TCEIP: Text Condition Embedded Regression Network for Dental Implant Position Prediction
This work addresses a domain-specific problem for dentists by improving implant position prediction in complex cases with multiple missing teeth, representing an incremental advancement.
The paper tackles the problem of predicting dental implant positions for multiple missing teeth, where existing methods fail, by integrating weak supervision text into a regression network, achieving superior performance over existing methods in experiments.
When deep neural network has been proposed to assist the dentist in designing the location of dental implant, most of them are targeting simple cases where only one missing tooth is available. As a result, literature works do not work well when there are multiple missing teeth and easily generate false predictions when the teeth are sparsely distributed. In this paper, we are trying to integrate a weak supervision text, the target region, to the implant position regression network, to address above issues. We propose a text condition embedded implant position regression network (TCEIP), to embed the text condition into the encoder-decoder framework for improvement of the regression performance. A cross-modal interaction that consists of cross-modal attention (CMA) and knowledge alignment module (KAM) is proposed to facilitate the interaction between features of images and texts. The CMA module performs a cross-attention between the image feature and the text condition, and the KAM mitigates the knowledge gap between the image feature and the image encoder of the CLIP. Extensive experiments on a dental implant dataset through five-fold cross-validation demonstrated that the proposed TCEIP achieves superior performance than existing methods.