Development and validation of an explainable model of brain injury in premature infants: A prospective cohort study

Zhijie He1, Ruiqi Zhang2, Pengfei Qu3

  • 1Northwest Women's and Children's Hospital, No. 1616 Yanxiang Road, Xi'an, 710061, China; College of Life Sciences, Northwest A&F University, Yangling, 712100, China.

Insights

This study developed PBIPred, a machine learning model for early preterm brain injury detection. It identifies key risk factors like mechanical ventilation and anemia, aiding in timely diagnosis and management of this critical condition.

Area of Science:

  • Neonatal Medicine
  • Computational Biology
  • Medical Informatics

Background:

  • Preterm brain injury (PBI) is a significant complication in preterm infants, causing severe neurological deficits.
  • Early detection of PBI is crucial but challenging due to non-specific early symptoms, leading to potential misdiagnosis.
  • Currently, no specific treatments exist for PBI, emphasizing the need for early identification and intervention.

Purpose of the Study:

  • To develop and validate an interpretable machine learning (ML) model for the early detection of PBI in preterm infants.
  • To identify patient-wide and individual risk factors associated with PBI.
  • To create accessible tools for clinical diagnosis and prediction of PBI.

Main Methods:

  • Utilized a cohort of 650 preterm infants' medical records (2019-2021) with PBI identified via cranial MRI.
  • Employed 14 ML models with 10-fold cross-validation and SHAP for interpretation, followed by feature selection.
  • Validated the final model on an independent test set to assess predictive performance.

Main Results:

  • The CatBoost model, refined into PBIPred (Preterm Brain Injury Predictor), achieved an AUC of 0.8229 on the independent test set.
  • PBIPred was constructed using seven key features, demonstrating strong predictive accuracy for PBI.
  • Identified positive risk factors for PBI: mechanical ventilation, weight, anemia of prematurity, respiratory distress syndrome, albumin, and white blood cell count.

Conclusions:

  • PBIPred offers a reliable and interpretable ML-based approach for early PBI detection.
  • The identified risk factors provide valuable insights for clinical risk assessment and management strategies.
  • Developed freely accessible webserver and tools (PBIPred) for clinical application and research.
Abstract