Automated Segmentation of Dental CBCT Image Using an Improved U-Net Network

Journal: Journal of Clinical Medicine Research DOI: 10.32629/jcmr.v4i2.1191

Zeyu Chen1, Senyang Chen2, Songming Liu3

1. School of Computer Science and Artificial Intelligence, Changzhou University, Changzhou 213164, China
2. College of Computer Science and Technology, Zhejiang University of Technology, Hangzhou 310023, China
3. Hangzhou Zhejiang Chuanran Intelligent Technology Group Co., Ltd. Hangzhou 310000, China

Abstract

In the field of clinical dental medicine, Cone Beam Computed Tomography (CBCT) is a useful tool for the measurement of various dimensions related to the oral cavity, including height and thickness. This provides invaluable guidance and reference for risk assessment in orthodontic treatment, selection of treatment plans and implant treatment. However, segmentation of the teeth region from CBCT images is a daunting task due to complex root morphology and indistinct boundaries between the root and the alveolar bone. Manual annotation of the teeth area is resource-intensive, and deep learning-based segmentation methods are susceptible to noise, reducing their efficiency. To tackle these complexities, a multi-filter attention module is proposed in this paper, which effectively minimizes the noise in CBCT images through utilization of multiple filters and self-attention techniques. Additionally, an Improved U-Net model is proposed, where the original convolution block in the U-Net is replaced with a Double ConvNeXt block to yield better network performance. Experimentally, the proposed Improved U-Net method showed remarkable progress as it achieved a Dice Similarity Coefficient of 86.95% in oral CBCT image segmentation, surpassing existing models and affirming the effectiveness and advancedness of the proposed model and method.

Keywords

deep learning, images segmentation, medical image processing, convolutional neural networks, image denoising

References

[1]Loubele M, Bogaerts R, Van Dijck E, et al. Comparison between effective radiation dose of CBCT and MSCT scanners for dentomaxillofacial applications[J]. European journal of radiology, 2009, 71(3): 461-468.
[2]Mah J K, Yi L, Huang R C, et al. Advanced applications of cone beam computed tomography in orthodontics[C]//Seminars in Orthodontics. WB Saunders, 2011, 17(1): 57-71.
[3]Hwang J J, Jung Y H, Cho B H, et al. An overview of deep learning in the field of dentistry[J]. Imaging science in dentistry, 2019, 49(1): 1-7.
[4]Liu X, Song L, Liu S, et al. A review of deep-learning-based medical image segmentation methods[J]. Sustainability, 2021, 13(3): 1224.
[5]Hesamian M H, Jia W, He X, et al. Deep learning techniques for medical image segmentation: achievements and challenges[J]. Journal of digital imaging, 2019, 32: 582-596.
[6]Shaheen E, Leite A, Alqahtani K A, et al. A novel deep learning system for multi-class tooth segmentation and classification on cone beam computed tomography. A validation study[J]. Journal of Dentistry, 2021, 115: 103865.
[7]Minnema J, van Eijnatten M, Hendriksen A A, et al. Segmentation of dental cone‐beam CT scans affected by metal artifacts using a mixed‐scale dense convolutional neural network[J]. Medical physics, 2019, 46(11): 5027-5035.
[8]Wang X, Meng X, Yan S. Deep learning-based image segmentation of cone-beam computed tomography images for oral lesion detection[J]. Journal of Healthcare Engineering, 2021, 2021: 1-7.
[9]Ma J, Yang X. Automatic dental root CBCT image segmentation based on CNN and level set method[C]//Medical Imaging 2019: Image Processing. SPIE, 2019, 10949: 668-674.
[10]Corbella S, Srinivas S, Cabitza F. Applications of deep learning in dentistry[J]. Oral Surgery, Oral Medicine, Oral Pathology and Oral Radiology, 2021, 132(2): 225-238.
[11]Minnema J, van Eijnatten M, Hendriksen A A, et al. Segmentation of dental cone‐beam CT scans affected by metal artifacts using a mixed‐scale dense convolutional neural network[J]. Medical physics, 2019, 46(11): 5027-5035.
[12]Wang H, Minnema J, Batenburg K J, et al. Multiclass CBCT image segmentation for orthodontics with deep learning[J]. Journal of dental research, 2021, 100(9): 943-949.
[13]Zheng Z, Yan H, Setzer F C, et al. Anatomically constrained deep learning for automating dental CBCT segmentation and lesion detection[J]. IEEE Transactions on Automation Science and Engineering
[14]Duan W, Chen Y, Zhang Q, et al. Refined tooth and pulp segmentation using U-Net in CBCT image[J]. Dentomaxillofacial Radiology, 2021, 50(6): 20200251.
[15]Duong D Q, Nguyen K C T, Kaipatur N R, et al. Fully automated segmentation of alveolar bone using deep convolutional neural networks from intraoral ultrasound images[C]//2019 41st Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC). IEEE, 2019: 6632-6635.
[16]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[J]. IEEE transactions on pattern analysis and machine intelligence, 2021, 44(10): 6562-6568.
[17]Awari H, Subramani N, Janagaraj A, et al. Three‐dimensional dental image segmentation and classification using deep learning with tunicate swarm algorithm[J]. Expert Systems, 2022: e13198.
[18]Lahoud P, Diels S, Niclaes L, et al. Development and validation of a novel artificial intelligence driven tool for accurate mandibular canal segmentation on CBCT[J]. Journal of dentistry, 2022, 116: 103891.
[19]Ronneberger O, Fischer P, Brox T. U-net: Convolutional networks for biomedical image segmentation[C]//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, 2015: 234-241.
[20]Simonyan K, Zisserman A. Very deep convolutional networks for large-scale image recognition[J]. arXiv preprint arXiv:1409.1556, 2014.
[21]Liu Z, Song Y Q, Sheng V S, et al. Liver CT sequence segmentation based with improved U-Net and graph cut[J]. Expert Systems with Applications, 2019, 126: 54-63.
[22]Chen K, Xuan Y, Lin A, et al. Lung computed tomography image segmentation based on U-Net network fused with dilated convolution[J]. Computer Methods and Programs in Biomedicine, 2021, 207: 106170.
[23]Diniz J O B, Ferreira J L, Cortes O A C, et al. An automatic approach for heart segmentation in CT scans through image processing techniques and Concat-U-Net[J]. Expert Systems with Applications, 2022, 196: 116632.
[24]Oktay O, Schlemper J, Folgoc L L, et al. Attention u-net: Learning where to look for the pancreas[J]. arXiv preprint arXiv:1804.03999, 2018.
[25]Chen J, Lu Y, Yu Q, et al. Transunet: Transformers make strong encoders for medical image segmentation[J]. arXiv preprint arXiv:2102.04306, 2021.
[26]Cao H, Wang Y, Chen J, et al. Swin-unet: Unet-like pure transformer for medical image segmentation[J]. preprint arXiv:2105.05537, 2021.
[27]Badrinarayanan V, Kendall A, Cipolla R. Segnet: A deep convolutional encoder-decoder architecture for image segmentation[J]. IEEE transactions on pattern analysis and machine intelligence, 2017, 39(12): 2481-2495.
[28]Chen L C, Zhu Y, Papandreou G, et al. Encoder-decoder with atrous separable convolution for semantic image segmentation[C]//Proceedings of the European conference on computer vision (ECCV). 2018: 801-818.
[29]He K, Zhang X, Ren S, et al. Deep residual learning for image recognition[C]//Proceedings of the IEEE conference on computer vision and pattern recognition. 2016: 770-778.
[30]Woo S, Debnath S, Hu R, et al. ConvNeXt V2: Co-designing and Scaling ConvNets with Masked Autoencoders[J]. arXiv preprint arXiv:2301.00808, 2023.

Copyright © 2023 Zeyu Chen, Senyang Chen, Songming Liu

Creative Commons License
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License