Site-Level Fine-Tuning with Progressive Layer Freezing: Towards Robust Prediction of Bronchopulmonary Dysplasia from Day-1 Chest Radiographs in Extremely Preterm Infants

• Day-1 chest radiograph predicts later lung disease risk (area under curve ≈ 0.78). • Chest-x-ray pretraining beats natural-image pretraining for neonatal scans. • Progressive layer freezing with a brief linear probe works best. • Routine respiratory distress grades only weakly predict later lung disease. Bronchopulmonary dysplasia (BPD) is a chronic lung disease affecting 35% of extremely low birth weight infants and is defined by oxygen dependence at 36 weeks postmenstrual age. Preventive interventions carry severe risks and early prediction is crucial to avoid unnecessary toxicity in low-risk infants. Admission radiographs of extremely preterm infants are routinely acquired within 24h of life and could serve as a non-invasive prognostic tool. We developed a deep learning approach using day 1 chest X-rays from 163 extremely low-birth-weight infants (≤32 weeks gestation, 401-999g). We fine-tuned a ResNet-50 pretrained specifically on adult chest radiographs, employing progressive layer freezing with discriminative learning rates to prevent overfitting and evaluated a CutMix augmentation and linear probing. Complementing prior insights that compare architectures and acquisition timing, we ablate the effects of initialization domain and compute-light fine-tuning choices on performance on small day-1 neonatal CXR cohorts, yielding practical training guidance for site-level and federated deployment. For moderate/severe BPD outcome prediction, our best performing model with progressive freezing, linear probing and CutMix achieved an AUROC of 0.78 ± 0.10, balanced accuracy of 0.69 ± 0.10, and an F1-score of 0.67 ± 0.11. In-domain pre-training significantly outperformed ImageNet initialization (p = 0.031) highlighting the importance of domain-specific pretraining. Routine IRDS grades showed limited prognostic value (AUROC 0.57 ± 0.11), motivating learned image markers. Our approach demonstrates that domain-specific pretraining enables accurate BPD prediction from routine day-1 radiographs. Through progressive freezing and linear probing, the method remains computationally feasible for site-level implementation.

Authors

Institutions

Publication Details

Journal
Machine Learning with Applications
Published
2026-03-01
DOI
https://doi.org/10.1016/j.mlwa.2026.100890
Primary Topic
Neonatal Respiratory Health Research
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Site-Level Fine-Tuning with Progressive Layer Freezing: Towards Robust Prediction of Bronchopulmonary Dysplasia from Day-1 Chest Radiographs in Extremely Preterm Infants

Dorothea Kerner, Sogand Nemat, Michelle Bous, Annika Engel et al.
Machine Learning with Applications
Neonatal Respiratory Health Research
article

Site-Level Fine-Tuning with Progressive Layer Freezing: Towards Robust Prediction of Bronchopulmonary Dysplasia from Day-1 Chest Radiographs in Extremely Preterm Infants

Dorothea Kerner, Sogand Nemat, Michelle Bous, Annika Engel, Dwayne Mohr, Mathias Kaspar, Pascal Hirsch, Matthias Flotho, Philipp Flotho, Hannah Wittig, Sascha Meyer, Michael Zemlin, Dino Milanovic, Andreas Keller, Sybelle Goedicke-Fritz
article en

Abstract

• Day-1 chest radiograph predicts later lung disease risk (area under curve ≈ 0.78). • Chest-x-ray pretraining beats natural-image pretraining for neonatal scans. • Progressive layer freezing with a brief linear probe works best. • Routine respiratory distress grades only weakly predict later lung disease. Bronchopulmonary dysplasia (BPD) is a chronic lung disease affecting 35% of extremely low birth weight infants and is defined by oxygen dependence at 36 weeks postmenstrual age. Preventive interventions carry severe risks and early prediction is crucial to avoid unnecessary toxicity in low-risk infants. Admission radiographs of extremely preterm infants are routinely acquired within 24h of life and could serve as a non-invasive prognostic tool. We developed a deep learning approach using day 1 chest X-rays from 163 extremely low-birth-weight infants (≤32 weeks gestation, 401-999g). We fine-tuned a ResNet-50 pretrained specifically on adult chest radiographs, employing progressive layer freezing with discriminative learning rates to prevent overfitting and evaluated a CutMix augmentation and linear probing. Complementing prior insights that compare architectures and acquisition timing, we ablate the effects of initialization domain and compute-light fine-tuning choices on performance on small day-1 neonatal CXR cohorts, yielding practical training guidance for site-level and federated deployment. For moderate/severe BPD outcome prediction, our best performing model with progressive freezing, linear probing and CutMix achieved an AUROC of 0.78 ± 0.10, balanced accuracy of 0.69 ± 0.10, and an F1-score of 0.67 ± 0.11. In-domain pre-training significantly outperformed ImageNet initialization (p = 0.031) highlighting the importance of domain-specific pretraining. Routine IRDS grades showed limited prognostic value (AUROC 0.57 ± 0.11), motivating learned image markers. Our approach demonstrates that domain-specific pretraining enables accurate BPD prediction from routine day-1 radiographs. Through progressive freezing and linear probing, the method remains computationally feasible for site-level implementation.

Machine Learning with Applications
Goethe University Frankfurt (DE), Okinawa Institute of Science and Technology Graduate University (JP), University Hospital Augsburg (DE), Helmholtz Institute for Pharmaceutical Research Saarland (DE), Saarland University (DE)
Zero hunger
Openalex Percentile: Top 6%
Neonatal Respiratory Health Research
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.