MRCM-UCTransNet: Automatic and Accurate 3D Tooth Segmentation Network From Cone-Beam CT Images

被引:1
作者
Wen, Xinyang [1 ,2 ]
Liu, Zhuoxuan [1 ,2 ]
Chu, Yanbo [1 ,2 ]
Le, Min [2 ]
Li, Liang [2 ,3 ]
机构
[1] Tsinghua Univ, WeiYang Coll, Beijing, Peoples R China
[2] Tsinghua Univ, Dept Engn Phys, Beijing, Peoples R China
[3] Tsinghua Univ, Inst Precis Med, Beijing, Peoples R China
基金
北京市自然科学基金;
关键词
attention mechanism; cone-beam CT; multi-scale residual convolution; tooth segmentation; UCTransNet;
D O I
10.1002/ima.23139
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Many scenarios in dental clinical diagnosis and treatment require the segmentation and identification of a specific tooth or the entire dentition in cone-beam computed tomography (CBCT) images. However, traditional segmentation methods struggle to ensure accuracy. In recent years, there has been significant progress in segmentation algorithms based on deep learning, garnering considerable attention. Inspired by models from present neuro networks such as UCTransNet and DC-Unet, this study proposes an MRCM-UCTransNet for accurate three-dimensional tooth segmentation from cone-beam CT images. To enhance feature extraction while preserving the multi-head attention mechanism, a multi-scale residual convolution module (MRCM) is integrated into the UCTransNet architecture. This modification addresses the limitations of traditional segmentation methods and aims to improve accuracy in tooth segmentation from CBCT images. Comparative experiments indicate that, in the situation with a specific image size and small data volume, the proposed method exhibits certain advantages in segmentation accuracy and precision. Compared to traditional Unet approaches, MRCM-UCTransNet's dice accuracy is improved by 7%, while its sensitivity is improved by about 10%. These findings highlight the efficacy of the proposed approach, particularly in scenarios with specific image size constraints and limited data availability. The proposed MRCM-UCTransNet algorithm integrates the latest architectural advancements in the Unet model which achieves effective segmentation of six types of teeth within the tooth. It was proved to be efficient for image segmentation on small datasets, requiring less training time and fewer parameters.
引用
收藏
页数:13
相关论文
共 33 条
[1]  
Akhoondali H., 2009, Journal of Applied Sciences, V9, P2031, DOI 10.3923/jas.2009.2031.2044
[2]   SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation [J].
Badrinarayanan, Vijay ;
Kendall, Alex ;
Cipolla, Roberto .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) :2481-2495
[3]  
Bahdanau D, 2016, Arxiv, DOI [arXiv:1409.0473, 10.48550/arXiv.1409.0473, DOI 10.48550/ARXIV.1409.0473]
[4]  
Cui Z., ToothNet: Automatic Tooth Instance Segmentation and Identification From Cone Beam CT Images, in IEEE/CVF Conference on Computer Vision and Pattern Recognition (Long Beach, CA: IEEE, 2019), 63616370
[5]   A fully automatic AI system for tooth and alveolar bone segmentation from cone-beam CT images [J].
Cui, Zhiming ;
Fang, Yu ;
Mei, Lanzhuju ;
Zhang, Bojun ;
Yu, Bo ;
Liu, Jiameng ;
Jiang, Caiwen ;
Sun, Yuhang ;
Ma, Lei ;
Huang, Jiawei ;
Liu, Yang ;
Zhao, Yue ;
Lian, Chunfeng ;
Ding, Zhongxiang ;
Zhu, Min ;
Shen, Dinggang .
NATURE COMMUNICATIONS, 2022, 13 (01)
[6]   Refined tooth and pulp segmentation using U-Net in CBCT image [J].
Duan, Wei ;
Chen, Yufei ;
Zhang, Qi ;
Lin, Xiang ;
Yang, Xiaoyu .
DENTOMAXILLOFACIAL RADIOLOGY, 2021, 50 (06)
[7]   Understanding Deep Learning Techniques for Image Segmentation [J].
Ghosh, Swarnendu ;
Das, Nibaran ;
Das, Ishita ;
Maulik, Ujjwal .
ACM COMPUTING SURVEYS, 2019, 52 (04)
[8]   Generative Adversarial Networks [J].
Goodfellow, Ian ;
Pouget-Abadie, Jean ;
Mirza, Mehdi ;
Xu, Bing ;
Warde-Farley, David ;
Ozair, Sherjil ;
Courville, Aaron ;
Bengio, Yoshua .
COMMUNICATIONS OF THE ACM, 2020, 63 (11) :139-144
[9]   A knowledge-based algorithm for automatic detection of cephalometric landmarks on CBCT images [J].
Gupta, Abhishek ;
Kharbanda, Om Prakash ;
Sardana, Viren ;
Balachandran, Rajiv ;
Sardana, Harish Kumar .
INTERNATIONAL JOURNAL OF COMPUTER ASSISTED RADIOLOGY AND SURGERY, 2015, 10 (11) :1737-1752
[10]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778