Comparative Analysis of the Ability of Machine Learning Models in Predicting In-hospital Postoperative Outcomes After Total Hip Arthroplasty.

Abstract

Machine learning (ML) methods have shown promise in a wide range of applications including the development of patient-specific predictive models before surgical interventions. The purpose of this study was to develop, test, and compare four distinct ML models to predict postoperative parameters after primary total hip arthroplasty. Data from the Nationwide Inpatient Sample were used to identify patients undergoing total hip arthroplasty from 2016 to 2017. Linear support vector machine (LSVM), random forest (RF), neural network (NN), and extreme gradient boost trees (XGBoost) predictive of mortality, length of stay, and discharge disposition were developed and validated using 15 predictive patient-specific and hospital-specific factors. Area under the curve of the receiver operating characteristic (AUCROC) curve and accuracy were used as validity metrics, and the strongest predictive variables under each model were assessed. A total of 177,442 patients were included in this analysis. For mortality, the XGBoost, NN, and LSVM models all had excellent responsiveness during validation while RF had fair responsiveness. LSVM had the highest responsiveness with an AUCROC of 0.973 during validation. For the length of stay, the LSVM and NN models had fair responsiveness while the XGBoost and random forest models had poor responsiveness. LSVM had the highest responsiveness with an AUCROC of 0.744 during validation. For the discharge disposition outcome, LSVM had good responsiveness while the XGBoost, NN, and RF models all had fair responsiveness. LSVM had the highest responsiveness with an AUCROC of 0.801. The ML methods tested demonstrated a range of poor-to-excellent responsiveness and accuracy in the prediction of the assessed metrics, with LSVM being the best performer. Such models should be further developed, with eventual integration into clinical practice to inform patient discussions and management decision making, with the potential for integration into tiered bundled payment models.