Skip to main content

ORIGINAL RESEARCH article

Front. Neurol.
Sec. Stroke
Volume 15 - 2024 | doi: 10.3389/fneur.2024.1446250

Development and Validation of An Explainable Machine Learning Prediction Model of Hemorrhagic Transformation after Intravenous Thrombolysis in Stroke

Provisionally accepted
  • 1 First Affiliated Hospital, Dalian Medical University, Dalian, China
  • 2 Dalian Institute of Chemical Physics, Chinese Academy of Sciences (CAS), Dalian, Liaoning Province, China

The final, formatted version of the article will be published soon.

    Objective: To develop and validate an explainable machine learning (ML) model predicting the risk of hemorrhagic transformation (HT) after intravenous thrombolysis.We retrospectively enrolled patients who received intravenous tissue plasminogen activator (IV-tPA) thrombolysis within 4.5 hours after symptom onset to form the original modeling cohort. HT was defined as any hemorrhage on head CT scan completed within 48 hours after IV-tPA administration. We utilized the Random Forest (RF), Multilayer Perceptron (MLP), Adaptive Boosting (AdaBoost), and Gaussian Naive Bayes (GauNB) algorithms to develop ML-HT models. The models' predictive performance was evaluated using confusion matrix (including accuracy, precision, recall, and F1 score), and discriminative analysis (area under the receiver-operatingcharacteristic curve, ROC-AUC) in the original cohort, followed by validation in an independent external cohort. The models' explainability was assessed using SHapley Additive exPlanations (SHAP) global feature plot, SHAP summary plot, and partial dependence plot.Results: A total of 1007 patients were included in the original modeling cohort, with an HT incidence of 8.94%. The RF-based ML-HT model showed metrics of 0.874 (accuracy), 0.972 (precision), 0.890 (recall), 0.929 (F1 score); with ROC-AUC of 0.7847 in the original cohort and 0.7119 in the external validation cohort.

    Keywords: Acute ischemic stroke, intravenous thrombolysis, Hemorrhagic transformation, machine learning, Explainability

    Received: 17 Jun 2024; Accepted: 20 Dec 2024.

    Copyright: © 2024 Lin, Li, Luo and Han. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) or licensor are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

    * Correspondence: Jie Han, First Affiliated Hospital, Dalian Medical University, Dalian, China

    Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.