Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Evaluating Synthetic Data Generation for Domain Generalization in Fetal Brain MRI Segmentation

About

Fetal brain tissue segmentation from magnetic resonance imaging (MRI) is crucial for studying neurodevelopment, but remains challenging due to data heterogeneity and limited annotations. Domain randomization (DR) has recently emerged as a promising strategy for single-source domain generalization by synthesizing training images with randomized artifacts, contrast, and resolution. In this work, we investigate how to maximize the out-of-domain (OOD) generalization of DR-based methods. We evaluate several synthetic data generation strategies for DR, with a particular focus on our recently proposed framework, FetalSynthSeg. We show that simple Gaussian mixture-based intensity modeling outperforms more complex physics-based simulations, and that intensity clustering (subdividing tissue classes based on intensity) improves OOD robustness. Evaluated on 348 fetal subjects from four sites spanning 0.55-3T and both T1w and T2w contrasts, FetalSynthSeg reaches state-of-the-art performance on several FeTA 2024 testing datasets (80-85 Dice score) and, for the first time, offers robust segmentation on modalities other than T2w for fetal brain segmentation (80 Dice on dHCP-T1w dataset). Compared with state-of-the-art methods such as BOUNTI, nnU-Net ensemble, and the FeTA 2024 winner, FetalSynthSeg delivers comparable or superior accuracy while maintaining strong robustness across domain shifts. Our code, model weights, and Docker image ready for easy inference are available at https://hub.docker.com/r/vzalevskyi/fetalsynthseg.

Vladyslav Zalevskyi, Thomas Sanchez, Margaux Roulet, Busra Bulut, H\'el\`ene Lajous, Jordina Aviles Verdera, Sara Neves Silva, Georg Langs, Gregor Kasprian, Roxane Licandro, Jana Hutter, Hamza Kebiri, Meritxell Bach Cuadra• 2024

Related benchmarks

TaskDatasetResultRank
Fetal Brain SegmentationdHCP T1w
DSC80.1
6
Fetal Brain SegmentationCHUV T2w
DSC83.2
6
Fetal Brain SegmentationdHCP T2w
DSC84
6
Fetal Brain SegmentationKCL T2w
DSC85.3
6
Fetal Brain SegmentationVIEN T2w
DSC79.5
6
Showing 5 of 5 rows

Other info

Follow for update