Machine Learning-Based Model for Prediction of Outcomes in Acute Stroke

Joon Nyung Heo, Jihoon G. Yoon, Hyungjong Park, Young Dae Kim, Hyo Suk Nam, Ji Hoe Heo

Research output: Contribution to journalArticle

15 Citations (Scopus)

Abstract

Background and Purpose - The prediction of long-term outcomes in ischemic stroke patients may be useful in treatment decisions. Machine learning techniques are being increasingly adapted for use in the medical field because of their high accuracy. This study investigated the applicability of machine learning techniques to predict long-term outcomes in ischemic stroke patients. Methods - This was a retrospective study using a prospective cohort that enrolled patients with acute ischemic stroke. Favorable outcome was defined as modified Rankin Scale score 0, 1, or 2 at 3 months. We developed 3 machine learning models (deep neural network, random forest, and logistic regression) and compared their predictability. To evaluate the accuracy of the machine learning models, we also compared them to the Acute Stroke Registry and Analysis of Lausanne (ASTRAL) score. Results - A total of 2604 patients were included in this study, and 2043 (78%) of them had favorable outcomes. The area under the curve for the deep neural network model was significantly higher than that of the ASTRAL score (0.888 versus 0.839; P<0.001), while the areas under the curves of the random forest (0.857; P=0.136) and logistic regression (0.849; P=0.413) models were not significantly higher than that of the ASTRAL score. Using only the 6 variables that are used for the ASTRAL score, the performance of the machine learning models did not significantly differ from that of the ASTRAL score. Conclusions - Machine learning algorithms, particularly the deep neural network, can improve the prediction of long-term outcomes in ischemic stroke patients.

Original languageEnglish
Pages (from-to)1263-1265
Number of pages3
JournalStroke
Volume50
Issue number5
DOIs
Publication statusPublished - 2019 May 1

Bibliographical note

Funding Information:
This study was supported by the Basic Science Research Program through the National Research Foundation of Korea (NRF) funded by the Ministry of Education (NRF-2018R1A2A3074996).

All Science Journal Classification (ASJC) codes

  • Clinical Neurology
  • Cardiology and Cardiovascular Medicine
  • Advanced and Specialised Nursing

Fingerprint Dive into the research topics of 'Machine Learning-Based Model for Prediction of Outcomes in Acute Stroke'. Together they form a unique fingerprint.

Cite this