FetSAM: Advanced Segmentation Techniques for Fetal Head Biometrics in Ultrasound Imagery
Mahmood Alzubaidi, Uzair Shah, Marco Agus, Mowafa Househ
College of Science and Engineering, Hamad Bin Khalifa University, Doha, Qatar
IEEE Open Journal of Engineering in Medicine and Biology, Vol. 5 · 2024
FetSAM is a cutting-edge deep learning model aimed at revolutionizing fetal head ultrasound segmentation, elevating prenatal diagnostic precision. Trained on the largest fetal head metrics dataset to date, the model incorporates prompt-based learning and a dual-loss mechanism combining Weighted DiceLoss and Weighted Lovasz Loss optimized through AdamW, with class weight adjustments for better segmentation balance. Benchmarked against U-Net, DeepLabV3, and Segformer, FetSAM delivers unparalleled accuracy and sets a new benchmark for AI-enhanced prenatal ultrasound analysis.
- DSC of 0.90117 — outperforms U-Net, DeepLabV3, Segformer
- Hausdorff Distance: 1.86484 · Average Surface Distance: 0.46645
- Prompt-based learning + dual-loss optimization (Weighted Dice + Lovasz)
- Targets fetal brain, CSP, and lateral ventricles (LV) segmentation