Mortality risk assessment and interpretability analysis of preterm infants in the NICU using machine learning models

Corresponding author: WU Xiayang, 56425477@qq.com

DOI: 10.12201/bmr.202503.00066

•  

  Abstract: Objective: Aimed at using machine learning algorithms to predict the risk of neonatal ICU mortality, providing clinicians with an early diagnosis and risk assessment tool to assist in decision-making. Methods: Retrospectively collecting clinical data of preterm infants from the PIC database. Cases were divided into mortality and survival groups based on ICU outcomes. Key clinical characteristics potentially affecting preterm infant outcomes were screened using LASSO regression analysis and multivariate logistic regression analysis. The study balanced the data using the SMOTE algorithm and constructed predictive models using seven machine learning models (e.g., LightGBM, random forest), evaluating their performance. Model interpretation was performed using the Shapley Additive Explanations (SHAP) algorithm. Results: A total of 923 infants were included in the final analysis. The survival group comprised 886 infants, and the death group comprised 37 infants. A total of 38 clinical characteristics were collected. LASSO screening identified 8 variables significantly associated with neonatal ICU mortality, including lactate, chloride concentration, neutrophils, and red blood cell distribution width. Multivariate logistic regression analysis revealed that lactate and respiratory rate were independent predictors of neonatal ICU outcomes. The LightGBM model achieved an AUC of 0.972 and outperformed other models in terms of accuracy and precision. Furthermore, SHAP analysis enhanced model interpretability. The results indicated that respiratory rate and lactate contributed most significantly to the prediction of infant mortality risk. Conclusion: This study provides reliable tools for early identification and intervention of preterm infant outcomes, emphasizing the importance of key physiological indicators. Future multi-center data validation is needed to enhance the models generalizability and further optimize algorithm performance.

  Key words: Premature infants, ICU mortality risk, machine learning, LightGBM model, risk prediction

  Submit time: 21 March 2025

  Copyright: The copyright holder for this preprint is the author/funder, who has granted biomedRxiv a license to display the preprint in perpetuity.
• html

• 图表

• WU Xiayang. Establishment and verification of a risk prediction model for neonatal sepsis in premature infants. 2025. doi: 10.12201/bmr.202505.00013

  Cheng Luping, Wu Siyang, Lu Bo. Analysis of Risk Factors and Prediction of Type 2 Diabetes Mellitus Based on Machine Learning. 2025. doi: 10.12201/bmr.202509.00003

  zhangwei, qindawei, tianwenjun, chenling, yanmi, guxiu, fehejian, tianchangjun. Risk factors and clinical outcome of meconium-stained amniotic fluid in preterm infants. 2025. doi: 10.12201/bmr.202502.00007

  guolianmei, sunxiaohong, zhangzhe, tianye. Construction of a Prediction Model for Lower Limb Deep Vein Thrombosis in Patients with Hemorrhagic Stroke Based on Machine Learning. 2025. doi: 10.12201/bmr.202510.00023

  Li Na, Xu Mengqi, Zhang Min, Wang Jun. Progress in risk prediction model for hypoproteinemia in orthopedic hospitalized patients. 2025. doi: 10.12201/bmr.202510.00017

  Chen Lili. Study on the immune regulatory mechanism of ω－3PUFA in premature infants with brain injury. 2024. doi: 10.12201/bmr.202408.00011

  Construction and application evaluation of risk prediction model and nomogram for shivering during cesarean sectio. 2025. doi: 10.12201/bmr.202501.00053

  wangyifan, shichaojun, maanning. Comparison of risk prediction models for atherosclerosis in type 2 diabetes mellitus. 2024. doi: 10.12201/bmr.202404.00007

  Xiao Jiali, Zhang Lijuan, Zhu Xiaohong, Zheng Ping. Application of postural assisted fixator in PICC chest X-ray tip localization in premature infants. 2025. doi: 10.12201/bmr.202508.00051

  Yang ShuJie, Zeng Ying, Yi HaiYing, Huang Jie, Gan Bin, Wei Xian, Tang Yanlin. Observation of the therapeutic effect of less invasive surfactant administration on respiratory distress syndrome in extremely premature infants. 2024. doi: 10.12201/bmr.202410.00018

• ID	Submit time	Number	Download
  1	2025-02-19	10.12201/bmr.202503.00066V1	Download

• Public  Anonymous  To author only Submit Comments