A comprehensive study on machine learning models combining with oversampling for bronchopulmonary

Dan Wang1,2,3, Shuwei Huang4, Jingke Cao1,2

  • 1Newborn Intensive Care Unit, Faculty of Pediatrics, the Seventh Medical Center of PLA General Hospital, Beiing, China.

PubMed

Insights

Machine learning models can predict bronchopulmonary dysplasia-associated pulmonary hypertension (BPD-PH) in infants. Early identification of BPD-PH risk improves diagnosis and treatment planning for better outcomes.

Area of Science:

  • Neonatal Medicine
  • Artificial Intelligence in Healthcare
  • Pulmonary Hypertension Research

Background:

  • Bronchopulmonary dysplasia-associated pulmonary hypertension (BPD-PH) significantly impacts preterm infant outcomes.
  • Early detection of BPD-PH is crucial for reducing morbidity and improving prognosis.
  • Machine learning offers a promising approach for identifying infants at risk of BPD-PH.

Purpose of the Study:

  • To develop and validate machine learning models for early prediction of BPD-PH in infants.
  • To identify key clinical features associated with the development of BPD-PH.
  • To provide clinicians with a tool for timely diagnosis and treatment planning.

Main Methods:

  • Utilized clinical data from 761 neonatology patients across four tertiary hospitals in China.
  • Applied Synthetic Minority Over-sampling Technique (SMOTE) to address imbalanced data.
  • Selected 5 key features including respiratory support duration, BPD severity, VAP, pulmonary hemorrhage, and early-onset PH for model development.

Main Results:

  • Four machine learning models were evaluated, with a selected model achieving 93.8% sensitivity, 85.0% accuracy, and 0.933 AUC.
  • A logistic regression formula score greater than 0 was identified as a significant warning sign for BPD-PH.
  • The developed model demonstrates high predictive performance for identifying infants at risk.

Conclusions:

  • The study successfully developed a robust machine learning model for predicting BPD-PH in infants.
  • This predictive model can aid pediatric clinicians in early diagnosis and personalized treatment strategies.
  • The findings support the integration of AI in neonatal care for improved management of BPD-PH.
Abstract

Related Concept Videos