Machine learning techniques for independent gait recovery prediction in acute anterior circulation ischemic stroke

This study aimed to develop and validate a machine learning-based predictive model for gait recovery in patients with acute anterior circulation ischemic stroke.

Between May and November 2023, 237 patients with acute anterior circulation ischemic stroke were enrolled. Patients were randomly divided into training and validation sets at a 7:3 ratio. Thirty-one medical characteristics were collected, and the Least Absolute Shrinkage and Selection Operator (LASSO) regression was applied to screen predictor variables. Predictive models were developed using the Random Survival Forest (RSF) and COX regression methods. The optimal model was identified based on C-index values. The SHapley Additive exPlanations (SHAP) method was employed to interpret the RSF model globally and locally.

Ten predictors were identified through LASSO regression, including age, gender, periventricular white matter hyperintensities (PVWMH), Montreal Cognitive Assessment (MoCA), National Institutes of Health Stroke Scale (NIHSS), enlarged perivascular spaces in basal ganglia (BG-EPVS), lacunes, parietal infarction, basal ganglia infarction, and Timed Up & Go (TUG) test score. The C-index values of the COX regression and RSF models were 0.741 and 0.761 in the training set and 0.705 and 0.725 in the validation set, respectively. SHAP analysis of the RSF model identified BG-EPVS, TUG, MoCA, age, and PVWMH as the top five most influential predictors of gait recovery.

The RSF model demonstrated superior performance to the COX regression model in predicting gait recovery, offering a reliable tool for clinical decision-making regarding stroke patients' prognoses.