arXiv:2502.02489cs.CVcs.AI2025-02被引 4

用自监督学习提升超声图像分割泛化能力,尤其在数据少时表现更优。

A Self-Supervised Framework for Improved Generalisability in Ultrasound B-mode Image Segmentation

  • 设计关系对比损失,通过可学习度量区分正负样本对。
  • 在三个乳腺超声数据集上,小样本下Dice评分提升3.7%至9%。
  • 在分布外数据上性能提升达20.6%,适合临床数据稀缺场景。

超声成像因其无创安全特性在临床上至关重要,但解读困难且易出错。深度学习可辅助分割,但传统监督方法依赖大量高质量标注数据,且在分布外数据上表现差。本文提出一种针对B-mode超声图像的对比自监督学习框架,引入新颖的关系对比损失(RCL),通过可学习度量区分正负样本对。同时设计空间与频域增强策略以提升表征学习效果。在三个公开乳腺超声数据集上,本方法显著优于传统监督分割模型,尤其在数据受限情况下:在BUSI数据集20%和50%样本下,Dice分数分别提升4%;在BrEaST数据集对应比例下分别提升近6%和9%;在UDIAT数据集对应比例下分别提升6.4%和3.7%。此外,在分布外的UDIAT数据集上,使用20%和50%的BUSI与BrEaST训练数据时,性能分别提升20.6%和13.6%。结果表明,领域启发的自监督学习能有效提升超声图像分割的泛化性。

原文摘要 · Abstract (English)

Ultrasound (US) imaging is clinically invaluable due to its noninvasive and safe nature. However, interpreting US images is challenging, requires significant expertise, and time, and is often prone to errors. Deep learning offers assistive solutions such as segmentation. Supervised methods rely on large, high-quality, and consistently labeled datasets, which are challenging to curate. Moreover, these methods tend to underperform on out-of-distribution data, limiting their clinical utility. Self-supervised learning (SSL) has emerged as a promising alternative, leveraging unlabeled data to enhance model performance and generalisability. We introduce a contrastive SSL approach tailored for B-mode US images, incorporating a novel Relation Contrastive Loss (RCL). RCL encourages learning of distinct features by differentiating positive and negative sample pairs through a learnable metric. Additionally, we propose spatial and frequency-based augmentation strategies for the representation learning on US images. Our approach significantly outperforms traditional supervised segmentation methods across three public breast US datasets, particularly in data-limited scenarios. Notable improvements on the Dice similarity metric include a 4% increase on 20% and 50% of the BUSI dataset, nearly 6% and 9% improvements on 20% and 50% of the BrEaST dataset, and 6.4% and 3.7% improvements on 20% and 50% of the UDIAT dataset, respectively. Furthermore, we demonstrate superior generalisability on the out-of-distribution UDIAT dataset with performance boosts of 20.6% and 13.6% compared to the supervised baseline using 20% and 50% of the BUSI and BrEaST training data, respectively. Our research highlights that domain-inspired SSL can improve US segmentation, especially under data-limited conditions.

超声分割自监督学习小样本医学影像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。