Diagnosis test for classification of skeletal discrepancies by machine learning based on three-dimensional facial profile

毛渤淳,田雅婧,肖宇嘉,周彦恒,李晶

DOI: https://doi.org/10.3760/cma.j.cn115797-20230404-23402

2023-01-01

Abstract:Objective:This study aims to distinguish sagittal and vertical skeletal discrepancies using three machine learning methods based on three-dimensional (3D) facial profile, and a comparison among the three methods was carried out, and the discriminant features were screened.Methods:Three dimensional facial profiles were collected from 292 patients before orthodontic treatment. Cephalometric analysis was conducted for diagnosis of sagittal and vertical skeletal discrepancies. The 3D morphable model method was used to obtain soft tissue landmarks. The coordinates of landmarks, along with their geometric features and demographic characteristics, were set as features for machine learning. Random Forest (RF), AdaBoost, and Multi-Layer Perceptron (MLP) were used for skeletal discrepancies discrimination. Each model was evaluated using the receiver operating characteristic curve (ROC) and the area under the curve (AUC) was calculated. Features with the highest Gini importance were selected.Results:The AUC of MLP was the largest for sagittal pattern (0.92 for Class Ⅱ and 0.97 for Class Ⅲ). For the vertical pattern, Random Forest had the largest AUC (hypodivergence 0.84, hyperdivergence 0.88). The accuracy of discrimination of the three algorithms for sagittal pattern (Class Ⅱ: RF 90.0%, AdaBoost 85.7%, MLP 95.0%; Class Ⅲ: RF 84.6%, AdaBoost 92.3%, MLP 93.3%) was better than that of vertical pattern (hypodivergence: RF 68.2%, AdaBoost 72.0%, MLP 76.9%; hyperdivergence: RF 76.2%, AdaBoost 77.8%, MLP 69.6%). The results of feature screening showed that the most significant features for the Class Ⅱ and hyperdivergence were the angle of the interbrow point to the premental point to the subnasal septum point, and the angle of the midpoint of the bilateral temple point to the midpoint of the bilateral cheilion point to the premental point.Conclusions:Machine learning methods can assist in identifying sagittal and vertical skeletal deformities to a certain extent through three-dimensional images, and the features with the most significance for discrimination of skeletal patterns were revealed.