Research on predicting risk factors for re-bleeding in the acute phase of intracerebral hemorrhage using machine learning algorithms.

Objective: To investigate risk factors for rebleeding in patients during the acute phase of intracerebral hemorrhage, compare the predictive performance of various machine learning models, and develop an optimal predictive model based on SHAP interpretation. Methods: A retrospective analysis of clinical data from 368 patients with intracerebral hemorrhage was conducted. First, the chi-square test and the Mann-Whitney U test were used to compare the rebleeding group with the non-rebleeding group; Lasso regression was then adopted to screen predictive factors. Based on the six core variables identified, the 10-fold cross-validation performance of nine models, XGBoost, logistic regression, LightGBM, random forest, AdaBoost, decision tree, GBDT, Gaussian Naive Bayes, and k-nearest neighbors, was compared. The most effective GBDT algorithm was used as the foundation, and four additional potential variables were incorporated based on clinical relevance to construct a final GBDT model comprising 10 variables. Bootstrap resampling (1,000 times) was used to calculate the confidence interval for the model's AUC. Calibration curves and decision curves were employed to evaluate the model's calibration and clinical utility, while SHAP values were utilized for global and local interpretation. SHAP dependency plots were generated for each feature, and a predictive website was ultimately developed based on this model. Results: There were statistically significant differences between the two groups in the history of anticoagulant or antiplatelet therapy, hematoma morphology, intraventricular hemorrhage, body temperature, white blood cell count, time from onset to CT scan, hematoma volume, Hematoma Heterogeneity Index (HII), GCS score, lactate, and partial pressure of carbon dioxide ( Conclusion: In this single-center retrospective study, a 10-variable GBDT model accurately predicted early rebleeding risk after acute intracerebral hemorrhage, showing stable Bootstrap performance and improved interpretability via SHAP, which can aid early detection of high-risk patients.