Traumatic Brain Injury Rehabilitation Outcome Prediction Using Machine Learning Methods

Model System:

Reference Type:

Accession No.:

Journal:

Year, Volume, Issue, Page(s):

Publication Website:

Abstract:

Objective:

To investigate the performance of machine learning (ML) methods for predicting outcomes from inpatient rehabilitation for subjects with TBI using a dataset with a large number of predictor variables. Our second objective was to identify top predictive features selected by the ML models for each outcome and to validate the interpretability of the models.

Design:

Secondary analysis using computational modeling of relationships between patients, injury and treatment activities and 6 outcomes, applied to the large multi-site, prospective, longitudinal observational dataset collected during the traumatic brain injury inpatient rehabilitation study.

Results:

Advanced ML models, specifically gradient boosting tree model, performed consistently better than all other models, including classical linear regression models. Top ranked predictive features were identified for each of the 6 outcome variables. Level of effort, days to rehabilitation admission, age at rehabilitation admission, and advanced mobility activities were the most frequently top ranked predictive features. The highest-ranking predictive feature differed across the specific outcome variable.

Conclusions:

Identifying patient, injury, and rehabilitation treatment variables that are predictive of better outcomes will contribute to cost-effective care delivery and guide evidence-based clinical practice. ML methods can contribute to these efforts.

Author(s):

Participating Centers: