Construction and Validation of Machine Learning Algorithms to Predict Chronic Post-Surgical Pain Among Patients Undergoing Total Knee Arthroplasty

Abstract

BackgroundChronic post-surgical pain (CPSP) is a common but undertreated condition with a high prevalence among patients undergoing total knee arthroplasty (TKA). An effective model for CPSP prediction has not been established yet. AimsTo construct and validate machine learning models for the early prediction of CPSP among patients undergoing TKA. DesignA prospective cohort study. Participants/SubjectsA total of 320 patients in the modeling group and 150 patients in the validation group were recruited from two independent hospitals between December 2021 and July 2022. They were followed up for 6 months to determine the outcomes of CPSP through telephone interviews. MethodsFour machine learning algorithms were developed through 10-fold cross-validation for five times. In the validation group, the discrimination and calibration of the machine learning algorithms were compared by the logistic regression model. The importance of the variables in the best model identified was ranked. ResultsThe incidence of CPSP in the modeling group was 25.3%, and that in the validation group was 27.6%. Compared with other models, the random forest model achieved the best performance with the highest C-statistic of 0.897 and the lowest Brier score of 0.119 in the validation group. The top three important factors for predicting CPSP were knee joint function, fear of movement, and pain at rest in the baseline. ConclusionsThe random forest model demonstrated good discrimination and calibration capacity for identifying patients undergoing TKA at high risk for CPSP. Clinical nurses would screen out high-risk patients for CPSP by using the risk factors identified in the random forest model, and efficiently distribute preventive strategy.