Skip to main content

ORIGINAL RESEARCH article

Front. Bioeng. Biotechnol., 14 October 2024
Sec. Biomaterials
This article is part of the Research Topic Advanced Technologies for Oral and Craniomaxillofacial Therapy View all 3 articles

Evaluation of four machine learning methods in predicting orthodontic extraction decision from clinical examination data and analysis of feature contribution

  • 1Department of Orthodontics, Shanghai Stomatological Hospital and School of Stomatology, Fudan University, Shanghai, China
  • 2Shanghai Key Laboratory of Craniomaxillofacial Development and Diseases, Fudan University, Shanghai, China
  • 3School of Stomatology, Fudan University, Shanghai, China
  • 4School of Medical Technology, Shanghai University of Medicine and Health Sciences, Shanghai, China
  • 5Department of Cariology and Endodontology, College of Stomatology, Shanghai Ninth People’s Hospital, Shanghai Jiao Tong University School of Medicine, Shanghai, China
  • 6National Clinical Research Center for Oral Diseases, National Center for Stomatology, Shanghai, China
  • 7Shanghai Key Laboratory of Stomatology and Shanghai Research Institute of Stomatology, Shanghai, China
  • 8Department of Oral and Maxillofacial-Head Neck Oncology, College of Stomatology, Shanghai Ninth People’s Hospital, Shanghai Jiao Tong University School of Medicine, Shanghai, China

Introduction: The study aims to predict tooth extraction decision based on four machine learning methods and analyze the feature contribution, so as to shed light on the important basis for experts of tooth extraction planning, providing reference for orthodontic treatment planning.

Methods: This study collected clinical information of 192 patients with malocclusion diagnosis and treatment plans. This study used four machine learning strategies, including decision tree, random forest, support vector machine (SVM) and multilayer perceptron (MLP) to predict orthodontic extraction decisions on clinical examination data acquired during initial consultant containing Angle classification, skeletal classification, maxillary and mandibular crowding, overjet, overbite, upper and lower incisor inclination, vertical growth pattern, lateral facial profile. Among them, 30% of the samples were randomly selected as testing sets. We used five-fold cross-validation to evaluate the generalization performance of the model and avoid over-fitting. The accuracy of the four models was calculated for the training set and cross-validation set. The confusion matrix was plotted for the testing set, and 6 indicators were calculated to evaluate the performance of the model. For the decision tree and random forest models, we observed the feature contribution.

Results: The accuracy of the four models in the training set ranges from 82% to 90%, and in the cross-validation set, the decision tree and random forest had higher accuracy. In the confusion matrix analysis, decision tree tops the four models with highest accuracy, specificity, precision and F1-score and the other three models tended to classify too many samples as extraction cases. In the feature contribution analysis, crowding, lateral facial profile, and lower incisor inclination ranked at the top in the decision tree model.

Conclusion: Among the machine learning models that only use clinical data for tooth extraction prediction, decision tree has the best overall performance. For tooth extraction decisions, specifically, crowding, lateral facial profile, and lower incisor inclination have the greatest contribution.

1 Introduction

Whether to extract teeth is one of the most important decisions in orthodontic treatment planning. Its frequency has fluctuated over the years, the extraction percentage was 30% in 1953, reached 76% in 1968, and declined to 28% in 1993, this variation was due to considerations in outcome stability, facial esthetics, and technological changes (Proffit, 1994). To date, scholars have been studying and exploring orthodontic extraction decisions to obtain healthier, more stable and more esthetic orthodontic outcomes (Jackson et al., 2017; Proffit, 1994). Occlusion, stability and esthetics are the three goals for a successful treatment plan, but no single rule can give the orthodontist a simple way to decide how to reach these goals, the extraction decision is multi-factorial, involving crowding (Janson et al., 2014; Boley et al., 2003), overjet and overbite (Janson et al., 2003), Bolton ratio (Hasija et al., 2014), Angle and skeletal classifications (Ker et al., 2008), transverse dimension (midline discrepancy, facial asymmetries) (Chang et al., 2011), incisors angulation, presence of root resorption (Maués et al., 2015), soft-tissue profile (Konstantonis et al., 2013), etc. The extraction plan embodies the experience and wisdom of orthodontists, which is difficult and confusing for young or general practitioners (Liu et al., 2021).

With the popularization of big data and artificial intelligence, more and more studies are trying to use machine learning algorithms to assist in extraction decision-making (Khanagar et al., 2021). Commonly used machine learning methods in decision prediction include linear regression (Köktürk et al., 2024), tree models (Köktürk et al., 2024; Etemad et al., 2021; Suhail et al., 2020), support vector machines (Köktürk et al., 2024), neural networks (Köktürk et al., 2024; Jung and Kim, 2016; Li et al., 2019; Xie et al., 2010), etc. from simple to complex. They usually included model measurements and cephalometric data to train the models. Different validation strategies for over-fitting were applied in the previous studies. We took categorical variables that are often recorded in clinical diagnosis, so we chose tree models, support vector machines, and neural networks. Since the number of variables in this study is not very large, we hope to include features to the maximum extent, so we chose the random forest while also trying the decision tree.

The current study aimed to predict tooth extraction decision based on four machine learning methods and analyze the feature contribution. This study invited senior specialists of the orthodontic department of Shanghai Stomatological Hospital to note down the diagnosis and tooth extraction plan of the patients, and applied four machine learning method for prediction of extraction decision-making. We used cross-validation to measure the generalization ability of the model and avoid over-fitting, and calculated feature contribution to explain the key variables that clinicians value when determining extraction planning.

2 Materials and methods

2.1 Data collection

The clinical materials were collected from consecutive patients visiting the department of orthodontics of Shanghai Stomatological Hospital from 2018 to 2020, including pre-treatment plaster models, clinical examination data, and treatment plans for malocclusion. Plaster models were scanned and stored in STL format.

Two senior doctors were asked to fill in a standardized form to record the patient’s Angle classification, skeletal classification, maxillary and mandibular crowding, overjet, overbite, upper and lower incisor inclination, vertical growth pattern, lateral facial profile. Angle’s classification, crowding, overbite and overjet were measured from dental models. And lateral facial profile was observed from facial photographs. Skeletal classification, upper/lower incisor inclination, and vertical growth pattern were measured in the cephalometric radiographs, using Steiner analysis (SNA SNB ANB U1-SN U1-NA L1-NB MP-SN), Tweed analysis (FMA IMPA FMIA), Wit’s appraisal, Ricketts’s analysis (lower lip to E plane). Experts’ decisions were comprehensive judgments based on the combination of objective measurement indicators and clinical observations.

As for the extraction decisions, only the consensus of the two experts was recorded, otherwise a third expert was invited and the majority opinion prevailed. An endodontic expert was invited to evaluate the preservation value of the residual crown. We marked the non-extraction cases, and then for the extraction cases, we recorded the specific teeth that are to be extracted, although the prediction of the extraction pattern was not involved in this study. The research plan has been approved by the Institutional Review Board of Shanghai Stomatological Hospital [Hu Kou Fang Lun Shen (2017) No. 0005]. Written informed consent was obtained from all the participants.

2.2 Inclusion and exclusion criteria

(1) Complete permanent dentition;

(2) No congenitally missing teeth or impacted teeth;

(3) The tooth extraction in the treatment plan is the classic extraction mode, that is, symmetric extraction of premolars;

(4) Complete pre-treatment model, clinical examination data and orthodontics treatment plan information;

(5) Informed consent signed by the patient or the parent (for teenager under 18).

In total, 192 patients with complete information were included, among whom, 30% were randomly selected as testing set and the rest were divided as training set (Figure 1).

Figure 1
www.frontiersin.org

Figure 1. Flowchart of data allocation, model training, cross-validation and model testing.

2.3 Model training

Four machine learning frameworks were applied for prediction of extraction and non-extraction planning, including decision tree (DT), random forest (RF), support vector machine (SVM) and multilayer perceptron (MLP) (Figure 1).

Decision Trees are built by continuously splitting the data into binary nodes that acquire the largest information gain until the terminal node outputs the predictions of classification (Suhail et al., 2020).

Random Forest is an ensemble of decision trees. Each decision tree performs the same classification prediction individually, and the final classification, or output, is determined by taking the most common predictions for discrete variables or the average of predictions for continuous variables (Etemad et al., 2021). The number of trees was set to 1,000 in our study.

Support Vector Machine (SVM) can efficiently perform a non-linear classification using what is called the kernel trick, representing the data only through a set of pairwise similarity comparisons between the original data points using a kernel function, which transforms them into coordinates in the higher dimensional feature space (Leavitt et al., 2023).

Multilayer Perceptron (MLP) is one of the simplest forms of artificial neural network (ANN), composed of input layer, output layer and hidden layer(s). Each layer consists of multiple nodes, called neuron, fully connected with nodes at adjacent layers (Etemad et al., 2021).

Due to the limited number of features, we removed data containing missing values. And convert continuous data into discrete data using the commonly used classification in orthodontics. Since age and gender nearly had no contribution (0%) to the model in the preliminary experiment, they were removed from the features in the formal study. Finally, nine feature variables were Angle’s classification, skeletal classification, crowding, overbite, overjet, upper and lower incisor inclination, vertical growth pattern and lateral facial profile (Table 1) and one classification variable was extraction/non-extraction decision.

Table 1
www.frontiersin.org

Table 1. Definitions of nine feature variables.

Since there is a difference between the proportion of tooth extraction cases (144 cases) and non-tooth extraction cases (48 cases) in the dataset, in order to deal with the problem of imbalanced classification, we adopted the threshold moving method (Collell et al., 2018). According to the ROC curve of the training set, we select the optimal classification threshold to prevent the model from being “occupied” by the classification with more data.

2.4 Cross-validation

Cross-validation is often used to measure the generalization ability of a model (Jonathan et al., 2000; Karkkainen, 2014). In order to ensure adequate number of samples in the cross-validation set, this study used a five-fold cross-validation method. It divided the dataset into five parts, taking turns to use four of them as training set and one as validation set. In each validation, the four training sets will generate a model, and the validation set will be input into the model for classification task. The proportion of the number of times the classifier decision matches the ground truth to the total number of tests is calculated as the accuracy of each validation. The average accuracy of the five validations is taken as the accuracy of a five-fold cross-validation.

2.5 Model testing

In the dataset, 30% (58 samples) of all patients were randomly selected as the testing set. The above four machine learning models were used for testing. The results of the testing set were represented by confusion matrix, and the following indicators were calculated (Figure 2):

• Accuracy: correctly classified data points in the testing set.

• Balanced accuracy: the average of sensitivity and specificity.

• Sensitivity/recall: positive data points correctly classified as positive.

• Specificity: negative data points correctly classified as negative.

• Precision: data points classified as positive that are actually positive.

• F1-score: the harmonic mean of precision and sensitivity. Precision and sensitivity may affect each other. Although it is ideal for both to be high, but in reality, it is often the case that the precision is high and the sensitivity is low, or otherwise. Therefore, F1-score is an indication of both at the same time.

Figure 2
www.frontiersin.org

Figure 2. Confusion matrix and six measurements for evaluation of model performance (positive cases represent extraction classification and negative cases represent non-extraction classification).

We also plotted the receiver operating characteristic (ROC) curve to visually demonstrate the performance of the model and calculated the area under the curve (AUC).

2.6 Feature contribution

Feature contribution in tree models can be measured by the number of times a feature participates in building the tree and the cumulative value of information gain when it is used as a split node. In the process of building a tree, the more times the feature is used to split nodes, the more important role the feature plays in the decision process. Cumulative value of information gain represents the amount of reduction in information entropy when the feature is used for node splitting. The greater the information gain, the greater the contribution of the feature to improving the predictive ability of the model. In addition, early participation of a feature (at the upper nodes of the tree) is usually considered more important than late participation and the cumulative sum was calculated in a weighted manner to reflect the contribution of the features.

3 Results

3.1 Descriptive statistics

The dataset of this study includes 48 non-extraction cases and 144 extraction cases. The proportion of each feature variables is shown in Figure 3.

Figure 3
www.frontiersin.org

Figure 3. Proportion of each feature variables.

Since crowding often changes in similar directions in the upper and lower dentitions, we combined it into one variable-crowding, by calculating the maximum of crowding in upper and lower dentitions. However, the inclination of the upper and lower anterior teeth can appear completely opposite trends, such as buccal inclination in upper incisor and lingual inclination in lower incisor, so they were considered separately in this study. Open bite and deep overbite reflect different degrees of vertical discrepancy of the upper and lower anterior teeth in different directions, so they were combined into one variable as overbite (Table 1; Figure 3).

3.2 Evaluation of model accuracy

The accuracy of the four models in the training set and cross-validation set is shown in Figure 4. The accuracy of the four models in the training set ranges from 80% to 93%, and in the cross-validation set, the decision tree and random forest had higher accuracy. We can evaluate the generalization ability of the model by comparing the difference between the accuracy of cross-validation set and training set. Generally, over-fitting is often seen if the accuracy in the cross-validation set is significantly lower than that in the training set, but there is not a definite border (Xu and Goodacre, 2018). In this study, random forest and SVM showed a tendency of, but not absolute, over-fitting. The performance of decision tree and MLP was acceptable, and the accuracy of decision tree was greater than that of MLP.

Figure 4
www.frontiersin.org

Figure 4. Accuracy of the four machine learning models in training sets, cross-validation sets and testing sets.

For the testing set, we list the confusion matrices of the four models, in which, positive cases represent extraction classification and negative cases represent non-extraction classification (Figure 5). In the random forest and SVM, the accuracy of the testing set was also significantly lower than the training set (Figures 4, 6). Among the performance of these four models, the decision tree showed the best overall prediction performance with the best accuracy, balanced accuracy, specificity, precision and F1-score (Figure 6). It is worth noting that in a more detailed analysis, RF and MLP showed a low specificity, indicating that they tended to classify too many non-extraction cases into extraction group (Figure 6).

Figure 5
www.frontiersin.org

Figure 5. Confusion matrix of the testing sets of the four machine learning models.

Figure 6
www.frontiersin.org

Figure 6. Performance of the four models in testing set.

From the ROC curve, we can see that the decision tree model performs best, followed by SVM, and MLP is the weakest (Figure 7).

Figure 7
www.frontiersin.org

Figure 7. Receiver operating characteristic (ROC) curve and area under the curve (AUC) of the four models in testing set.

3.3 Evaluation of feature contribution

The tree model has strong interpretability and is significantly better than support vector machines and neural networks. In the decision tree model, crowding is the most important indicator for tooth extraction (30.20%), followed by lateral facial profile (26.00%), lower incisor inclination (13.30%), overbite (9.20%), upper anterior lip inclination (8.10%), and skeletal classification (7.30%) (Figure 8). In the random forest model, the most important indicator is lateral facial profile (29.4%), followed by overbite (6.80%), crowding (11.90%), upper (10.9%) and lower incisor inclination (10.20%), vertical growth pattern (7.90%), etc. (Figure 8).

Figure 8
www.frontiersin.org

Figure 8. Feature contribution in decision tree and random forest models.

Figure 9 shows one of the decision tree models generated in this study, which facilitates intuitive analysis of the specific role of each feature. The feature selection standard of the decision tree in this study is set to information gain (entropy). Entropy represents the uncertainty of the information of the node, taking a value between 0 and 1. The larger the entropy, the greater the uncertainty. That is to say, for the training set of this tree, the closer the entropy is to 0, the greater the confidence of this classification.

Figure 9
www.frontiersin.org

Figure 9. One example of a decision tree generated in our study.

The root node, or 0-layer node, used in the tree in Figure 9 are lateral facial profile, upper and lower incisor inclination, crowding, overbite, vertical growth pattern, skeletal classification. The root node is the lateral facial profile. The model is divided into two categories according to straight, convex and concave lateral facial profile. However, it is unable to make a tooth extraction prediction based on the lateral facial profile alone, so node splits into two branches as first-layer nodes. A more detailed analysis shows that the most commonly used feature was crowding (grey box in Figure 9), and when we look at the most left and right boxes in the second-level nodes, if the crowding score was between 1 and 3 (larger than 2 mm crowding), it tends to extraction decision, and when it is greater than 3 or equal to 4 (spacing), it tends to non-extraction decision. For vertical growth pattern (orange box in the third-level node), a score over 2 (hypodivergent) indicates non-extraction decision. Moreover, Class II and Class III classification (pink box, scored 2 and 3) points to tooth extraction and buccal incisor inclination (blue box in the fourth-level node, scored 1) suggests the tendency to tooth extraction.

4 Discussion

Orthodontic treatment planning embodies experience and wisdom of orthodontists, and the formulation of tooth extraction plans is one of the most critical decisions in orthodontic practice, and it is also one of the biggest challenges for young doctors and general practitioners (Liu et al., 2021). Although there are many factors that may influence orthodontic tooth extraction decisions. For most orthodontic cases, orthodontic experts have relatively consistent judgments on tooth extraction planning (Suhail et al., 2020; Xu and Huang, 2002). In this context, it is meaningful to perform automatic decision-making assisted by machine learning.

Because natural prevalence of different categories of malocclusion differs among different populations (Lombardo et al., 2020). At the same time, patients’ motivation to seek orthodontic treatment also varies. For example, Chinese patients may have stronger will to correct convex facial profile (Yin et al., 2014). Therefore, each category of malocclusion in this study cannot be completely balanced. If the proportion of each category of malocclusion is calibrated during the inclusion process, randomization cannot be ensured, and it may be difficult for the model to adapt to clinical reality.

Classification algorithm is an important technology in data mining. In this study, the decision tree was very effective in classifying the sample set. In orthodontic tooth extraction decisions, random forests usually perform better than decision trees (Prasad et al., 2022; Etemad et al., 2021). Random forests may randomly select a certain proportion of features to be included in each model (Belgiu and Dragut, 2016). When the scale of the decision tree is too large, it may lead to over-fitting. Therefore, this study appropriately limited the depth and number of nodes of the tree during the parameter adjustment process. Random forest is suitable for most types of datasets, but in this study, no matter how the parameters are adjusted, random forest still shows some over-fitting tendency. Neural networks are more suitable for large and complex datasets. In the analysis of the confusion matrix, RF, SVM and MLP all showed low specificity (<0.4) at the first experiment, showing a strong preference to the tooth extraction decision. This may be attributed to the imbalance of the classification variable of the dataset. Among the consecutive patients in our hospital, the number of extraction treatment is larger than non-extraction, resulting in more positive cases in the testing set. Therefore, we adopted threshold moving method to select an optimal classification threshold (Collell et al., 2018). As a result, the specificity improved in SVM. Still, in RF and MLP models, adding the judgment of the extraction decision may improve the accuracy, resulting in decision bias. However, the decision tree shows better discernment of non-extraction cases. The more complex the model, the greater the possibility of over-fitting. Decision tree is more resistible to uneven category in this study. Decision tree is based on information entropy for classification, so it essentially looks at the correlation between features and categories. Even if there is little data in this category, as long as its correlation with the features is strong, it will not be misclassified. Random forest has a step of random sampling of data. If the category distribution is uneven, this class may not exist in some sub-sampling, which will affect the results.

There is occasionality in the generation of the model, and high accuracy in a single test does not mean stable performance of the model. In order to evaluate the generalization of the model, we conducted cross-validation. In cross-validation, each sample will appear once in the validation set, therefore, the model needs to have good generalization ability for all samples in the dataset in order to achieve a high average cross-validation accuracy (Jonathan et al., 2000; Karkkainen, 2014). Xie et al. (2010) used an artificial neural network model to predict the extraction plan and obtained an accuracy of 100% in the training set and 80% in the testing set, indicating over-fitting of the model. Köktürk et al. (2024) used multiple machine learning methods for tooth extraction decision-making, and the best algorithm was Gradient Boosted Trees, with an accuracy of 83.3%. We obtained an accuracy of 81.5% using MLP. Li et al. (2019) and Jung and Kim (2016) both used deep neural networks to train tooth extraction prediction models on samples of 156 and 302 patients, respectively, achieving an accuracy rate of over 93%, but their study did not specify the validation method they used. However, Köktürk et al. (2024), Etemad et al. (2021) and the current study also used the MLP method, but none of the accuracy exceeded 82%. They further divided the dataset into a correctly predicted group and an incorrectly predicted group according to the classification results of this preliminary model, and used the same method to calculate the accuracy, which was improved in both groups (Etemad et al., 2021). However, such a division has no clinical implication, and therefore impossible to find the corresponding external testing set, so it cannot be said that this result can be generalized. Interestingly, Etemad et al. (2021) found that the accuracy of the model obtained by using 117 variables did not improve compared with that using 22 variables. Our study only used 9 variables and obtained an accuracy exceeding their study. Suhail et al. (2020), unlike previous studies, significantly reduced the number of diagnostic features (nineteen) and demonstrated that finite feature sets and machine learning algorithms can accurately predict the extraction process. The ensemble of simpler models outperforms more complex models, like neural network.

There is a huge difference in the tooth extraction rate between Eastern and Western people. The tooth extraction and correction rate of Western people is around 10%–30%, much lower than that of East Asian people (Jackson et al., 2017). Economic, psychological, physiological and anatomical conditions may cause various extraction rate and complex influencing factors among different countries (Del Real et al., 2022), resulting in distinct difficulty of model training. Del Real et al. (2022) obtained an accuracy of more than 90% in one of their models in a Chilean population, but in their sample, cases of skeletal class I accounted for 65%, and skeletal class III only 6%, and the average angle cases were also close to 50%, the proportion of normal facial type was significantly higher than in our study.

Tooth extraction plan can be affected by the patient’s own factors. Some patients strongly resist tooth extraction, which may lead to the actual number of tooth extraction classifications in the final included data being less than the number of tooth extraction classifications that should be, which may be one of the reasons for the misclassification of the prediction model. Therefore, in this study, we recorded the experts’ preferred plans rather than the actual plans in the final dataset to predict the tooth extraction plan.

The interpretability of the tree models significantly outperforms other models like deep neural networks. Since the decision tree model worked better in this study, and a tendency of over-fitting occurred in the random forest, the results of the decision tree should be referred to in the feature contribution analysis. The most important indicators of extraction treatment were crowding, lateral facial profile, and lower incisor inclination. These features can all reflect the tooth-bone volume discrepancy. On the other hand, indicators such as Angle’s classification and overjet may reflect the sagittal discrepancy of maxilla and mandible and does not influence orthodontic extraction decisions from our study. Li et al. (2019) found out that upper and lower crowding, and U1-NA°, are the three most important features. Su et al. (2022) studied the tooth extraction pattern of patients with skeletal Class II average angle and deep overjet and showed that deep overjet and distal molar relationship are the main reasons for tooth extraction in Class II patients. Guo et al. (2014) founded that lower anterior crowding, molar relationship, and growth pattern were the three most relevant influential factors to the extraction decisions for Angle’s class II division 1 malocclusions. The above three studies all found that crowding is the most important influencing factor in determining tooth extraction decisions, but other factors are slightly different from this study. However, Bishara et al. (1995) pointed out that tooth size-arch length discrepancy and lip protrusion are the main reference factors for tooth extraction correction, which is highly consistent with the results of our study.

It should be noted that the purpose of introducing machine learning to assist in the classification of orthodontic tooth extraction plans is not to pursue a wiser judgment than that of orthodontists in critical cases, because whether it is expert judgment or machine learning classifier, the plan formulation of critical cases needs to weigh non-orthodontic indicators such as periodontal conditions, patient’s own will, and public aesthetics, rather than just the accuracy as a measurement standard. Machine learning methods should be able to better summarize the logic and ideas of experts, explain the causes and nature of malocclusion, and help doctors formulate more comprehensive treatment plans. Any mathematical principle that affects decision-making cannot be isolated from clinical examination and communication.

5 Conclusion

The decision tree algorithm in machine learning outperformed other machine learning models in predicting orthodontic extraction plans, with an average accuracy of 86%. Crowding is the most important factor for experts to decide on extraction treatment, followed by lateral facial profile and lower incisor inclination, indicating that tooth extraction is an important treatment method for tooth-bone volume discrepancy. Clinically, lack of space, protrusive anterior lips and convex facial profile indicates the need for tooth extraction. Machine learning should not replace but help doctors formulate more comprehensive treatment plans.

Data availability statement

The raw data supporting the conclusions of this article will be made available by the authors, without undue reservation.

Ethics statement

The studies involving humans were approved by the Institutional Review Board of Shanghai Stomatological Hospital. The studies were conducted in accordance with the local legislation and institutional requirements. Written informed consent for participation was not required from the participants or the participants’ legal guardians/next of kin in accordance with the national legislation and institutional requirements.

Author contributions

JH: Conceptualization, Data curation, Formal Analysis, Funding acquisition, Investigation, Project administration, Resources, Software, Supervision, Validation, Writing–original draft, Writing–review and editing. I-TC: Data curation, Formal Analysis, Investigation, Writing–review and editing. ZW: Data curation, Formal Analysis, Investigation, Writing–review and editing. XD: Data curation, Formal Analysis, Investigation, Writing–review and editing. YJ: Data curation, Formal Analysis, Investigation, Writing–review and editing. CY: Conceptualization, Formal Analysis, Funding acquisition, Supervision, Writing–review and editing. YP: Conceptualization, Data curation, Formal Analysis, Investigation, Methodology, Project administration, Resources, Software, Supervision, Validation, Writing–original draft, Writing–review and editing.

Funding

The author(s) declare that financial support was received for the research, authorship, and/or publication of this article. The research was supported by National Natural Science Foundation of China (82001104).

Acknowledgments

The authors would like to thank all the doctors of Department of Orthodontics, Shanghai Stomatological Hospital who have provided clinical materials and Yibin Liu, vision engineer of Changzhou Changchuan Technology Co., Ltd. for technical support.

Conflict of interest

The authors declare that the research was conducted in the absence of any commercial or financial relationships that could be construed as a potential conflict of interest.

The reviewer CO declared a shared affiliation with the authors JH and I-TC to the handling editor at the time of review.

Publisher’s note

All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article, or claim that may be made by its manufacturer, is not guaranteed or endorsed by the publisher.

References

Belgiu, M., and Dragut, L. (2016). Random forest in remote sensing: a review of applications and future directions. Isprs J. Photogramm. 114, 24–31. doi:10.1016/j.isprsjprs.2016.01.011

CrossRef Full Text | Google Scholar

Bishara, S. E., Cummins, D. M., and Jakobsen, J. R. (1995). The morphologic basis for the extraction decision in Class II, division 1 malocclusions: a comparative study. Am. J. Orthod. Dentofac. Orthop. 107, 129–135. doi:10.1016/s0889-5406(95)70127-3

PubMed Abstract | CrossRef Full Text | Google Scholar

Boley, J. C., Mark, J. A., Sachdeva, R. C., and Buschang, P. H. (2003). Long-term stability of Class I premolar extraction treatment. Am. J. Orthod. Dentofac. Orthop. 124, 277–287. doi:10.1016/s0889-5406(03)00448-7

PubMed Abstract | CrossRef Full Text | Google Scholar

Chang, C. A., Fields, H. W., Beck, F. M., Springer, N. C., Firestone, A. R., Rosenstiel, S., et al. (2011). Smile esthetics from patients’ perspectives for faces of varying attractiveness. Am. J. Orthod. Dentofac. Orthop. 140, e171–e180. doi:10.1016/j.ajodo.2011.03.022

PubMed Abstract | CrossRef Full Text | Google Scholar

Collell, G., Prelec, D., and Patil, K. R. (2018). A simple plug-in bagging ensemble based on threshold-moving for classifying binary and multiclass imbalanced data. Neurocomputing 275, 330–340. doi:10.1016/j.neucom.2017.08.035

PubMed Abstract | CrossRef Full Text | Google Scholar

Del Real, A., Del Real, O., Sardina, S., and Oyonarte, R. (2022). Use of automated artificial intelligence to predict the need for orthodontic extractions. Korean J. Orthod. 52, 102–111. doi:10.4041/kjod.2022.52.2.102

PubMed Abstract | CrossRef Full Text | Google Scholar

Etemad, L., Wu, T. H., Heiner, P., Liu, J., Lee, S., Chao, W. L., et al. (2021). Machine learning from clinical data sets of a contemporary decision for orthodontic tooth extraction. Orthod. Craniofac Res. 24 (Suppl. 2), 193–200. doi:10.1111/ocr.12502

PubMed Abstract | CrossRef Full Text | Google Scholar

Guo, Y., Han, X., Xu, H., Ai, D., Zeng, H., and Bai, D. (2014). Morphological characteristics influencing the orthodontic extraction strategies for Angle’s class II division 1 malocclusions. Prog. Orthod. 15, 44. doi:10.1186/s40510-014-0044-y

PubMed Abstract | CrossRef Full Text | Google Scholar

Hasija, N., Bala, M., and Goyal, V. (2014). Estimation of tooth size discrepancies among different malocclusion groups. Int. J. Clin. Pediatr. Dent. 7, 82–85. doi:10.5005/jp-journals-10005-1242

PubMed Abstract | CrossRef Full Text | Google Scholar

Jackson, T. H., Guez, C., Lin, F. C., Proffit, W. R., and Ko, C. C. (2017). Extraction frequencies at a university orthodontic clinic in the 21st century: demographic and diagnostic factors affecting the likelihood of extraction. Am. J. Orthod. Dentofac. Orthop. 151, 456–462. doi:10.1016/j.ajodo.2016.08.021

PubMed Abstract | CrossRef Full Text | Google Scholar

Janson, G., Araki, J., Estelita, S., and Camardella, L. T. (2014). Stability of class II subdivision malocclusion treatment with 3 and 4 premolar extractions. Prog. Orthod. 15, 67. doi:10.1186/s40510-014-0067-4

PubMed Abstract | CrossRef Full Text | Google Scholar

Janson, G., Valarelli, F. P., Henriques, J. F., de Freitas, M. R., and Cançado, R. H. (2003). Stability of anterior open bite nonextraction treatment in the permanent dentition. Am. J. Orthod. Dentofac. Orthop. 124, 265–276. doi:10.1016/s0889-5406(03)00449-9

CrossRef Full Text | Google Scholar

Jonathan, P., Krzanowski, W. J., and Mccarthy, W. V. (2000). On the use of cross-validation to assess performance in multivariate prediction. Statistics and Comput. 10, 209–229. doi:10.1023/a:1008987426876

CrossRef Full Text | Google Scholar

Jung, S. K., and Kim, T. W. (2016). New approach for the diagnosis of extractions with neural network machine learning. Am. J. Orthod. Dentofac. Orthop. 149, 127–133. doi:10.1016/j.ajodo.2015.07.030

PubMed Abstract | CrossRef Full Text | Google Scholar

Karkkainen, T. (2014). On cross-validation for MLP model evaluation. Int. Workshop Stat. Tech. Pattern Recognit., 291–300. doi:10.1007/978-3-662-44415-3_30

CrossRef Full Text | Google Scholar

Ker, A. J., Chan, R., Fields, H. W., Beck, M., and Rosenstiel, S. (2008). Esthetics and smile characteristics from the layperson's perspective: a computer-based survey study. J. Am. Dent. Assoc. 139, 1318–1327. doi:10.14219/jada.archive.2008.0043

PubMed Abstract | CrossRef Full Text | Google Scholar

Khanagar, S. B., Al-Ehaideb, A., Vishwanathaiah, S., Maganur, P. C., Patil, S., Naik, S., et al. (2021). Scope and performance of artificial intelligence technology in orthodontic diagnosis, treatment planning, and clinical decision-making - a systematic review. J. Dent. Sci. 16, 482–492. doi:10.1016/j.jds.2020.05.022

PubMed Abstract | CrossRef Full Text | Google Scholar

Köktürk, B., Pamukçu, H., and Gözüaçik, Ö. (2024). Evaluation of different machine learning algorithms for extraction decision in orthodontic treatment. Orthod. Craniofac Res. doi:10.1111/ocr.12811

CrossRef Full Text | Google Scholar

Konstantonis, D., Anthopoulou, C., and Makou, M. (2013). Extraction decision and identification of treatment predictors in Class I malocclusions. Prog. Orthod. 14, 47. doi:10.1186/2196-1042-14-47

PubMed Abstract | CrossRef Full Text | Google Scholar

Leavitt, L., Volovic, J., Steinhauer, L., Mason, T., Eckert, G., Dean, J. A., et al. (2023). Can we predict orthodontic extraction patterns by using machine learning?. Orthod. Craniofac Res. 26, 552–559. doi:10.1111/ocr.12641

PubMed Abstract | CrossRef Full Text | Google Scholar

Li, P., Kong, D., Tang, T., Su, D., Yang, P., Wang, H., et al. (2019). Orthodontic treatment planning based on artificial neural networks. Sci. Rep. 9, 2037. doi:10.1038/s41598-018-38439-w

PubMed Abstract | CrossRef Full Text | Google Scholar

Liu, J., Chen, Y., Li, S., Zhao, Z., and Wu, Z. (2021). Machine learning in orthodontics: challenges and perspectives. Adv. Clin. Exp. Med. 30, 1065–1074. doi:10.17219/acem/138702

PubMed Abstract | CrossRef Full Text | Google Scholar

Lombardo, G., Vena, F., Negri, P., Pagano, S., Barilotti, C., Paglia, L., et al. (2020). Worldwide prevalence of malocclusion in the different stages of dentition: a systematic review and meta-analysis. Eur. J. Paediatr. Dent. 21, 115–122. doi:10.23804/ejpd.2020.21.02.05

PubMed Abstract | CrossRef Full Text | Google Scholar

Maués, C. P. R., Nascimento, R. R. D., and Vilella, O. D. V. (2015). Severe root resorption resulting from orthodontic treatment: prevalence and risk factors. Dent. Press J. Orthod. 20, 52–58. doi:10.1590/2176-9451.20.1.052-058.oar

CrossRef Full Text | Google Scholar

Prasad, J., Mallikarjunaiah, D. R., Shetty, A., Gandedkar, N., Chikkamuniswamy, A. B., and Shivashankar, P. C. (2022). Machine learning predictive model as clinical decision support system in orthodontic treatment planning. Dent. J. (Basel) 11, 1. doi:10.3390/dj11010001

PubMed Abstract | CrossRef Full Text | Google Scholar

Proffit, W. R. (1994). Forty-year review of extraction frequencies at a university orthodontic clinic. Angle Orthod. 64, 407–414. doi:10.1043/0003-3219(1994)064<0407:FROEFA>2.0.CO;2

PubMed Abstract | CrossRef Full Text | Google Scholar

Su, H., Han, B., Wang, X., and Xu, T. (2022). Analysis of craniofacial features of patients with skeletal Class II average angle deepoverjet in three different tooth extraction modes. Chin. J. Orthod. 29, 24–28. doi:10.3760/cma.j.cn115797-20211112-22105

CrossRef Full Text | Google Scholar

Suhail, Y., Upadhyay, M., Chhibber, A., and Kshitiz, (2020). Machine learning for the diagnosis of orthodontic extractions: a computational analysis using ensemble learning. Bioengineering 7, 55. doi:10.3390/bioengineering7020055

PubMed Abstract | CrossRef Full Text | Google Scholar

Xie, X., Wang, L., and Wang, A. (2010). Artificial neural network modeling for deciding if extractions are necessary prior to orthodontic treatment. Angle Orthod. 80, 262–266. doi:10.2319/111608-588.1

PubMed Abstract | CrossRef Full Text | Google Scholar

Xu, T., and Huang, W. (2002). Interclinician agreement on the decision of extraction or non-extraction orthodontic treatment. Chin. J. Orthod. 9, 14–16. doi:10.3760/cma.j.issn.1674-5760.2002.02.004

CrossRef Full Text | Google Scholar

Xu, Y., and Goodacre, R. (2018). On splitting training and validation set: a comparative study of cross-validation, bootstrap and systematic sampling for estimating the generalization performance of supervised learning. J. Anal. Test. 2, 249–262. doi:10.1007/s41664-018-0068-2

PubMed Abstract | CrossRef Full Text | Google Scholar

Yin, L., Jiang, M., Chen, W., Smales, R. J., Wang, Q., and Tang, L. (2014). Differences in facial profile and dental esthetic perceptions between young adults and orthodontists. Am. J. Orthod. Dentofac. Orthop. 145, 750–756. doi:10.1016/j.ajodo.2014.01.021

PubMed Abstract | CrossRef Full Text | Google Scholar

Keywords: orthodontic treatment, tooth extraction decision, decision tree, machine learning, cross validation

Citation: Huang J, Chan I-T, Wang Z, Ding X, Jin Y, Yang C and Pan Y (2024) Evaluation of four machine learning methods in predicting orthodontic extraction decision from clinical examination data and analysis of feature contribution. Front. Bioeng. Biotechnol. 12:1483230. doi: 10.3389/fbioe.2024.1483230

Received: 19 August 2024; Accepted: 04 October 2024;
Published: 14 October 2024.

Edited by:

Bolei Cai, The Fourth Military Medical University, China

Reviewed by:

Chun Ouyang, Fudan University, China
Nishant Kumar, K L University, India
Meritxell Sanchez Molins, University of Barcelona, Spain
Nuno Gustavo Correia d’Oliveira, University of Barcelona, Spain, in collaboration with reviewer MS

Copyright © 2024 Huang, Chan, Wang, Ding, Jin, Yang and Pan. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) and the copyright owner(s) are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

*Correspondence: Congchong Yang, eWFuZ2NvbmdjaG9uZzE5ODdAMTI2LmNvbQ==; Yichen Pan, NjAyNzUyODc3QHFxLmNvbQ==

Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.