Classification of skeletal discrepancies by machine learning based on three-dimensional facial scans.

Journal: International journal of oral and maxillofacial surgery
Published Date:

Abstract

The aim of this study was to use machine learning (ML) to classify sagittal and vertical skeletal discrepancies in three-dimensional (3D) facial scans, as well as to evaluate shape variability. 3D facial scans from 435 pre-orthodontic patients were subjected to cephalometric analysis and 3D facial landmark identification. Three ML models were used for the discrimination of skeletal discrepancy: random forest, AdaBoost, and multi-layer perceptron. Each model was evaluated by receiver operating characteristic curve and calculating the area under the curve (AUC). Principal component analysis was conducted to evaluate shape variability. The AUCs for Class II and III patients ranged from 0.91 to 0.95. Random forest achieved the highest accuracy for sagittal classification (88.5% for Class II, 95.5% for Class III). Multi-layer perceptron exhibited the best performance for vertical classification (accuracy of 78.8% for hypodivergent, 86.2% for hyperdivergent). Six principal components explained 94.0% of facial morphology variation. ML methods show promise for assisting in the discrimination of sagittal and vertical skeletal discrepancies based on 3D facial scans. 3D facial soft tissue features appear to be suitable for the discrimination of skeletal discrepancies in most cases.

Authors

  • B Mao
    Department of Orthodontics, Peking University School and Hospital of Stomatology, Beijing, China; National Centre for Stomatology, Beijing, China; National Clinical Research Centre for Oral Diseases, Beijing, China; National Engineering Research Centre of Oral Biomaterials and Digital Medical Devices, Beijing, China; Beijing Key Laboratory of Digital Stomatology, Beijing, China; Research Centre of Engineering and Technology for Computerized Dentistry Ministry of Health, Beijing, China.
  • Y Tian
    Melbourne Neuropsychiatry Centre, The University of Melbourne, Victoria, 3010, Australia.
  • Y Xiao
    Centers of System Biology, Data Information and Reproductive Health, School of Basic Medical Science, School of Basic Medical Science, Central South University, Changsha, 410008, Hunan, China.
  • J Li
    Department of Pulmonary and Critical Care Medicine, Beijing Anzhen Hospital, Capital Medical University, Beijing 100029, China.
  • Y Zhou
    Department of Radiation Oncology, First Affiliated Hospital, Bengbu Medical College, Bengbu, Anhui 233004, China.
  • X Wang
    3 Laboratory Animal Center, Wenzhou Medical University, Wenzhou, China.