A machine learning-based model for “In-time” prediction of periprosthetic joint infection

Abstract

Background Previous criteria had limited value in early diagnosis of periprosthetic joint infection (PJI). Here, we constructed a novel machine learning (ML)-derived, “in-time” diagnostic system for PJI and proved its validity. Methods We filtered “in-time” diagnostic indicators reported in the literature based on our continuous retrospective cohort of PJI and aseptic prosthetic loosening patients. With the indicators, we developed a two-level ML model with six base learners including Elastic Net, Linear Support Vector Machine, Kernel Support Vector Machine, Extra Trees, Light Gradient Boosting Machine and Multilayer Perceptron), and one meta-learner, Ensemble Learning of Weighted Voting. The prediction performance of this model was compared with those of previous diagnostic criteria (International Consensus Meeting in 2018 (ICM 2018), etc.). Another prospective cohort was used for internal validation. Based on our ML model, a user-friendly web tool was developed for swift PJI diagnosis in clinical practice. Results A total of 254 patients (199 for development and 55 for validation cohort) were included in this study with 38.2% of them diagnosed as PJI. We included 21 widely accessible features including imaging indicators (X-ray and CT) in the model. The sensitivity and accuracy of our ML model were significantly higher than ICM 2018 in development cohort (90.6% vs. 76.1%, P = 0.032; 94.5% vs. 86.7%, P = 0.020), which was supported by internal validation cohort (84.2% vs. 78.6%; 94.6% vs. 81.8%). Conclusions Our novel ML-derived PJI “in-time” diagnostic system demonstrated significantly improved diagnostic potency for surgical decision-making compared with the commonly used criteria. Moreover, our web-based tool greatly assisted surgeons in distinguishing PJI patients comprehensively. Level of evidence Diagnostic Level III.