Systematic comparison of machine learning algorithms to develop and validate predictive models for periodontitis
Nasir Zeeshan Bashir, Zahid Rahman, Sam Li‐Sheng Chen
Abstract
AIM: The aim of this study was to compare the validity of different machine learning algorithms to develop and validate predictive models for periodontitis. MATERIALS AND METHODS: Using national survey data from Taiwan (n = 3453) and the United States (n = 3685), predictors of periodontitis were extracted from the datasets and pre-processed, and then 10 machine learning algorithms were trained to develop predictive models. The models were validated both internally (bootstrap sampling) and externally (alternative country's dataset). The algorithms were compared across six performance metrics ([i] area under the curve for the receiver operating characteristic [AUC], [ii] accuracy, [iii] sensitivity, [iv] specificity, [v] positive predictive value, and [vi] negative predictive value) and two methods of data pre-processing ([i] machine-learning-based feature selection and [ii] dimensionality reduction into principal components). RESULTS: Many algorithms showed extremely strong performance during internal validation (AUC > 0.95, accuracy > 95%). However, this was not replicated in external validation, where predictive performance of all algorithms dropped off drastically. Furthermore, predictive performance differed according to data pre-processing methodology and the cohort on which they were trained. CONCLUSIONS: Larger sample sizes and more complex predictors of periodontitis are required before machine learning can be leveraged to its full potential.