arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2502.02489cs.CVcs.AIcs.LG

一种用于提高超声B模式图像分割泛化能力的自监督框架

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

Satisfai Health(Satisfai Health公司)

Edward Ellis, Andrew Bulpitt, Nasim Parsa, Michael F Byrne, Sharib Ali

AI总结:

本文提出一种针对超声B模式图像的自监督对比学习框架,引入关系对比损失及空间与频率增强策略,在多个乳腺超声数据集上显著提升分割性能与泛化能力,尤其在数据有限场景下表现突出。

摘要:

超声(US)成像因其无创且安全的特性在临床上具有重要价值。然而,解读超声图像具有挑战性,需要大量专业知识与时间,且容易出错。深度学习提供了分割等辅助解决方案。监督方法依赖大规模、高质量且标注一致的数据集,而这类数据集的构建十分困难。此外,这些方法在分布外数据上往往表现不佳,限制了其临床实用性。自监督学习(SSL)已成为一种有前景的替代方案,它利用无标注数据来提升模型性能和泛化能力。我们提出了一种针对B模式超声图像定制的对比自监督学习方法,其中引入了一种新颖的关系对比损失(Relation Contrastive Loss,RCL)。RCL通过可学习的度量来区分正负样本对,从而促使模型学习具有区分性的特征。此外,我们提出了基于空间和频率的增强策略,用于超声图像的表示学习。我们的方法在三个公开乳腺超声数据集上显著优于传统的监督分割方法,尤其是在数据有限的场景下。在Dice相似度指标上的显著改进包括:在BUSI数据集的20%和50%数据上分别提升4%,在BrEaST数据集的20%和50%数据上分别提升近6%和9%,在UDIAT数据集的20%和50%数据上分别提升6.4%和3.7%。此外,我们在分布外的UDIAT数据集上展示了更优的泛化能力,与使用20%和50%的BUSI和BrEaST训练数据的监督基线相比,性能分别提升了20.6%和13.6%。我们的研究表明,领域启发的自监督学习可以改善超声分割,尤其是在数据受限的条件下。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑