Predicting futile recanalization, malignant cerebral edema, and cerebral herniation using intelligible ensemble machine learning following mechanical thrombectomy for acute ischemic stroke

Abstract

PurposeTo establish an ensemble machine learning (ML) model for predicting the risk of futile recanalization, malignant cerebral edema (MCE), and cerebral herniation (CH) in patients with acute ischemic stroke (AIS) who underwent mechanical thrombectomy (MT) and recanalization.MethodsThis prospective study included 110 patients with premorbid mRS ≤ 2 who met the inclusion criteria. Futile recanalization was defined as a 90-day modified Rankin Scale score >2. Clinical and imaging data were used to construct five ML models that were fused into a logistic regression algorithm using the stacking method (LR-Stacking). We added the Shapley Additive Explanation method to display crucial factors and explain the decision process of models for each patient. Prediction performances were compared using area under the receiver operating characteristic curve (AUC), F1-score, and decision curve analysis (DCA).ResultsA total of 61 patients (55.5%) experienced futile recanalization, and 34 (30.9%) and 22 (20.0%) patients developed MCE and CH, respectively. In test set, the AUCs for the LR-Stacking model were 0.949, 0.885, and 0.904 for the three outcomes mentioned above. The F1-scores were 0.882, 0.895, and 0.909, respectively. The DCA showed that the LR-Stacking model provided more net benefits for predicting MCE and CH. The most important factors were the hypodensity volume and proportion in the corresponding vascular supply area.ConclusionUsing the ensemble ML model to analyze the clinical and imaging data of AIS patients with successful recanalization at admission and within 24 h after MT allowed for accurately predicting the risks of futile recanalization, MCE, and CH.