LLM2D
自我监督的框架以提高超声B模式图像分割的泛化能力
A Self-Supervised Framework for Improved Generalisability in Ultrasound B-mode Image Segmentation
作者: Edward Ellis, Andrew Bulpitt, Nasim Parsa, Michael F Byrne, Sharib Ali
发布日期: 2/5/2025
arXiv ID: 2502.02489

摘要

arXiv:2502.02489v1 交叉类型: cross 摘要:超声(US)成像由于其非侵入性和安全性,在临床中非常有价值。然而,解释US图像具有挑战性,需要显著的专业知识和时间,并且经常会出现错误。深度学习提供了诸如分割之类的辅助解决方案。监督方法依赖于大型、高质量且一致性标注的数据集,而这些数据集的收集非常具有挑战性。此外,这些方法在处理分布外数据时往往会表现不佳,限制了它们在临床中的实用性。自监督学习(SSL)作为一种有前途的替代方案已崭露头角,通过利用未标注的数据来增强模型的性能和泛化能力。我们介绍了一种针对B模式US图像的对比SSL方法,其中包含一种新颖的Relational Contrastive Loss (RCL)。RCL通过可学习度量来区分正样本和负样本对,鼓励学习不同的特征。此外,我们提出了空间和频率基的增强策略,用于US图像的表示学习。我们的方法在三个公开的乳腺US数据集上显著优于传统监督分割方法,特别是在数据受限的情况下。在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%。我们的研究强调,领域启发的SSL可以改善US分割,尤其是在数据受限的条件下。