Development and validation of a machine learning model for in-hospital mortality prediction in children under 5 years

Huasheng Lv1, Fengyu Sun2, Teng Yuan1

  • 1Department of Cardiology, The First Affiliated Hospital of Xinjiang Medical University, Urumqi, China.

PubMed

Insights

A new machine learning model accurately predicts in-hospital mortality in young children with heart failure (HF). This tool aids early risk stratification for better pediatric HF outcomes.

Area of Science:

  • Pediatric Cardiology
  • Computational Biology
  • Biomedical Informatics

Background:

  • Heart failure (HF) in children under five presents a high mortality risk.
  • Current pediatric risk tools lack specificity for this age group.
  • Reliable, interpretable prediction models for pediatric HF are critically needed.

Purpose of the Study:

  • To develop and validate a machine learning model for predicting in-hospital mortality in young children with heart failure.
  • To identify key predictors of mortality in this population.

Main Methods:

  • Retrospective analysis of 630 pediatric HF cases (2013-2024).
  • Feature selection using the Boruta algorithm identified seven key predictors.
  • Extreme Gradient Boosting (XGB) model developed and interpreted using SHAP; externally validated on 73 cases.

Main Results:

  • The XGB model demonstrated high predictive performance (AUC: 0.916 training, 0.851 internal, 0.846 external validation).
  • Key predictors identified: NT-proBNP, pH, PCT, LDH, WBC, creatinine, and platelet count.
  • SHAP analysis confirmed the clinical significance of these predictors.

Conclusions:

  • A reliable and interpretable machine learning model for predicting pediatric HF mortality has been developed.
  • This model can facilitate early risk stratification and timely interventions.
  • The model shows potential to improve outcomes for high-risk pediatric HF patients.
Abstract