ICU Readmission Prediction for Intracerebral Hemorrhage Patients using MIMIC III and MIMIC IV Databases

Abstract

Background: Intracerebral hemorrhage (ICH) is a critical form of stroke resulting from bleeding within the brain, with a mortality rate of 40-50\% within a few days and significant risk of long-term disability. Despite the high incidence of ICU readmissions among ICH patients, the specific factors contributing to these readmissions remain unclear. This study utilizes MIMIC-III and MIMIC-IV databases to develop machine learning models that predict ICU readmissions in ICH patients. Methods: Data from 2,144 patients were extracted using ICD-9 and ICD-10 codes. Four machine learning models - AdaBoost, Random Forest, XGBoost, and LightGBM - were implemented. Recursive Feature Elimination with Cross-Validation reduced features from 50 to 18 key predictors. The RandomUnderSampler technique addressed class imbalance by reducing majority class samples to 60\% of the minority class, while the Optuna framework with Tree-structured Parzen Estimator optimized model parameters. Performance was primarily evaluated using AUROC, which effectively handles class imbalance and provides threshold-independent assessment, complemented by accuracy, sensitivity, and specificity metrics. Results: The AdaBoost model achieved an AUROC of 0.877 (95\% CI: 0.815-0.913) and accuracy of 0.810, improving from the previous best AUROC of 0.736. Sensitivity notably increased from 22.6\% to 84.0\%, demonstrating substantial improvement in identifying high-risk patients. This enhancement resulted from effective class imbalance handling and AdaBoost's adaptive weighting mechanism. Through comprehensive SHAP and ablation analyses, we identified oxygen saturation and cardiovascular disease as crucial predictive features for ICU readmission risk assessment, providing new insights into patient monitoring and care management. Conclusions: Our preprocessing methodology and model selection strategy significantly improved high-risk ICH patient identification. Through comprehensive improvements combining advanced hyperparameter optimization, balanced sampling techniques, and dual feature importance analysis, we achieved a 19.2\% AUROC improvement while reducing feature dimensionality from 51 to 18. This integrated approach demonstrates the potential of machine learning in enhancing clinical decision-making. The framework provides a promising foundation for developing clinical decision support tools in ICU settings, improving resource allocation and enabling more personalized patient care interventions.

Competing Interest Statement

The authors have declared no competing interest.

Funding Statement

No Funding

Author Declarations

I confirm all relevant ethical guidelines have been followed, and any necessary IRB and/or ethics committee approvals have been obtained.

Yes

The details of the IRB/oversight body that provided approval or exemption for the research described are given below:

The Medical Information Mart for Intensive Care III (MIMIC-III) is a comprehensive dataset, available to the public via https://physionet.org/content/mimiciii/1.4/. The Medical Information Mart for Intensive Care IV (MIMIC-IV) is a comprehensive dataset, available to the public via https://physionet.org/content/mimiciv/2.0/.

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.

Yes

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).

Yes

I have followed all appropriate research reporting guidelines, such as any relevant EQUATOR Network research reporting checklist(s) and other pertinent material, if applicable.

Yes

Data Availability

The Medical Information Mart for Intensive Care III (MIMIC-III) is a comprehensive dataset, available to the public via https://physionet.org/content/mimiciii/1.4/. The Medical Information Mart for Intensive Care IV (MIMIC-IV) is a comprehensive dataset, available to the public via https://physionet.org/content/mimiciv/2.0/.

https://physionet.org/content/mimiciii/1.4/

https://physionet.org/content/mimiciv/2.0/

留言 (0)

沒有登入
gif