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: oai:arXiv.org:2502.02489v1

摘要

arXiv:2502.02489v1 声波类型: cross 摘要:超声波(US)成像因其无创性和安全性而在临床中极为宝贵。然而,解读US图像具有挑战性,需要大量的专业知识和时间,并且经常会出现错误。深度学习提供了诸如分割之类的辅助解决方案。监督方法依赖于大量高质量且一致标注的训练数据集,这在收集上具有挑战性。此外,这些方法在处理分布外数据时往往会表现不佳,限制了其在临床中的实用性。自我监督学习(SSL)已作为一种有前途的替代方案出现,它利用未标注的数据来增强模型的性能和泛化能力。我们介绍了一种针对B模式US图像的对比SSL方法,结合了一种新颖的关系对比损失(RCL)。RCL通过可学习的度量将正样本和负样本对区分开来,从而鼓励学习不同的特征。此外,我们还提出了空间和频率基础的增强策略,用于US图像的表示学习。我们的方法在三个公开的乳腺US数据集中相比传统监督分割方法取得了显著的性能提升,尤其是在数据受限的情况下。在Dice相似度指标上,我们的方法在BUSI数据集的20%和50%部分分别提高了4%和5%,在BrEaST数据集的20%和50%部分分别提高了近6%和9%,在UDIAT数据集的20%和50%部分分别提高了6.4%和3.7%。此外,我们展示了在分布外UDIAT数据集上的更强泛化能力,使用20%和50%的BUSI和BrEaST训练数据,相较于监督基线分别提升了20.6%和13.6%。我们的研究强调了领域启发式SSL可以在数据受限条件下提高US分割的性能。