Machine Learning-Based Prediction of Subsequent Vascular Events After 6 Months in Chinese Patients with Minor Ischemic Stroke

Systematic Review of Machine Learning Applied to the Secondary Prevention of Ischemic Stroke

Ischemic stroke is a serious disease posing significant threats to human health and life, with the highest absolute and relative risks of a poor prognosis following the first occurrence, and more than 90% of strokes are attributable to modifiable risk factors. Currently, machine learning (ML) is widely used for the prediction of ischemic stroke outcomes. By identifying risk factors, predicting the risk of poor prognosis and thus developing personalized treatment plans, it effectively reduces the probability of poor prognosis, leading to more effective secondary prevention. This review includes 41 studies since 2018 that used ML algorithms to build prognostic prediction models for ischemic stroke, transient ischemic attack (TIA), and acute ischemic stroke (AIS). We analyzed in detail the risk factors used in these studies, the sources and processing methods of the required data, the model building and validation, and their application in different prediction time windows. The results indicate that among the included studies, the top five risk factors in terms of frequency were cardiovascular diseases, age, sex, national institutes of health stroke scale (NIHSS) score, and diabetes. Furthermore, 64% of the studies used single-center data, 65% of studies using imbalanced data did not perform data balancing, 88% of the studies did not utilize external validation datasets for model validation, and 72% of the studies did not provide explanations for their models. Addressing these issues is crucial for enhancing the credibility and effectiveness of the research, consequently improving the development and implementation of secondary prevention measures.

Traumatic Brain Injury Rehabilitation Outcome Prediction Using Machine Learning Methods

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.

Setting

Acute inpatient rehabilitation.

Participants

1946 patients aged 14 years or older, who sustained a severe, moderate, or complicated mild TBI, and were admitted to 1 of 9 US inpatient rehabilitation sites between 2008 and 2011 (N=1946).

Main Outcome Measures

Rehabilitation length of stay, discharge to home, FIM cognitive and FIM motor at discharge and at 9-months post discharge.

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.