ImplantMamba: Long-range Sequential Modeling Mamba For Dental Implant Position Prediction

TL;DR

ImplantMamba is a novel AI architecture that combines CNNs and Mamba layers for long-range modeling to accurately predict dental implant positions and angles from medical images.

Contribution

The paper introduces a hybrid CNN-Mamba network with a slope-coupled prediction branch for improved implant position and angulation prediction.

Findings

ImplantMamba outperforms existing methods on a large-scale dental dataset.

The approach effectively models global contextual dependencies in scan volumes.

Coupling implant position with slope improves prediction consistency.

Abstract

In the design of surgical guides for implant placement, determining the precise implant position is a critical step. However, the implant region itself is often characterized by a lack of distinctive texture in medical images. Consequently, artificial intelligence (AI) models must infer the correct implant position and angulation (slope) primarily by analyzing the texture of the surrounding teeth, which poses a significant challenge. To address this, we propose ImplantMamba, a network architecture designed for long-range sequential modeling to integrate texture information from adjacent teeth. Our approach explicitly couples the regression of the implant position with its slope. The core of ImplantMamba is a hybrid encoder that combines Convolutional Neural Networks (CNNs) with Mamba layers. This design enables the network to hierarchically extract local anatomical features through CNNs while simultaneously modeling global contextual dependencies across the entire scan volume via Mamba's selective scan operations, leading to a more comprehensive understanding of the implant site. Furthermore, we introduce a Slope-Coupled Prediction Branch (SCP). This branch is designed to connect the prediction of implant position with the slope, ensuring internal consistency and anatomical plausibility by thereby enforcing a coherent relationship between the predicted implant location and its angulation. Extensive experiments on a large-scale dental implant dataset demonstrate that the proposed ImplantMamba achieves superior performance compared to existing methods.