Miao Juzheng, Zhou Si-Ping, Zhou Guang-Quan, Wang Kai-Ni, Yang Meng, Zhou Shoujun, Chen Yang
IEEE Trans Med Imaging. 2024 Apr;43(4):1347-1364. doi: 10.1109/TMI.2023.3336534. Epub 2024 Apr 3.
Image segmentation achieves significant improvements with deep neural networks at the premise of a large scale of labeled training data, which is laborious to assure in medical image tasks. Recently, semi-supervised learning (SSL) has shown great potential in medical image segmentation. However, the influence of the learning target quality for unlabeled data is usually neglected in these SSL methods. Therefore, this study proposes a novel self-correcting co-training scheme to learn a better target that is more similar to ground-truth labels from collaborative network outputs. Our work has three-fold highlights. First, we advance the learning target generation as a learning task, improving the learning confidence for unannotated data with a self-correcting module. Second, we impose a structure constraint to encourage the shape similarity further between the improved learning target and the collaborative network outputs. Finally, we propose an innovative pixel-wise contrastive learning loss to boost the representation capacity under the guidance of an improved learning target, thus exploring unlabeled data more efficiently with the awareness of semantic context. We have extensively evaluated our method with the state-of-the-art semi-supervised approaches on four public-available datasets, including the ACDC dataset, M&Ms dataset, Pancreas-CT dataset, and Task_07 CT dataset. The experimental results with different labeled-data ratios show our proposed method's superiority over other existing methods, demonstrating its effectiveness in semi-supervised medical image segmentation.
在大规模标注训练数据的前提下,图像分割通过深度神经网络取得了显著进展,而在医学图像任务中确保大规模标注训练数据非常费力。最近,半监督学习(SSL)在医学图像分割中显示出巨大潜力。然而,在这些半监督学习方法中,通常忽略了未标注数据的学习目标质量的影响。因此,本研究提出了一种新颖的自校正协同训练方案,以从协作网络输出中学习更接近真实标签的更好目标。我们的工作有三个突出亮点。首先,我们将学习目标生成提升为一项学习任务,通过自校正模块提高对未标注数据的学习信心。其次,我们施加结构约束,以进一步鼓励改进后的学习目标与协作网络输出之间的形状相似性。最后,我们提出了一种创新的逐像素对比学习损失,以在改进后的学习目标的指导下提升表示能力,从而在语义上下文感知的情况下更有效地探索未标注数据。我们使用包括ACDC数据集、M&Ms数据集、胰腺CT数据集和Task_07 CT数据集在内的四个公开可用数据集,用最先进的半监督方法对我们的方法进行了广泛评估。不同标注数据比例的实验结果表明,我们提出的方法优于其他现有方法,证明了其在半监督医学图像分割中的有效性。