用信息差异提升胎儿超声图像分割的标注效率与鲁棒性
Leveraging Information Divergence for Robust Semi-Supervised Fetal Ultrasound Image Segmentation
- 双模型交叉监督,通过信息差异损失约束预测一致性
- 仅5%标注数据时Dice提升2.39%, Hausdorff距离降低14.90
- 适合医疗图像少样本分割场景,代码开源可用
产前超声是监测胎儿发育的主要手段,但自动化分割因高质量标注稀缺而困难。为解决此问题,我们提出一种半监督学习框架,利用信息差异实现鲁棒的胎儿超声图像分割。方法采用轻量级卷积网络(1.47M参数)与基于Transformer的网络,联合使用标注数据进行标准监督,以及未标注数据通过交叉监督训练。为促进一致且自信的预测,引入信息差异损失,结合像素级Kullback-Leibler散度与互信息差距,有效减少两模型间的预测分歧。此外,对未标注样本应用mixup以进一步增强鲁棒性。在两个胎儿超声数据集上的实验表明,本方法持续优于七种先进半监督方法。当仅5%训练数据标注时,该框架将Dice分数提升2.39%,95% Hausdorff距离降低14.90,平均表面距离减少4.18。结果验证了利用信息差异实现标注高效与鲁棒医学图像分割的有效性。代码已公开于GitHub。
原文摘要 · Abstract (English)
Maternal-fetal Ultrasound is the primary modality for monitoring fetal development, yet automated segmentation remains challenging due to the scarcity of high-quality annotations. To address this limitation, we propose a semi-supervised learning framework that leverages information divergence for robust fetal ultrasound segmentation. Our method employs a lightweight convolutional network (1.47M parameters) and a Transformer-based network, trained jointly with labelled data through standard supervision and with unlabelled data via cross-supervision. To encourage consistent and confident predictions, we introduce an information divergence loss that combines per-pixel Kullback-Leibler divergence and Mutual Information Gap, effectively reducing prediction disagreement between the two models. In addition, we apply mixup on unlabelled samples to further enhance robustness. Experiments on two fetal ultrasound datasets demonstrate that our approach consistently outperforms seven state-of-the-art semi-supervised methods. When only 5% of training data is labelled, our framework improves the Dice score by 2.39%, reduces the 95% Hausdorff distance by 14.90, and decreases the Average Surface Distance by 4.18. These results highlight the effectiveness of leveraging information divergence for annotation-efficient and robust medical image segmentation. Our code is publicly available on GitHub.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。