arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2305.00673cs.CV

面向半监督医学图像分割的双向复制粘贴方法

Bidirectional Copy-Paste for Semi-Supervised Medical Image Segmentation

East China Normal University(华东师范大学);Shanghai Jiao Tong University(上海交通大学)

Yunhao Bai, Duowen Chen, Qingli Li, Wei Shen, Yan Wang

AI总结:

针对半监督医学图像分割中标记与未标记数据分布失配的问题,提出基于Mean Teacher架构的双向复制粘贴方法,通过双向混合数据与一致监督缩小分布差距,在多个数据集上取得显著性能提升。

摘要:

在半监督医学图像分割任务中,标记数据与未标记数据的分布存在经验性失配问题。若将标记数据与未标记数据分开处理或以不一致的方式处理,从标记数据中学到的知识可能会被大量丢弃。我们提出了一种缓解该问题的直接方法——在简易的Mean Teacher(均值教师)架构中,对标记数据和未标记数据进行双向复制粘贴。该方法促使未标记数据从内、外两个方向从标记数据中学习全面的共同语义。更重要的是,标记数据与未标记数据的一致学习流程可大幅缩小经验分布差距。具体而言,我们分别将标记图像(前景)的随机裁剪块复制粘贴到未标记图像(背景)上,以及将未标记图像(前景)的随机裁剪块复制粘贴到标记图像(背景)上。这两幅混合图像被输入Student(学生)网络,并由伪标签与真实标签的混合监督信号进行监督。我们发现,标记数据与未标记数据之间的双向复制粘贴这一简单机制已足够有效,实验表明,在多个半监督医学图像分割数据集上,该方法相比其他现有最优方法均有稳定提升(例如,在标记数据占比为5%的ACDC数据集上,Dice指标提升超过21%)。代码可在https://github.com/DeepMed-Lab-ECNU/BCP获取。

英文摘要:

In semi-supervised medical image segmentation, there exist empirical mismatch problems between labeled and unlabeled data distribution. The knowledge learned from the labeled data may be largely discarded if treating labeled and unlabeled data separately or in an inconsistent manner. We propose a straightforward method for alleviating the problem - copy-pasting labeled and unlabeled data bidirectionally, in a simple Mean Teacher architecture. The method encourages unlabeled data to learn comprehensive common semantics from the labeled data in both inward and outward directions. More importantly, the consistent learning procedure for labeled and unlabeled data can largely reduce the empirical distribution gap. In detail, we copy-paste a random crop from a labeled image (foreground) onto an unlabeled image (background) and an unlabeled image (foreground) onto a labeled image (background), respectively. The two mixed images are fed into a Student network and supervised by the mixed supervisory signals of pseudo-labels and ground-truth. We reveal that the simple mechanism of copy-pasting bidirectionally between labeled and unlabeled data is good enough and the experiments show solid gains (e.g., over 21% Dice improvement on ACDC dataset with 5% labeled data) compared with other state-of-the-arts on various semi-supervised medical image segmentation datasets. Code is available at https://github.com/DeepMed-Lab-ECNU/BCP}.

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

↑