Domain-Guided YOLO26 with Composite BCE-Dice-Lov'asz Loss for Multi-Class Fetal Head Ultrasound Segmentation
Not on the map. Why?
Abstract (arXiv)
Segmenting fetal head structures from prenatal ultrasound remains a practical bottleneck in obstetric imaging. The current state-of-the-art baseline, proposed alongside the published dataset, adapts the Segment Anything Model with per-class Dice and Lov'asz losses but still depends on bounding-box prompts at test time. We build a prompt-free pipeline on top of YOLO26-Seg that jointly detects and segments three structures, Brain, Cavum Septi Pellucidi (CSP), and Lateral Ventricles (LV), in a single forward pass. Three modifications are central to our approach: (i) a composite BCE-Dice-Lov'asz segmentation loss with inverse-frequency class weighting, injected into the YOLO26 training loop via runtime monkey-patching; (ii) domain-guided copy-paste augmentation that transplants minority-class structures while respecting their anatomical location relative to the brain boundary; and (iii) inter-patient stratified splitting to prevent data leakage. On 575 held-out test images, the composite loss variant reaches a mean Dice coefficient of 0.9253, exceeding the baseline (0.9012) by 2.68 percentage points, despite reporting over three foreground classes only, whereas the baseline's reported mean includes the easy background class. We further ablate each component and discuss annotation-quality and class-imbalance effects on CSP and LV performance.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 24 of its 38 references that are in the map. First posted on arXiv 23 Mar 2026.
Builds on
24 of 38 references.
References · 24
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2014Microsoft COCO: Common Objects in Contextcited by 17,498
- 2023Segment Anythingcited by 8,158
- 2017Mask R-CNNcited by 7,259
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,415
- 2016Feature Pyramid Networks for Object Detectioncited by 5,021
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
- 2015You Only Look Once: Unified, Real-Time Object Detectioncited by 4,829
- 2018Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentationcited by 3,210
12 more
- 2021SegFormer: Simple and Efficient Design for Semantic Segmentation with Transformerscited by 2,923
- 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,518
- 2017Mixed Precision Trainingcited by 1,135
- 2018Path Aggregation Network for Instance Segmentationcited by 796
- 2020The Future of Digital Health with Federated Learningcited by 625
- 2020Simple Copy-Paste is a Strong Data Augmentation Method for Instance Segmentationcited by 595
- 2019YOLACT: Real-time Instance Segmentationcited by 441
- 2019CSPNet: A New Backbone that can Enhance Learning Capability of CNNcited by 381
- 2023A Comprehensive Review of YOLO Architectures in Computer Vision: From YOLOv1 to YOLOv8 and YOLO-NAScited by 149
- 2025YOLO26: Key Architectural Enhancements and Performance Benchmarking for Real-Time Object Detectioncited by 76
- 2022A Review on Deep-Learning Algorithms for Fetal Ultrasound-Image Analysiscited by 29
- 2022Domain-guided data augmentation for deep learning on medical imagingcited by 5
Built on by
Not cited yet.