论文标题
与嘈杂标签的稳健医学图像分割的联合类亲和力损失校正
Joint Class-Affinity Loss Correction for Robust Medical Image Segmentation with Noisy Labels
论文作者
论文摘要
以有限的注释成本收集的嘈杂标签可阻止医疗图像分割算法学习精确的语义相关性。以前使用嘈杂标签的学习的细分艺术仅执行以像素的方式来保存语义,例如像素标签校正,但忽略了配对的方式。实际上,我们观察到,捕获像素之间亲和力关系的成对方式可以大大降低标签噪声率。在这一观察结果的推动下,我们通过纳入像素和配对的举止,分别从嘈杂的阶级和亲和力标签中得出了监督,从而提出了缓解噪音的新观点。通过统一像素和配对的举止,我们提出了一个强大的联合类亲和力分段(JCAS)框架,以应对医学图像分割中的标签噪声问题。考虑到成对的亲和力结合了上下文依赖性,通过推理有关类内和类的亲和力关系的推理,设计了一个差异化的亲和力推理(DAR)模块来纠正像素段预测。为了进一步增强噪声阻力,旨在通过类和亲和力标签中建模的噪声标签分布来纠正监督信号的类亲和力损失校正(计算)策略。同时,CALC策略通过理论得出的一致性正则化来互动像素和成对的方式。合成和现实世界噪声标签下的广泛实验证实了所提出的JCAS框架的功效,并且对上限性能的最小间隙。源代码可在\ url {https://github.com/cityu-aim-group/jcas}中获得。
Noisy labels collected with limited annotation cost prevent medical image segmentation algorithms from learning precise semantic correlations. Previous segmentation arts of learning with noisy labels merely perform a pixel-wise manner to preserve semantics, such as pixel-wise label correction, but neglect the pair-wise manner. In fact, we observe that the pair-wise manner capturing affinity relations between pixels can greatly reduce the label noise rate. Motivated by this observation, we present a novel perspective for noisy mitigation by incorporating both pixel-wise and pair-wise manners, where supervisions are derived from noisy class and affinity labels, respectively. Unifying the pixel-wise and pair-wise manners, we propose a robust Joint Class-Affinity Segmentation (JCAS) framework to combat label noise issues in medical image segmentation. Considering the affinity in pair-wise manner incorporates contextual dependencies, a differentiated affinity reasoning (DAR) module is devised to rectify the pixel-wise segmentation prediction by reasoning about intra-class and inter-class affinity relations. To further enhance the noise resistance, a class-affinity loss correction (CALC) strategy is designed to correct supervision signals via the modeled noise label distributions in class and affinity labels. Meanwhile, CALC strategy interacts the pixel-wise and pair-wise manners through the theoretically derived consistency regularization. Extensive experiments under both synthetic and real-world noisy labels corroborate the efficacy of the proposed JCAS framework with a minimum gap towards the upper bound performance. The source code is available at \url{https://github.com/CityU-AIM-Group/JCAS}.