通过学生模型差异发现偏差,自动修正医学图像分割中的错误伪标签。
SDCL: Students Discrepancy-Informed Correction Learning for Semi-supervised Medical Image Segmentation
- 用两个学生模型的分割差异识别潜在偏差区域
- 在偏差区通过距离最小化和熵最大化提升分割精度
- 在3个数据集上超越最新方法,接近甚至超过全监督效果
半监督医学图像分割(SSMIS)有望缓解标注数据不足的问题。然而,由于错误的伪标签,当前基于教师-学生框架的方法易受确认偏见和认知偏差影响。为此,本文改进了均值教师方法,提出学生差异感知修正学习(SDCL)框架,包含两个学生与一个不可训练的教师。该框架利用两个学生间的分割差异来引导自修正学习。其核心思想是将分割差异区域识别为潜在偏差区域,并在此类区域鼓励模型重新审视并纠正自身认知偏差。为实现持续的审查与修正,引入两种修正损失函数:最小化正确分割体素的距离,最大化错误分割体素的熵。在三个公开医学图像数据集(两个3D数据集:CT、MRI,一个2D数据集:MRI)上进行实验。结果表明,本方法在胰腺(Pancreas)、心脏左室(LA)和ACDC数据集上的Dice分数分别优于当前最先进方法2.57%、3.04%和2.34%。此外,在ACDC数据集上性能接近全监督方法,而在胰腺和LA数据集上甚至超过全监督方法。(代码见:https://github.com/pascalcpp/SDCL)
原文摘要 · Abstract (English)
Semi-supervised medical image segmentation (SSMIS) has been demonstrated the potential to mitigate the issue of limited medical labeled data. However, confirmation and cognitive biases may affect the prevalent teacher-student based SSMIS methods due to erroneous pseudo-labels. To tackle this challenge, we improve the mean teacher approach and propose the Students Discrepancy-Informed Correction Learning (SDCL) framework that includes two students and one non-trainable teacher, which utilizes the segmentation difference between the two students to guide the self-correcting learning. The essence of SDCL is to identify the areas of segmentation discrepancy as the potential bias areas, and then encourage the model to review the correct cognition and rectify their own biases in these areas. To facilitate the bias correction learning with continuous review and rectification, two correction loss functions are employed to minimize the correct segmentation voxel distance and maximize the erroneous segmentation voxel entropy. We conducted experiments on three public medical image datasets: two 3D datasets (CT and MRI) and one 2D dataset (MRI). The results show that our SDCL surpasses the current State-of-the-Art (SOTA) methods by 2.57\%, 3.04\%, and 2.34\% in the Dice score on the Pancreas, LA, and ACDC datasets, respectively. In addition, the accuracy of our method is very close to the fully supervised method on the ACDC dataset, and even exceeds the fully supervised method on the Pancreas and LA dataset. (Code available at \url{https://github.com/pascalcpp/SDCL}).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。