arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2301.04465cs.CV

基于高置信度伪标签协同训练的半监督医学图像分割

Co-training with High-Confidence Pseudo Labels for Semi-supervised Medical Image Segmentation

Northeastern University(东北大学);Fujian Normal University(福建师范大学);DAMO Academy, Alibaba Group(阿里巴巴集团达摩院);University of Alberta(阿尔伯塔大学);Alberta Machine Intelligence Institute(阿尔伯塔机器智能研究所)

Zhiqiang Shen, Peng Cao, Hua Yang, Xiaoli Liu, Jinzhu Yang, Osmar R. Zaiane

AI总结:

针对半监督医学图像分割中协同训练过早收敛和伪标签置信度低的问题,本文提出 UCMT 模型,通过协同均值教师和不确定性引导区域混合保持模型分歧并提升伪标签质量,在四个公开数据集上取得最优性能。

摘要:

一致性正则化和基于伪标记的半监督方法利用多视图输入的伪标签进行协同训练。然而,这类协同训练模型往往过早收敛到一致状态,退化为自训练模型,并在训练过程中从扰动输入中产生低置信度的伪标签。为解决这些问题,我们提出了一种不确定性引导的协同均值教师模型(UCMT),用于具有高置信度伪标签的半监督语义分割。具体而言,UCMT 包含两个主要组件:1)协同均值教师(CMT),用于鼓励模型分歧并在子网络之间进行协同训练;2)不确定性引导的区域混合(UMIX),根据 CMT 的不确定性图操纵输入图像,并促使 CMT 生成高置信度的伪标签。将 UMIX 与 CMT 的优势相结合,UCMT 能够保持模型分歧并提高协同训练分割中伪标签的质量。在包括 2D 和 3D 模态的四个公开医学图像数据集上的大量实验表明,UCMT 优于当前最先进的方法。代码可在 https://github.com/Senyh/UCMT 获取。

英文摘要:

Consistency regularization and pseudo labeling-based semi-supervised methods perform co-training using the pseudo labels from multi-view inputs. However, such co-training models tend to converge early to a consensus, degenerating to the self-training ones, and produce low-confidence pseudo labels from the perturbed inputs during training. To address these issues, we propose an Uncertainty-guided Collaborative Mean-Teacher (UCMT) for semi-supervised semantic segmentation with the high-confidence pseudo labels. Concretely, UCMT consists of two main components: 1) collaborative mean-teacher (CMT) for encouraging model disagreement and performing co-training between the sub-networks, and 2) uncertainty-guided region mix (UMIX) for manipulating the input images according to the uncertainty maps of CMT and facilitating CMT to produce high-confidence pseudo labels. Combining the strengths of UMIX with CMT, UCMT can retain model disagreement and enhance the quality of pseudo labels for the co-training segmentation. Extensive experiments on four public medical image datasets including 2D and 3D modalities demonstrate the superiority of UCMT over the state-of-the-art. Code is available at: https://github.com/Senyh/UCMT.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑