Rectified Contrastive Pseudo Supervision用于半监督医学图像分割的校正对比伪监督

摘要

医学图像分割方法通常设计为全监督以保证模型性能，这需要大量的专家注释样本，成本高且费力。半监督图像分割可以通过利用大量未标记图像和有限标记图像来缓解该问题。然而，由于伪标签中的潜在噪声和特征空间中的类可分离性不足，从大量未标记图像中学习鲁棒表示仍然具有挑战性，这破坏了当前半监督分割方法的性能。为了解决上述问题，我们提出了一种新的半监督分割方法，称为校正对比伪监督（RCPS），它结合了校正伪监督和体素级对比学习来提高半监督分割的有效性。特别是，我们为基于不确定性估计和一致性正则化的伪监督方法设计了一种新的整改策略，以减少伪标签中的噪声影响。此外，我们向网络引入了双向体素对比损失，以确保特征空间中的类内一致性和类间对比度，从而增加了分割中的类可分离性。所提出的 RCPS 分割方法已在两个公共数据集和一个内部临床数据集上得到验证。实验结果表明，与半监督医学图像分割中的最新方法相比，所提出的方法具有更好的分割性能。源代码可在 https://github.com/hsiangyuzhao/RCPS 获取。
索引词—医学图像分割，半监督学习，对比学习，伪监督

这项工作的贡献包括三个方面：

• 我们引入校正伪监督来学习图像空间中不同外观下的稳健表示。引入了预测不确定性估计和一致性正则化两个整改，以提高伪监督的有效性。
• 我们提出了一种双向体素对比损失和一种新颖的置信负采样策略，以提高特征空间中不同语义类之间的可分离性。我们鼓励两个增强视图应该相似，并且两个视图应该远离特征空间中的负样本。
• 我们在两个公共基准数据集上验证了提议的 RCPS，包括 MRI 中的左心房腔分割 [9] 和 CT 中的胰腺分割 [10]。实验结果表明，所提出的 RCPS 在两个基准测试中都优于其他最先进的方法。我们还包括一个内部临床数据集，用于对创伤性脑损伤 (TBI) 患者进行大脑分割，与我们之前的作品相比，实现了最先进的分割性能。