Development and validation of an early predictive model for Hemiplegic Shoulder Pain: A Comparative Study of Logistic Regression, Support Vector Machine, and Random Forest

Qiang Wu¹Fang Zhang¹Yuchang Fei²Zhenfen Sima¹Shanshan Gong³Qifeng Tong¹Qingchuan Jiao¹Hao Wu¹Jianqiu Gong^1*

• ¹Department of Rehabilitation Medicine, Shaoxing People's Hospital, Shaoxing, China
• ²Department of Integrated Chinese and Western Medicine, Jiashan First People's Hospital, Jiaxing, China
• ³Department of Gastroenterology, Third Affiliated Hospital of Zhejiang Chinese Medical University, Hangzhou, Zhejiang Province, China

Objective In this study, we aim to identify the predictive variables for hemiplegic shoulder pain(HSP) through machine learning algorithms, select the optimal model and predict the occurrence of HSP. Methods Data of 332 stroke patients admitted to a tertiary hospital in Zhejiang Province from January 2022 to January 2023 were collected. After screening predictive variables by LASSO regression, three predictive models selected using the LazyPredict package, namely Logistic Regression(LR), Support Vector Machine(SVM) and Random Forest(RF), were established respectively. The performance parameters (accuracy, precision, recall, F1 score) of the models were calculated, the receiver operating characteristic curve(ROC) and the Decision Curve Analysis(DCA) were plotted to compare the performance of the three models. An explainability analysis (SHAP) was conducted on the optimal model. Results The RF model performed the best, with accuracy: 0.90, precision: 0.89, recall: 0.88, F1 score: 0.86, AUC-ROC: 0.94, and the range of the threshold probability in DCA: 7% -99%. Based on the SHAP analysis of the explainability of the RF model, the contribution degrees of the early HSP predictive variables from high to low are as follows: multiple injuries, shoulder joint flexion (p), biceps tendon effusion, sensory disorder, supraspinatus tendinopathy, subluxation, diabetes, and age.The RF prediction model has a good predictive effect on HSP and has good clinical explainability. It can provide objective references for the early warning and stratified management of HSP.