Death risk prediction model for patients with non-traumatic intracerebral hemorrhage

Yidan Chen1, Xuhui Liu2, Mingmin Yan3

  • 1Jianghan University School of Medicine, Wuhan, China.

Insights

A machine learning model effectively predicts death risk in intracerebral hemorrhage (ICH) patients. Key predictors include GCS motor score, age, GCS eye score, LDL, albumin, atrial fibrillation, and gender.

Area of Science:

  • Neurology
  • Medical Informatics
  • Biostatistics

Background:

  • Intracerebral hemorrhage (ICH) poses a significant mortality risk.
  • Accurate prediction of ICH-related mortality is crucial for patient management.
  • Machine learning offers potential for improving risk stratification in ICH.

Purpose of the Study:

  • To develop and validate a machine learning model for predicting the risk of death in patients with non-traumatic intracerebral hemorrhage (ICH).
  • To identify key clinical and demographic factors associated with mortality in ICH patients.

Main Methods:

  • Retrospective analysis of 1274 ICH patients from the MIMIC IV 3.0 database.
  • Feature selection using LASSO and multivariable logistic regression.
  • Development of an XGBoost machine learning model for mortality prediction.
  • External validation using data from the Second Hospital of Lanzhou University.
  • Model performance evaluation using ROC, calibration curves, and accuracy metrics.
  • Interpretation of model predictions using SHapley Additive exPlanations (SHAP).

Main Results:

  • The study included 1274 ICH patients with a mortality rate of 44.9%.
  • The XGBoost model demonstrated strong predictive performance with high AUC values in training, validation, and testing datasets.
  • Key predictors identified by SHAP analysis included GCS motor score, age, GCS eye score, LDL, albumin, atrial fibrillation, and gender.

Conclusions:

  • The developed XGBoost model is effective in predicting the risk of death in ICH patients.
  • The SHAP method provides valuable interpretability for the machine learning model's predictions.
  • This model can aid in clinical decision-making and risk stratification for ICH.
Abstract