Machine learning predicts improvement of functional outcomes in spinal cord injury patients after inpatient rehabilitation

IntroductionSpinal cord injury (SCI) presents a significant burden to patients, families, and the healthcare system. The ability to accurately predict functional outcomes for SCI patients is essential for optimizing rehabilitation strategies, guiding patient and family decision making, and improving...

Full description

Saved in:
Bibliographic Details
Main Authors: Mohammad Rasoolinejad, Irene Say, Peter B. Wu, Xinran Liu, Yan Zhou, Nathan Zhang, Emily R. Rosario, Daniel C. Lu
Format: Article
Language:English
Published: Frontiers Media S.A. 2025-08-01
Series:Frontiers in Rehabilitation Sciences
Subjects:
Online Access:https://www.frontiersin.org/articles/10.3389/fresc.2025.1594753/full
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:IntroductionSpinal cord injury (SCI) presents a significant burden to patients, families, and the healthcare system. The ability to accurately predict functional outcomes for SCI patients is essential for optimizing rehabilitation strategies, guiding patient and family decision making, and improving patient care.MethodsWe conducted a retrospective analysis of 589 SCI patients admitted to a single acute rehabilitation facility and used the dataset to train advanced machine learning algorithms to predict patients' rehabilitation outcomes. The primary outcome was the Functional Independence Measure (FIM) score at discharge, reflecting the level of independence achieved by patients after comprehensive inpatient rehabilitation.ResultsTree-based algorithms, particularly Random Forest (RF) and XGBoost, significantly outperformed traditional statistical models and Generalized Linear Models (GLMs) in predicting discharge FIM scores. The RF model exhibited the highest predictive accuracy, with an R-squared value of 0.90 and a Mean Squared Error (MSE) of 0.29 on the training dataset, while achieving 0.52 R-squared and 1.37 MSE on the test dataset. The XGBoost model also demonstrated strong performance, with an R-squared value of 0.74 and an MSE of 0.75 on the training dataset, and 0.51 R-squared with 1.39 MSE on the test dataset. Our analysis identified key predictors of rehabilitation outcomes, including the initial FIM scores and specific demographic factors such as level of injury and prehospital living settings. The study also highlighted the superior ability of tree-based models to capture the complex, non-linear relationships between variables that impact recovery in SCI patients.DiscussionThis research underscores the potential of machine learning models to enhance the accuracy of outcome predictions in SCI rehabilitation. The findings support the integration of these advanced predictive tools in clinical settings to better guide decision making for patients and families, tailor rehabilitation plans, allocate resources efficiently, and ultimately improve patient outcomes.
ISSN:2673-6861