Skip to main content

ORIGINAL RESEARCH article

Front. Med.
Sec. Intensive Care Medicine and Anesthesiology
Volume 11 - 2024 | doi: 10.3389/fmed.2024.1410702
This article is part of the Research Topic Clinical Application of Artificial Intelligence in Emergency and Critical Care Medicine, Volume V View all 5 articles

Survival Prediction for Heart Failure Complicated by Sepsis: Based on Machine Learning Methods

Provisionally accepted
Qitian Zhang Qitian Zhang 1Lizhen Xu Lizhen Xu 2*Weibin He Weibin He 1*Xinqi Lai Xinqi Lai 1*Xiaohong Huang Xiaohong Huang 1*
  • 1 Zhangzhou Affiliated Hospital of Fujian Medical University, Zhangzhou, China
  • 2 Fujian Provincial Hospital, Fuzhou, Fujian Province, China

The final, formatted version of the article will be published soon.

    Background: Heart failure is a cardiovascular disorder, while sepsis is a common non-cardiac cause of mortality. Patients with combined heart failure and sepsis have a significantly higher mortality rate and poor prognosis, making early identification of high-risk patients and appropriate allocation of medical resources critically important.We constructed a survival prediction model for patients with heart failure and sepsis using the eICU-CRD database and externally validated it using the MIMIC-IV database. Our primary outcome is the 28-day all-cause mortality rate. The boruta method is used for initial feature selection, followed by feature ranking using the XGBoost algorithm. Four machine learning models were compared, including Logistic Regression (LR), eXtreme Gradient Boosting (XGBoost), Adaptive Boosting (AdaBoost), and Gaussian Naive Bayes (GNB). Model performance was assessed using metrics such as area under the curve (AUC), accuracy, sensitivity, and specificity, and the SHAP method was utilized to visualize feature importance and interpret model results. Additionally, we conducted external validation using the MIMIC-IV database.We developed a survival prediction model for heart failure complicated by sepsis using data from 3891 patients in the eICU-CRD and validated it externally with 2928 patients from the MIMIC-IV database. The LR model outperformed all other machine learning algorithms with a validation set AUC of 0.746 (XGBoost: 0.726, AdaBoost: 0.744, GNB: 0.722), alongside accuracy (0.685), sensitivity (0.666), and specificity (0.712). The final model incorporates 10 features: age, ventilation, norepinephrine, white blood cell count, total bilirubin, temperature, phenylephrine, respiratory rate, neutrophil count, and systolic blood pressure. We employed the SHAP method to enhance the interpretability of the model based on the LR algorithm. Additionally, external validation was conducted using the MIMIC-IV database, with an external validation AUC of 0.699.Based on the LR algorithm, a model was constructed to effectively predict the 28-day allcause mortality rate in patients with heart failure complicated by sepsis. Utilizing our model predictions, clinicians can promptly identify high-risk patients and receive guidance for clinical practice.

    Keywords: Heart Failure, Sepsis, eICU-CRD, MIMIC-IV, machine learning, Hospital Mortality

    Received: 01 Apr 2024; Accepted: 17 Sep 2024.

    Copyright: © 2024 Zhang, Xu, He, Lai and Huang. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) or licensor are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

    * Correspondence:
    Lizhen Xu, Fujian Provincial Hospital, Fuzhou, 350001, Fujian Province, China
    Weibin He, Zhangzhou Affiliated Hospital of Fujian Medical University, Zhangzhou, China
    Xinqi Lai, Zhangzhou Affiliated Hospital of Fujian Medical University, Zhangzhou, China
    Xiaohong Huang, Zhangzhou Affiliated Hospital of Fujian Medical University, Zhangzhou, China

    Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.