摘要
arXiv:2504.14737v1 宣布类型: 交叉
摘要: 医学图像分割是一项关键但具有挑战性的任务,主要是由于难以获得大量高质量、专家注释的图像数据集。对比学习为解决这一问题提供了潜在但仍然存在一些问题的解决方案。由于现有的大多数方法集中在提取实例级或像素到像素的表示,而忽略了图像内相似像素组之间的特征。此外,在考虑对比学习对的生成时,大多数最先进的方法主要依赖于人工设定阈值,这需要大量的梯度实验,并且效率低下且缺乏通用性。为了解决这些问题,我们提出了一种名为 SuperCL 的新颖的对比学习方法,用于医学图像分割预训练。具体而言,我们的 SuperCL 通过引入两种新颖的对比学习对生成策略:图像内局部对比学习对 (ILCP) 生成和图像间全局对比学习对 (IGCP) 生成,利用图像的结构先验和像素相关性。考虑到超像素集群很好地符合对比学习对生成的概念,我们利用超像素图来生成用于 ILCP 和 IGCP 的伪掩码,以指导监督对比学习。此外,我们还提出了两个模块:平均超像素特征图生成 (ASP) 和连接组件标签生成 (CCL),以更好地利用先验结构信息来辅助 IGCP。最后,在 8 个医学图像数据集上的实验表明,我们的 SuperCL 在性能上优于现有的 12 种方法。具体来说,我们的 SuperCL 在可视化图中的预测更为精准,在 MMWHS、CHAOS 和 Spleen(10% 注释)数据集上,Dice 系数分别高出前最佳结果 3.15%、5.44% 和 7.89%。我们的代码将在接收后发布。