DOI QR코드

DOI QR Code

Improving Accuracy of Instance Segmentation of Teeth

  • Jongjin Park (Department of Computer Engineering, Chungwoon University)
  • Received : 2024.01.15
  • Accepted : 2024.02.09
  • Published : 2024.02.29

Abstract

In this paper, layered UNet with warmup and dropout tricks was used to segment teeth instantly by using data labeled for each individual tooth and increase performance of the result. The layered UNet proposed before showed very good performance in tooth segmentation without distinguishing tooth number. To do instance segmentation of teeth, we labeled teeth CBCT data according to tooth numbering system which is devised by FDI World Dental Federation notation. Colors for labeled teeth are like AI-Hub teeth dataset. Simulation results show that layered UNet does also segment very well for each tooth distinguishing tooth number by color. Layered UNet model using warmup trick was the best with IoU values of 0.80 and 0.77 for training, validation data. To increase the performance of instance segmentation of teeth, we need more labeled data later. The results of this paper can be used to develop medical software that requires tooth recognition, such as orthodontic treatment, wisdom tooth extraction, and implant surgery.

Keywords

Acknowledgement

This work was conducted with support from the 2023 Chungwoon University Academic Research Fund.

References

  1. T. He, et al., "Bag of Tricks for Image Classification with Convolutional Neural Networks," in 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA, 2019 pp. 558-567. DOI: https://doi.org/10.48550/arXiv.1812.01187
  2. J. J. Park, et al., "Teeth Segmentation for Orthodontics based on Deep Learning," The Transactions of the Korean Institute of Electrical Engineers, vol. 72, no. 3, pp. 440~446, 2023. DOI: https://doi.org/10.5370/KIEE.2023.72.3.440
  3. J. J. Park, et al., "Semantic Segmentation of Teeth using Layered UNet," The Transactions of the Korean Institute of Electrical Engineers, vol. 72, no. 11, pp. 1470~1476, 2023. DOI: https://doi.org/10.5370/KIEE.2023.72.11.1470
  4. https://keymakr.com/blog/semantic-segmentation-uses-and-applications/
  5. https://en.wikipedia.org/wiki/FDI_World_Dental_Federation_notation
  6. K. He, X. Zhang, S. Ren, and J. Sun, "Deep residual learning for image recognition," Proceedings of the IEEE conference on computer vision and pattern recognition, pages 770-778, 2016.
  7. P. Goyal, P. Dollar, R. B. Girshick, P. Noordhuis, ' L. Wesolowski, A. Kyrola, A. Tulloch, Y. Jia, and K. He, "Accurate, large minibatch SGD: training imagenet in 1 hour," CoRR, abs/1706.02677, 2017.
  8. Yo-wei Chen, et al., "Artificial intelligence in dentistry: current applications and future perspectives," QUINTESSENCE INTERNATIONAL, vol. 51, no. 3, March 2020. DOI: https://doi.org/10.3290/j.qi.a43952
  9. H. Ding et al., "Artificial intelligence in dentistry-A review," vol. 4, Front. Dent. Med. Sec. Dental Materials, 20 February 2023 DOI: https://doi.org/10.3389/fdmed.2023.1085251
  10. Khanagar SB, Al-ehaideb A, Maganur PC, Vishwanathaiah S, Patil S, Baeshen HA, et al, "Developments, application, and performance of artificial intelligence in dentistry-a systematic review," J Dent Sci. 16(1):508-22, 2021. DOI: https://doi.org/10.1016/j.jds.2020.06.019