Abstract:Objective: Risk stratification of patients with congestive heart failure (HF) is vital in clinical practice. The aim of this study was to construct a machine learning model to predict the in-hospital all-cause mortality for intensive care unit (ICU) patients with HF. Methods: eXtreme Gradient Boosting algorithm (XGBoost) was used to construct a new prediction model ( XGBoost model ) from the Medical Information Mart for Intensive Care IV database (MIMIC-IV) (training set). The eICU Collaborative Research Database dataset (eICU-CRD) was used for the external validation (test set). The XGBoost model performance was compared with a logistic regression model and an existing model (Get with the guideline-Heart Failure model) for mortality in the test set. Area under the receiver operating characteristic cure and Brier score were employed to evaluate the discrimination and the calibration of the three models. The SHapley Additive exPlanations (SHAP) value was applied to explain XGBoost model and calculate the importance of its features. Results: The total of 11,156 and 9,837 patients with congestive HF from the training set and test set, respectively, were included in the study. In-hospital all-cause mortality occurred in 13.3% (1,484/11,156) and 13.4% (1,319/9,837) of patients, respectively. In the training set, of 17 features with the highest predictive value were selected into the models with LASSO regression. Acute Physiology Score III (APS III), age and Sequential Organ Failure Assessment (SOFA) were strongest predictors in SHAP. In the external validation, the XGBoost model performance was superior to that of conventional risk predictive methods, with an area under the curve of 0.771 (95% confidence interval, 0.757–0.784) and a Brier score of 0.100. In the evaluation of clinical effectiveness, the machine learning model brought a positive net benefit in the threshold probability of 0%–90%, prompting evident competitiveness compare to the other two models. This model has been translated into an online calculator which is accessible freely to the public (https://nkuwangkai-app-for-mortality-prediction-app-a8mhkf.streamlit.app). Conclusion: This study developed a valuable machine learning risk stratification tool to accurately assess and stratify the risk of in-hospital all-cause mortality in ICU patients with congestive HF. This model was translated into a web-based calculator which access freely.

Personalized risk prediction of mortality and rehospitalization for heart failure in patients undergoing mitral valve repair surgery

Machine Learning-Based Personalized Risk Prediction Model for Mortality of Patients Undergoing Mitral Valve Surgery: the PRIME Score

Risk Prediction Model for In-Hospital Mortality in Women with ST-elevation Myocardial Infarction: A Machine Learning Approach

Predicting the risk of 1-year mortality among patients hospitalized for acute heart failure in China

Application of machine learning to predict in-hospital mortality after transcatheter mitral valve repair

Multi-parametric system for risk stratification in mitral regurgitation: A multi-task Gaussian prediction approach

Development and validation of a predicting model for risk of mortality in heart failure patients with mildly reduced ejection fraction post discharge

A Model for the Prediction of Mortality and Hospitalization in Chinese Heart Failure Patients

Derivation and validation of predictive indices for cardiac readmission after coronary and valvular surgery - A multicenter study

Development and Validation of a Score Predicting Mortality for Older Patients with Mitral Regurgitation.

Machine learning-based in-hospital mortality risk prediction tool for intensive care unit patients with heart failure

Development of an Electronic Frailty Index for Predicting Mortality in Patients Undergoing Transcatheter Aortic Valve Replacement Using Machine Learning

Risk prediction models for mortality and readmission in patients with acute heart failure: A protocol for systematic review, critical appraisal, and meta-analysis

A Novel Model for Predicting a Composite Outcome of Major Complications after Valve Surgery

The Prognostic Multivariate Models For Severe Complications After Heart Valve Surgery

Development and Validation of a Risk Score for Predicting 1-Year Mortality Risk of STEMI Based on a Nationwide Cohort in China

Machine Learning–Based Risk Model for Predicting Early Mortality after Surgery for Infective Endocarditis

Mortality Prediction System for Heart Failure with Orthogonal Relief and Dynamic Radius Means.

Application of model for end-stage liver disease as disease classification in cardiac valve surgery: a retrospective study based on the INSPIRE database

Risk prediction model of in-hospital heart failure in patients with coronary heart disease after percutaneous coronary intervention:development and validation

Prediction model of in-hospital mortality in intensive care unit patients with heart failure: machine learning-based, retrospective analysis of the MIMIC-III database