无需提示即可一次完成胎儿头颅多结构分割,精度显著提升。
Domain-Guided YOLO26 with Composite BCE-Dice-Lovász Loss for Multi-Class Fetal Head Ultrasound Segmentation
- 用复合损失+领域引导增强实现无提示端到端分割
- 平均Dice达0.925,比基线高2.68个百分点
- 适合产科超声自动化分析与小样本结构分割
孕期超声中胎儿头颅结构的分割仍是临床实践中的瓶颈。现有最佳基线方法虽采用Segment Anything模型并结合逐类Dice与Lovász损失,但仍需测试时提供边界框提示。本文基于YOLO26-Seg构建无提示流程,单次前向传播联合检测并分割脑部、透明隔腔(CSP)和侧脑室(LV)三个结构。核心改进包括:(i) 在训练中注入带逆频率类别加权的复合BCE-Dice-Lovász损失,通过运行时猴子补丁实现;(ii) 领域引导的复制粘贴增强,将少数类结构移植至符合解剖位置关系的脑边界区域;(iii) 按患者分层划分数据集以防止泄漏。在575张保留测试图像上,复合损失版本达到0.9253的平均Dice系数,较基线(0.9012)提升2.68个百分点,且仅报告三个前景类,而基线包含易分的背景类。进一步消融实验分析了标注质量与类别不平衡对CSP和LV性能的影响。
原文摘要 · Abstract (English)
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ász 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ász 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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。