Evolutionary algorithms and decision trees for predicting poor outcome after endovascular treatment for acute ischemic stroke

N. Kappelhof, L. A. Ramos*, M. Kappelhof, H. J. A. van Os, Vicky Chalos, K. R. van Kranendonk, N. D. Kruyt, Y. B. W. E. M. Roos, W. H. van Zwam, I. C. van der Schaaf, M. A. A. van Walderveen, M. J. H. Wermer, R. J. van Oostenbrugge, Hester Lingsma, Diederik Dippel, C. B. L. M. Majoie, H. A. Marquering

*Corresponding author for this work

Research output: Contribution to journalArticleAcademicpeer-review

3 Citations (Web of Science)

Abstract

Despite the large overall beneficial effects of endovascular treatment in patients with acute ischemic stroke, severe disability or death still occurs in almost one-third of patients. These patients, who might not benefit from treatment, have been previously identified with traditional logistic regression models, which may oversimplify relations between characteristics and outcome, or machine learning techniques, which may be difficult to interpret. We developed and evaluated a novel evolutionary algorithm for fuzzy decision trees to accurately identify patients with poor outcome after endovascular treatment, which was defined as having a modified Rankin Scale score (mRS) higher or equal to 5. The created decision trees have the benefit of being comprehensible, easily interpretable models, making its predictions easy to explain to patients and practitioners. Insights in the reason for the predicted outcome can encourage acceptance and adaptation in practice and help manage expectations after treatment. We compared our proposed method to CART, the benchmark decision tree algorithm, on classification accuracy and interpretability. The fuzzy decision tree significantly outperformed CART: using 5-fold cross-validation with on average 1090 patients in the training set and 273 patients in the test set, the fuzzy decision tree misclassified on average 77 (standard deviation of 7) patients compared to 83 (+/- 7) using CART. The mean number of nodes (decision and leaf nodes) in the fuzzy decision tree was 11 (+/- 2) compared to 26 (+/- 1) for CART decision trees. With an average accuracy of 72% and much fewer nodes than CART, the developed evolutionary algorithm for fuzzy decision trees might be used to gain insights into the predictive value of patient characteristics and can contribute to the development of more accurate medical outcome prediction methods with improved clarity for practitioners and patients.

Original languageEnglish
Article number104414
Number of pages11
JournalComputers in Biology and Medicine
Volume133
DOIs
Publication statusPublished - Jun 2021

Keywords

  • Acute ischemic stroke
  • Machine learning
  • Prognostics
  • Decision trees
  • Evolutionary algorithms
  • Fuzzy
  • Endovascular treatment

Cite this