Rehospitalization in Preterm Infants: Machine Learning Prediction Model and Associated Risk Factors

Dana Benni1,2, Roni Ramon-Gonen3, Gil Klinger4,5

  • 1Department of Management, Bar-Ilan University, Ramat Gan, Israel, dana1509.db@gmail.com.

Neonatology
|April 24, 2026
PubMed

Insights

A new machine learning model predicts one-year rehospitalization risk in preterm infants after Neonatal Intensive Care Unit discharge. Key factors include early gestational age, low birth weight, and specific medical conditions, aiding early intervention.

Area of Science:

  • Neonatal Medicine
  • Machine Learning in Healthcare
  • Predictive Analytics

Background:

  • Preterm birth is a significant global health challenge.
  • Preterm infants face elevated risks of morbidity and rehospitalization post-Neonatal Intensive Care Unit (NICU) discharge.
  • Predictive modeling for these outcomes remains underexplored.

Purpose of the Study:

  • To develop a machine learning model predicting one-year rehospitalization in preterm infants.
  • To identify clinical factors associated with increased rehospitalization risk.
  • To enhance early intervention strategies for high-risk preterm infants.

Main Methods:

  • Retrospective cohort study of 2,226 preterm infants (2018-2023).
  • Data sourced from NICU and inpatient records.
  • eXtreme Gradient Boosting (XGBoost) machine learning algorithm utilized with 20 clinical predictors.

Main Results:

  • 16.1% of preterm infants were rehospitalized within one year.
  • The predictive model achieved an Area Under the Curve (AUC) of 0.69.
  • Significant predictors included early gestational age, low birth weight, prolonged NICU stay, trisomy, low socioeconomic status, specific medical conditions (gastrointestinal, neurological, bronchopulmonary dysplasia), surgical interventions, and abnormal lab values.

Conclusions:

  • This study presents a novel machine learning model for predicting one-year rehospitalization in preterm infants.
  • The model integrates clinical data for improved risk stratification.
  • Findings can inform targeted interventions for high-risk preterm infants, addressing a critical gap in neonatal care.
Abstract

Related Concept Videos