摘要
arXiv:2309.11876v3 通告类型: replace-cross
摘要:医学图像分割是一项基本但具有挑战性的任务,由于从专家那里获取大量高质量标注数据的过程非常艰辛。对比学习提供了一种有希望但仍然存在问题的解决方案。首先,现有的医学对比学习策略主要关注提取图像级别的表示,而忽视了丰富的多级别表示。此外,它们要么通过随机初始化,要么通过分开预训练编码器和解码器来未充分利用解码器,从而忽视了编码器和解码器之间的潜在合作。为了应对这些问题,我们提出了一种新型的多级别不对称对比学习框架MACL,用于体医学图像分割的预训练。具体而言,我们设计了一种不对称对比学习结构,同时对编码器和解码器进行预训练,以提供更好的分割模型初始化。此外,我们开发了一种多级别对比学习策略,将特征级别、图像级别和像素级别的对应关系整合在一起,以确保在预训练阶段,编码器和解码器能够从不同尺度和粒度的表示中捕获全面的细节。最后,实验结果表明,我们的MACL框架优于现有的11种对比学习策略。具体而言,我们的MACL在使用10%标注数据的情况下,在ACDC、MMWHS、HVSMR和CHAOS等8个医学图像数据集上,分别取得了ACDC、MMWHS、HVSMR和CHAOS数据集上比之前最好结果高出1.72%、7.87%、2.49%和1.48%的Dice分数,且从可视化图中可以看出有更好的精确预测结果。此外,我们的MACL在5种变体UNet骨干网络之间具有很强的泛化能力。我们的代码将在 https://github.com/stevezs315/MACL 释放。