A Machine Learning Model for Prediction of Amputation in Diabetics

Abstract

Background: Diabetic foot ulcer (DFU) and the resulting lower extremity amputation are associated with a poor survival prognosis. The objective of this study is to generate a model for predicting the probability of major amputation in hospitalized patients with DFU. Methods: The National Inpatient Sample (NIS) database from 2008 to 2014 was used to select patients with DFU, who were then further divided by major amputation status. International Classification of Diseases, Ninth Edition, Clinical Modification ( ICD-9-CM) and Agency for Healthcare Research and Quality (AHRQ) comorbidity codes were used to compare patient characteristics. For the descriptive statistics, the Student t test, the χ2 test, and the Spearman correlation were utilized. The five most predictive variables were identified. A decision tree model (CTREE) based on conditional inference framework algorithm and a random forest model were used to develop the algorithm. Results: A total of 326 853 inpatients with DFU were identified, and 5.9% underwent major amputation. The top five contributory variables (all with P < .001) were gangrene (odds ratio [OR] = 11.8, 95% confidence interval [CI] = 11.5-12.2), peripheral vascular disease (OR = 2.9, 95% CI = 2.8-3.0), weight loss (OR = 2.6, 95% CI = 2.5-2.8), systemic infection (OR = 2.5, 95% CI = 2.4-2.53), and osteomyelitis (OR = 1.7, 95% CI = 1.6-1.73). The model performance of the training data was 77.7% (76.1% sensitivity and 79.3% specificity) and of the testing data was 77.8% (76.2% sensitivity and 79.4% specificity). The model was further validated with boosting and random forest models which demonstrated similar performance and area under the curve (AUC) (0.84, 95% CI = 0.83-0.85). Conclusion: Utilizing machine learning methods, we have developed a clinical algorithm that predicts the risk of major lower extremity amputation for inpatients with diabetes with 77.8% accuracy.