Scientific article
OA Policy
English

Machine learning for early dynamic prediction of functional outcome after stroke

Published inCommunications medicine, vol. 4, no. 1, 232
Publication date2024-11-13
First online date2024-11-13
Abstract

Background: Prediction of outcome after stroke is critical for treatment planning and resource allocation but is complicated by fluctuations during the first days after onset. We propose a machine learning model that can provide hourly predictions based on the integration of continuous variables acquired within 72 h of hospital admission.

Methods: We analyzed 2492 admissions for ischemic stroke in the Geneva University Hospital from 01.01.2018 to 31.12.2021, amounting to 2'131'752 unique data points. We developed a transformer model that continuously included clinical, physiological, imaging, and biological data recorded within 72 h of admission. This model was trained to generate hourly predictions of mortality and morbidity. Shapley additive explanations were used to identify the most relevant predictors to explain outcomes for each patient. The MIMIC-III database was used for external validation.

Results: Our transformer model predicts mortality, with an area under the receiver operating characteristic curve of 0.830 (95% CI 0.763-0.885) on admission, reaching 0.893 (95% CI 0.839-0.933) 72 h later for a 3-month outcome. Validated in an independent cohort, it outperforms all static models. Based on their mean explanatory weights, the top predictors included continuous clinical evaluation, baseline patient characteristics, timing from admission to acute treatment, and markers of inflammation and organ dysfunction.

Conclusions: The performance of our transformer model demonstrates the potential of machine learning models integrating clinical, physiological, imaging, and biological variables over time after stroke. The clinical applicability of our model is further strengthened by access to hourly updated predictions along with accompanying explanations.

Citation (ISO format)
KLUG, Julian et al. Machine learning for early dynamic prediction of functional outcome after stroke. In: Communications medicine, 2024, vol. 4, n° 1, p. 232. doi: 10.1038/s43856-024-00666-w
Main files (1)
Article (Published version)
Secondary files (6)
Appendix
accessLevelPublic
Appendix
accessLevelPublic
Appendix
accessLevelPublic
Show more
Identifiers
Additional URL for this publicationhttps://www.nature.com/articles/s43856-024-00666-w
Journal ISSN2730-664X
58views
342downloads

Technical informations

Creation28/06/2025 04:20:21
First validation07/07/2025 09:23:12
Update07/07/2025 09:23:12
Status update07/07/2025 09:23:12
Last indexation07/07/2025 09:23:13
All rights reserved by Archive ouverte UNIGE and the University of GenevaunigeBlack