Abstract:Background: The early prediction of cerebral edema changes in patients with spontaneous intracerebral hemorrhage (SICH) may facilitate earlier interventions and result in improved outcomes. This study aimed to develop and validate machine learning models to predict cerebral edema changes within 72 h, using readily available clinical parameters, and to identify relevant influencing factors. Methods: An observational study was conducted between April 2021 and October 2023 at the Quzhou Affiliated Hospital of Wenzhou Medical University. After preprocessing the data, the study population was randomly divided into training and internal validation cohorts in a 7:3 ratio (training: N = 150; validation: N = 65). The most relevant variables were selected using Support Vector Machine Recursive Feature Elimination (SVM-RFE) and Least Absolute Shrinkage and Selection Operator (LASSO) algorithms. The predictive performance of random forest (RF), GDBT, linear regression (LR), and XGBoost models was evaluated using the area under the receiver operating characteristic curve (AUROC), precision–recall curve (AUPRC), accuracy, F1-score, precision, recall, sensitivity, and specificity. Feature importance was calculated, and the SHapley Additive exPlanations (SHAP) and Local Interpretable Model-Agnostic Explanations (LIME) methods were employed to explain the top-performing model. Results: A total of 84 (39.1%) patients developed cerebral edema changes. In the validation cohort, GDBT outperformed LR and RF, achieving an AUC of 0.654 (95% CI: 0.611–0.699) compared to LR of 0.578 (95% CI, 0.535–0.623, DeLong: p = 0.197) and RF of 0.624 (95% CI, 0.588–0.687, DeLong: p = 0.236). XGBoost also demonstrated similar performance with an AUC of 0.660 (95% CI, 0.611–0.711, DeLong: p = 0.963). However, in the training set, GDBT still outperformed XGBoost, with an AUC of 0.603 ± 0.100 compared to XGBoost of 0.575 ± 0.096. SHAP analysis revealed that serum sodium, HDL, subarachnoid hemorrhage volume, sex, and left basal ganglia hemorrhage volume were the top five most important features for predicting cerebral edema changes in the GDBT model. Conclusion: The GDBT model demonstrated the best performance in predicting 72-h changes in cerebral edema. It has the potential to assist clinicians in identifying high-risk patients and guiding clinical decision-making.

Predicting cerebral edema in patients with spontaneous intracerebral hemorrhage using machine learning

The Value of Machine Learning Models in Predicting Factors Associated with the Need for Permanent Shunting in Patients with Intracerebral Hemorrhage Requiring Emergency Cerebrospinal Fluid Diversion

Machine learning for the prediction of in-hospital mortality in patients with spontaneous intracerebral hemorrhage in intensive care unit

Enhancing Hospital Course and Outcome Prediction in Patients with Traumatic Brain Injury: A Machine Learning Study.

Prediction of hematoma expansion in intracerebral hemorrhage in 24 hours by machine learning algorithm

Predicting Intraventricular Hemorrhage Growth with a Machine Learning-Based, Radiomics-Clinical Model

A machine learning approach for predicting perihematomal edema expansion in patients with intracerebral hemorrhage

Predicting the Outcome of Patients with Aneurysmal Subarachnoid Hemorrhage: A Machine-Learning-Guided Scorecard

Prediction of Delayed Cerebral Ischemia after Cerebral Aneurysm Rupture Using Explainable Machine Learning Approach.

A model comprising the blend sign and black hole sign shows good performance for predicting early intracerebral haemorrhage expansion: a comprehensive evaluation of CT features

Machine learning models for predicting early hemorrhage progression in traumatic brain injury

Predicting the recurrence of spontaneous intracerebral hemorrhage using a machine learning model

An Explainable Artificial Intelligence Model to Predict Malignant Cerebral Edema after Acute Anterior Circulating Large-Hemisphere Infarction

Predicting hematoma expansion using machine learning: An exploratory analysis of the ATACH 2 trial

Machine learning prediction of hematoma expansion in acute intracerebral hemorrhage

Explainable machine learning for predicting neurological outcome in hemorrhagic and ischemic stroke patients in critical care

Machine Learning-Based Approaches for Prediction of Patients’ Functional Outcome and Mortality after Spontaneous Intracerebral Hemorrhage

Machine learning for early prediction of sepsis-associated acute brain injury

Research on predicting hematoma expansion in spontaneous intracerebral hemorrhage based on deep features of the VGG-19 network

Mortality Prediction in Cerebral Hemorrhage Patients Using Machine Learning Algorithms in Intensive Care Units

Prognostic prediction of hypertensive intracerebral hemorrhage using CT radiomics and machine learning