arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2304.04441cs.CVcs.LG

面向半监督医学图像分割的双不确定性自训练方法

Self-training with dual uncertainty for semi-supervised medical image segmentation

Hubei University of Technology(湖北工业大学)

Zhanhong Qiu, Haitao Gan, Ming Shi, Zhongwei Huang, Zhi Yang

AI总结:

针对半监督医学图像分割标注数据不足的问题,提出融合样本级与像素级双不确定性的自训练方法,在两个公开数据集上均优于现有5种半监督方法,具备有效性与鲁棒性。

摘要:

在半监督医学图像分割领域,标注数据短缺是核心问题。如何有效从无标注图像中学习图像特征以提升分割精度,是该领域的主要研究方向。传统自训练方法通过生成伪标签进行迭代训练,可部分解决标注数据不足的问题,但训练过程中模型不确定性产生的噪声会直接影响分割结果。为此,我们在自训练框架基础上,加入样本级与像素级不确定性以稳定训练过程。具体而言,我们保存预训练过程中模型的多个时刻状态,利用它们对无标注样本预测结果的差异作为该样本的样本级不确定性估计,随后在训练过程中按从易到难的顺序逐步加入无标注样本。同时,我们为分割网络添加一个采用不同上采样方式的解码器,将两个解码器输出的差异作为像素级不确定性。简言之,我们对无标注样本进行选择性重训练,并为伪标签赋予像素级不确定性,以此优化自训练过程。我们在公开的2017 ACDC数据集和2018 Prostate数据集上,将所提模型的分割结果与5种半监督方法进行了对比。在相同设置下,本文方法在两个数据集上均取得了更优的分割性能,证明了其有效性、鲁棒性,以及迁移到其他医学图像分割任务的潜力。

英文摘要:

In the field of semi-supervised medical image segmentation, the shortage of labeled data is the fundamental problem. How to effectively learn image features from unlabeled images to improve segmentation accuracy is the main research direction in this field. Traditional self-training methods can partially solve the problem of insufficient labeled data by generating pseudo labels for iterative training. However, noise generated due to the model's uncertainty during training directly affects the segmentation results. Therefore, we added sample-level and pixel-level uncertainty to stabilize the training process based on the self-training framework. Specifically, we saved several moments of the model during pre-training, and used the difference between their predictions on unlabeled samples as the sample-level uncertainty estimate for that sample. Then, we gradually add unlabeled samples from easy to hard during training. At the same time, we added a decoder with different upsampling methods to the segmentation network and used the difference between the outputs of the two decoders as pixel-level uncertainty. In short, we selectively retrained unlabeled samples and assigned pixel-level uncertainty to pseudo labels to optimize the self-training process. We compared the segmentation results of our model with five semi-supervised approaches on the public 2017 ACDC dataset and 2018 Prostate dataset. Our proposed method achieves better segmentation performance on both datasets under the same settings, demonstrating its effectiveness, robustness, and potential transferability to other medical image segmentation tasks. Keywords: Medical image segmentation, semi-supervised learning, self-training, uncertainty estimation

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

↑