Abstract
Backgrounds Early and accurate identification of patients with spontaneous intracerebral hemorrhage(sICH) who are at high risk of in-hospital death can help intensive care unit (ICU) physicians make optimal clinical decisions. The aim of this study was to develop a machine learning(ML)-based tool to predict the risk of in-hospital death in patients with sICH in ICU.
Methods We conducted a retrospective administrative database study using the MIMIC-IV and Zhejiang Hospital database. The outcome of the study was in-hospital mortality. To develop and validate the final model, we employed the LASSO regression to screen and select relevant variables. Five algorithms, namely Logistic Regression (LR), K-Nearest Neighbors (KNN), Adaptive Boosting (AdaBoost), Random Forest (RF), and eXtreme Gradient Boosting (XGBoost), were utilized. The selection of the best model was based on the area under the curve (AUC) in the validation cohort. Furthermore, we employ the SHapley Additive exPlanations (SHAP) methodology to elucidate the contributions of individual features to the model and analyze their impact on the model’s outputs. To facilitate accessibility, we also created a visual online calculation page for the model.
Results In the final cohort comprising 1596 patients from MIMIC-IV and Zhejiang Hospital, 367 individuals (23%) experienced in-hospital mortality during the inpatient follow-up period. After extracting 46 variables from the database, LASSO regression identified 14 predictor variables for further analysis. Among the five evaluated models, the XGBoost model demonstrated superior discriminative power in both the internal validation set (AUC = 0.907) and the external validation set (AUC = 0.787). Furthermore, through the SHAP technique, we identified the top 5 predictors in the feature importance rankings: Glasgow Coma Scale (GCS), Sequential Organ Failure Assessment (SOFA), anticoagulant medication, mannitol medication and oxygen saturation.
Conclusions Among the five models, the XGBoost model exhibited superior performance in predicting mortality for patients with sICH in the ICU, indicating its potential significance in the development of early warning systems.
Competing Interest Statement
The authors have declared no competing interest.
Funding Statement
This work was supported by the grants from Special Support Program for High Level Talents of Zhejiang Province?2022R52038?, Key Research and Development Project of Zhejiang Provincial Department of Science and Technology(2021C03105), Natural Science Foundation of Zhejiang Province (LY21H090008) and Medical Health Science and Technology Project of Zhejiang Provincial Health Commission (WKJ-ZJ-2340).
Author Declarations
I confirm all relevant ethical guidelines have been followed, and any necessary IRB and/or ethics committee approvals have been obtained.
Not Applicable
The details of the IRB/oversight body that provided approval or exemption for the research described are given below:
The study protocol was approved by the Ethics Committee of Zhejiang Hospital (No. 2023 Pro-examination (58K)).
I confirm that all necessary patient/participant consent has been obtained and the appropriate institutional forms have been archived, and that any patient/participant/sample identifiers included were not known to anyone (e.g., hospital staff, patients or participants themselves) outside the research group so cannot be used to identify individuals.
Not Applicable
I understand that all clinical trials and any other prospective interventional studies must be registered with an ICMJE-approved registry, such as ClinicalTrials.gov. I confirm that any such study reported in the manuscript has been registered and the trial registration ID is provided (note: if posting a prospective study registered retrospectively, please provide a statement in the trial ID field explaining why the study was not registered in advance).
Not Applicable
I have followed all appropriate research reporting guidelines, such as any relevant EQUATOR Network research reporting checklist(s) and other pertinent material, if applicable.
Not Applicable
Data Availability
The datasets generated during and/or analyzed during the current study are available from the corresponding author on reasonable request.
