An open deep learning-based framework and model for tooth instance segmentation in dental CBCT.
Authors
Affiliations (4)
Affiliations (4)
- Periodontology & Implant Dentistry, Faculty of Dentistry, The Prince Philip Dental Hospital, The University of Hong Kong, 34 Hospital Road, Sai Ying Pun, Hong Kong.
- Department of Stomatology, Nanjing Drum Tower Hospital, The Affiliated Hospital of Nanjing University Medical School, No. 321 Zhongshan Road, Nanjing, Jiangsu, China.
- Applied Oral Sciences & Community Dental Care, Faculty of Dentistry, The Prince Philip Dental Hospital, The University of Hong Kong, 34 Hospital Road, Sai Ying Pun, Hong Kong.
- Periodontology & Implant Dentistry, Faculty of Dentistry, The Prince Philip Dental Hospital, The University of Hong Kong, 34 Hospital Road, Sai Ying Pun, Hong Kong. [email protected].
Abstract
Current dental CBCT segmentation tools often lack accuracy, accessibility, or comprehensive anatomical coverage. To address this, we constructed a densely annotated dental CBCT dataset and developed a deep learning model, OraSeg, for tooth-level instance segmentation, which is then deployed as a one-click tool and made freely accessible for non-commercial use. We established a standardized annotated dataset covering 35 key oral anatomical structures and employed UNetR as the backbone network, combining Swin Transformer and the spatial Mamba module for multi-scale residual feature fusion. The OralSeg model was designed and optimized for precise instance segmentation of dental CBCT images, and integrated into the 3D Slicer platform, providing a graphical user interface for one-click segmentation. OralSeg had a Dice similarity coefficient of 0.8316 ± 0.0305 on CBCT instance segmentation compared to SwinUNETR and 3D U-Net. The model significantly improves segmentation performance, especially in complex oral anatomical structures, such as apical areas, alveolar bone margins, and mandibular nerve canals. The OralSeg model presented in this study provides an effective solution for instance segmentation of dental CBCT images. The tool allows clinical dentists and researchers with no AI background to perform one-click segmentation, and may be applicable in various clinical and research contexts. OralSeg can offer researchers and clinicians a user-friendly tool for tooth-level instance segmentation, which may assist in clinical diagnosis, educational training, and research, and contribute to the broader adoption of digital dentistry in precision medicine.