Abstract
Computer-assisted modeling of patient-specific 3D teeth is a clinically important technology for the development of dental diagnosis and treatment. This technology often relies on accurately segmenting the target tooth and its surrounding tissues from CBCT images. Most of the previous methods consume extensive memory for generating bounding box proposals in a detection manner, while in this paper, we propose a novel stagewise tooth instance segmentation framework from localization to segmentation. Specifically, our method follows the process of tooth centroid prediction, candidate centroid analysis, and mapping of centroids to accurately localize the ROI of individual teeth, instead of generating bounding box proposals for tooth positioning regression. To improve the segmentation quality, we propose a new loss function referred to as potential energy loss, which measures the feature similarity among voxels in a neighborhood to focus more on local information, regulating potential energy to obtain optimal segmentation. Moreover, the proposed fine segmentation network introduces a dual-branch structure and spectrum filter connections to enhance hierarchical features and anti-noise capability. Experimental results demonstrate that the proposed method surpasses state-of-the-art methods with improvements of 1.05%, 5.77%, and 16.67% on average DSC, HD95, and ASSD, respectively.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Notes
- 1.
This dataset is available in public from [9] only for research use.
References
Majanga, V., Viriri, S.: Dental images’ segmentation using threshold connected component analysis. Computational Intelligence and Neuroscience 2021 (2021)
Syuhada, F., et al.: Multi-projection segmentation on dental cone beam computed tomography images using level set method. J. Computer Science and Informatics Eng. (J-Cosine). 5, 130–139 (2021)
Qaddoura, R., Manaseer, W.A., Abushariah, M.A.M., et al.: Dental radiography segmentation using expectation-maximization clustering and grasshopper optimizer. Multimedia Tools and Appl. 79, 22027–22045 (2020)
Cui, Z., Li, C., Wang, W.: ToothNet: automatic tooth instance segmentation and identification from cone beam CT images. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 6368–6377 (2019)
Jang, T.J., Kim, K.C., Cho, H.C., et al.: A fully automated method for 3D individual tooth identification and segmentation in dental CBCT. IEEE Trans. Pattern Anal. Mach. Intell. 44(10), 6562–6568 (2021)
Ezhov, M., Zakirov, A., Gusarev, M.: Coarse-to-fine volumetric segmentation of teeth in cone-beam ct. 2019 IEEE 16th International Symposium on Biomedical Imaging (ISBI 2019). IEEE, pp. 52–56(2019)
Chen, Y., Du, H., Yun, Z., et al.: Automatic segmentation of individual tooth in dental CBCT images from tooth surface map by a multi-task FCN. IEEE Access. 8, 97296–97309 (2020)
Li, P., Liu, Y., Cui, Z., et al.: Semantic graph attention with explicit anatomical association modeling for tooth segmentation from CBCT images. IEEE Trans. Med. Imaging 41(11), 3116–3127 (2022)
Cui, Z., Fang, Y., Mei, L., et al.: A fully automatic AI system for tooth and alveolar bone segmentation from cone-beam CT images. Nat. Commun. 13(1), 2096 (2022)
Ronneberger, O., Fischer, P., Brox, T.: U-net: convolutional networks for biomedical image segmentation. Medical Image Computing and Computer-Assisted Intervention–MICCAI 2015: 18th International Conference, Munich, Germany, October 5–9, 2015, Proceedings, Part III 18. Springer International Publishing, pp. 234–241 (2015)
Geng, Z., Sun, K., Xiao, B., et al.: Bottom-up human pose estimation via disentangled keypoint regression. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 14676–14686 (2021)
Yu, F., Koltun, V.: Multi-Scale Context Aggregation by Dilated Convolutions. arXiv preprint arXiv:1511.07122 (2015)
Wang, P., Chen, P., Yuan, Y., et al.: Understanding convolution for semantic segmentation. 2018 IEEE Winter Conference on Applications of Computer Vision (WACV). IEEE, pp. 1451–1460 (2018)
Park, J.S., Fadnavis, S., Garyfallidis, E..: EVC-Net: Multi-scale V-Net with Conditional Random Fields for Brain Extraction. arXiv preprint arXiv:2206.02837 (2022)
Strauss, D.J.: Hammersley–clifford theorem. Encyclopedia of Statistical Sciences 5 (2004)
LeCun, Y., Bottou, L., Bengio, Y., et al.: Gradient-based learning applied to document recognition. Proc. IEEE 86(11), 2278–2324 (1998)
Vaswani, A., Shazeer, N., Parmar, N., et al.: Attention is all you need. Advances in Neural Information Processing Syst. 30 (2017)
Ho, J., Jain, A., Abbeel, P.: Denoising diffusion probabilistic models. Adv. Neural. Inf. Process. Syst. 33, 6840–6851 (2020)
Chen, J., Lu, Y., Yu, Q., et al.: Transunet: Transformers make Strong Encoders for Medical Image Segmentation. arXiv preprint arXiv:2102.04306 (2021)
Peiris, H., Hayat, M., Chen, Z., et al.: A robust volumetric transformer for accurate 3d tumor segmentation. Medical Image Computing and Computer Assisted Intervention–MICCAI 2022: 25th International Conference, Singapore, September 18–22, 2022, Proceedings, Part V. Cham: Springer Nature Switzerland, pp: 162–172 (2022). https://doi.org/10.1007/978-3-031-16443-9_16
Wu, J., Fang, H., Zhang, Y., et al.: MedSegDiff: Medical Image Segmentation with Diffusion Probabilistic Model. arXiv preprint arXiv:2211.00611 (2022)
Amit, T., Nachmani, E., Shaharbany, T., et al.: Segdiff: Image Segmentation with Diffusion Probabilistic Models. arXiv preprint arXiv:2112.00390 (2021)
Isensee, F., Petersen, J., Klein, A., et al.: nnu-net: Self-Adapting Framework for u-Net-based Medical Image Segmentation. arXiv preprint arXiv:1809.10486 (2018)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Cao, K. et al. (2024). A Stagewise Deep Learning Framework for Tooth Instance Segmentation in CBCT Images. In: Liu, F., Sadanandan, A.A., Pham, D.N., Mursanto, P., Lukose, D. (eds) PRICAI 2023: Trends in Artificial Intelligence. PRICAI 2023. Lecture Notes in Computer Science(), vol 14325. Springer, Singapore. https://doi.org/10.1007/978-981-99-7019-3_38
Download citation
DOI: https://doi.org/10.1007/978-981-99-7019-3_38
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-7018-6
Online ISBN: 978-981-99-7019-3
eBook Packages: Computer ScienceComputer Science (R0)