Segmentation of teeth in panoramic dental X-ray images using U-Net with a loss function weighted on the tooth edge

Radiol Phys Technol. 2021 Mar;14(1):64-69. doi: 10.1007/s12194-020-00603-1. Epub 2021 Jan 5.

Abstract

Panoramic dental X-ray imaging is an established method for the diagnosis of dental problems. However, the resolution of panoramic dental X-ray images is relatively low. Thus, early lesions are often overlooked. As the first step in the development of a computer-aided diagnosis scheme for panoramic dental X-ray images, we propose a computerized method for the segmentation of teeth using U-Net with a loss function weighted on the tooth edge. Our database consisted of 162 panoramic dental X-ray images. The training dataset consisted of 102 images, while the remaining 60 images were used as the test dataset. The loss function obtained by the cross entropy (CE) in the entire image is usually used in training U-Net. To improve the segmentation accuracy of the tooth edge, a loss function weighted on the tooth edge is proposed by adding the CE in the tooth edge region to the CE for the entire image. The mean Jaccard index and Dice index for U-Net with the loss function combining the CEs for the entire image and tooth edge were 0.864 and 0.927, respectively, which were significantly larger than those for U-Net with the CE for the entire image (0.802 and 0.890, p < 0.001) and U-Net with the CE for the tooth edge (0.826 and 0.905, p < 0.001). U-Net with the new loss function exhibited a higher segmentation accuracy of the tooth in panoramic dental X-ray images than that obtained by U-Net with the conventional loss function.

Keywords: Loss function; Panoramic dental X-ray; Tooth segmentation; U-net.

MeSH terms

  • Image Processing, Computer-Assisted*
  • Radiography, Panoramic
  • Tooth* / diagnostic imaging
  • X-Rays