文献类型：期刊文献

英文题名：An interpretable machine learning model for predicting NICU admission in preterm infants: a single-center retrospective cohort study

作者：Ma, Zhanying[1];Zhang, Jianzhi[1];Ma, Hong[2];Sun, Yonghong[3];Yang, Yue[3];Yu, Yaqiong[2]

第一作者：马志元

通信作者：Yu, YQ[1]

机构：[1]Gansu Univ Chinese Med, Clin Med Coll 1, Lanzhou, Peoples R China;[2]Gansu Prov Hosp, Dept Intervent Oncol, 204 Donggang West Rd, Lanzhou 730000, Peoples R China;[3]Gansu Prov Hosp, Dept Pediat, Lanzhou, Peoples R China

第一机构：甘肃中医药大学

通信机构：[1]corresponding author), Gansu Prov Hosp, Dept Intervent Oncol, 204 Donggang West Rd, Lanzhou 730000, Peoples R China.

年份：2026

卷号：15

期号：4

外文期刊名：TRANSLATIONAL PEDIATRICS

收录：;Scopus(收录号：2-s2.0-105040014783);WOS:【SCI-EXPANDED(收录号:WOS:001773355100041)】；

语种：英文

外文关键词：Preterm infants; neonatal intensive care; machine learning (ML); interpretability; Shapley Additive Explanations (SHAP)

摘要：Background: Admission to the neonatal intensive care unit (NICU) is a critical event for preterm infants, with significant implications for resource allocation and parental counseling. However, existing prediction tools are often limited by low accuracy or lack of interpretability. This study aimed to develop an interpretable machine learning (ML) model for predicting NICU admission in preterm infants using readily available prenatal and intrapartum features, with a focus on both the overall cohort and the clinically challenging subgroup of late preterm infants (34-37 weeks). Methods: A retrospective cohort of 2,610 preterm infants was analyzed. Features were selected using Boruta and least absolute shrinkage and selection operator (LASSO). Multiple models were trained and optimized via 5-fold cross-validation. The optimal model was evaluated using area under the curve (AUC), calibration, and decision curve analysis. Subgroup analysis was performed in late preterm infants (34-37 weeks) to assess model performance in this population. Interpretability was assessed with Shapley Additive exPlanations (SHAP). Results: The random forest (RF) model demonstrated superior and robust performance, achieving an AUC of 0.861 [95% confidence interval (CI): 0.830-0.891] in the validation set and 0.869 (0.841-0.897) in the testing set. SHAP analysis identified birth weight (mean |SHAP| value =0.17), prenatal checkup status (0.13), and gestational age (0.09) as the three most influential predictors. Low birth weight, lack of prenatal care, and gestational age below 32 weeks were associated with a significantly elevated risk of NICU admission. In the late preterm subgroup (34-37 weeks), the RF model maintained robust performance with an AUC of 0.842 (validation) and 0.838 (test), demonstrating good calibration and positive net benefit on decision curve analysis. Conclusions: The interpretable ML model developed in this study accurately identifies preterm infants at high risk of NICU admission, with consistent performance in the late preterm subgroup. By providing individualized risk quantification and visual explanation via SHAP, it facilitates timely clinical decision-making and enhances clinician-parent communication. This tool holds significant potential for optimizing resource allocation and improving perinatal care pathways.