An automatic cephalometric landmark detection method based on heatmap regression and Monte Carlo dropout.
Journal:
Annual International Conference of the IEEE Engineering in Medicine and Biology Society. IEEE Engineering in Medicine and Biology Society. Annual International Conference
PMID:
38083204
Abstract
Cephalometric analysis plays an important role in orthodontic diagnosis and treatment planning. It depends on the detection of multiple landmarks, while the process is time-consuming and tedious. Although some deep learning-based automatic landmark detection algorithms have achieved excellent performance, most of them adopt multi-stage models increasing the complexity and detection time. Meanwhile, few studies focused on the uncertainty of detection results, thereby ignoring its significant clinical value. In this paper, we propose a novel approach based on heatmap regression for landmark detection, which can achieve competitive accuracy and good robustness with only one step. Furthermore, by applying Monte Carlo dropout to a U-shaped convolutional neural network, we can obtain not only the coordinate of each landmark but also the corresponding simple uncertainty, so that doctors can pay more attention to those landmarks with higher uncertainty. The evaluation results showed the mean radial error is 1.39±1.06mm and the average successful detection rate is 79.65%, 97.22% within 2mm, 4mm for the IEEE ISBI2015 Test Dataset 1, the indicators for the IEEE ISBI2015 Test Dataset 2 are 1.33±0.93mm, 80.05% and 97.53%, respectively. Our method has the potential to become an assistant tool in clinical practice. Automatic and accurate detection with uncertainty analysis is expected to help guide the doctor's judgment.