FETALFusion: Resolution-Aware State-Space Modeling for Multi-Domain Fetal Head Segmentation
Organizations: School of Computer Engineering, KIIT Deemed to be University Bhubaneswar, India
Abstract
Head circumference (HC) measured from 2D ultrasound is a routine marker of fetal growth, and automating it needs segmentations whose boundaries hold up on individual scans, not only on average. We present FETALFusion, a dual-path CNN-Mamba encoder-decoder for fetal head segmentation across acquisition types. It applies four-directional state-space scans only at the two highest-resolution stages, routes bottleneck features through domain-specific channel gates that are averaged at test time, and adds a weak landmark shape prior. We train jointly on transabdominal (HC-18) and transperineal (PSFHS) images and compare ten methods on a held-out test set of 471 images, each trained under one schedule and scored by surface distances at native resolution. FETALFusion reaches a Dice score of 0.963 and an HD95 of 2.10 mm without test-time augmentation or a domain label. It improves on U-Mamba on all five reported metrics (HD95 reduced by 0.31 mm; Holm-corrected (), a lead that holds under test-time augmentation and across two independent training runs, and in the mean on U-Net, Attention U-Net and TransUNet. Residual encoders close the gap: a ResNet-34 UNet++ exceeds FETALFusion, and an ImageNet initialised ResNet-34 U-Net is best overall (HD95 1.75 mm). The ablation agrees. Calibrated against run-to-run variation, it attributes FETALFusion's measurable gains to the landmark shape prior and, on transperineal images, to the domain gates, while removing the state-space branch does not measurably change accuracy. HC from the predicted masks agrees with the clinical reference at an intraclass correlation of 0.9986.
Figures & tables
| No test-time augmentation | With TTA | ||||||||||
| Method | Params (M) | ms/img | DSC | IoU | HD95 | HD95 med. | ASD | NSD | 10 mm | DSC | HD95 |
| U-Net [ 9 ] | 31.0 | 40.6 | .9600 | .9272 | 2.32 | 1.51 | 0.78 | .8816 | 9 | .9608 | 2.10 |
| Att. U-Net [ 13 ] | 31.4 | 40.8 | .9603 | .9272 | 2.26 | 1.51 | 0.78 | .8804 | 9 | .9610 | 2.15 |
| ResU-Net (nnU-Net-style) [ 11 ] | 5.9 | 9.4 | .9557 | .9180 | 2.70 | 1.73 | 0.88 | .8556 | 12 | .9592 | 2.42 |
| ResNet-34 U-Net [ 17 ] | 24.4 | 6.4 | .9605 | .9278 | 2.02 | 1.46 | 0.75 | .8850 | 4 | .9626 | 1.88 |
| UNet++ (ResNet-34) [ 18 ] | 26.1 | 11.9 | .9640 | .9323 | 1.88 | 1.48 | 0.70 | .8929 | 2 | .9645 | 1.85 |
| Variant | DSC | HD95 | NSD | |||
|---|---|---|---|---|---|---|
| Full model (2 runs) | .9629 | .0004 | 2.047 | .076 | .8863 | .0012 |
| w/o SSM branch | .9628 | .0001 | 1.944 | .103 | .8887 | .0024 |
| w/o DPAF | .9628 | .0001 | 2.135 | .088 | .8871 | .0008 |
| w/o landmark prior | .9589 | .0040 | 2.143 | .096 | .8684 | .0179 |
| Shared gate, no CDFR | .9612 | .0017 | 2.093 | .046 | .8777 | .0086 |
| Schedule | DSC | HD95 | NSD | SSM | ms/img | Mem |
|---|---|---|---|---|---|---|
| 1-1-1-1 | .9620 | 2.048 | .8777 | 2.44 | 22.0 | 4.64 |
| 4-1-1-1 | .9587 | 2.179 | .8670 | 2.67 | 31.4 | 6.47 |
| 4-4-1-1 (ours) | .9629 | 2.047 | .8863 | 3.20 | 35.9 | 7.18 |
| 4-4-4-4 | .9624 | 1.957 | .8797 | 9.20 | 42.4 | 7.63 |
Medical Image Segmentation
Medical image segmentation delineates organs, lesions, and other structures in scans to support measurement, diagnosis, treatment planning, and research.
See all 786 papers