arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.09854eess.IVcs.CV
收藏

自环不确定性:一种新的半监督医学图像分割伪标记

Self-Loop Uncertainty: A Novel Pseudo-Label for Semi-Supervised Medical Image Segmentation

Tencent Jarvis Lab(腾讯嘉里实验室)

Yuexiang Li, Jiawei Chen, Xinpeng Xie, Kai Ma, Yefeng Zheng

摘要:

随着深度学习神经网络在自然图像处理中的成功,越来越多的研究提出了开发基于深度学习的医学图像分割框架。然而,由于医学图像的逐像素注释是费力且昂贵的,因此注释数据的量通常不足以很好地训练神经网络。本文提出了一种半监督的方法来训练具有有限标记数据和大量未标记图像的神经网络用于医学图像分割。一种新的伪标签(即自循环不确定性),通过循环优化的神经网络与自我监督的任务,被采用作为地面真理的未标记的图像,以增加训练集,提高分割精度。提出的自循环不确定性可以被视为通过集成多个模型产生的不确定性估计的近似,从而显着减少推理时间。在两个公开数据集上的实验结果证明了半监督方法的有效性。

英文摘要:

Witnessing the success of deep learning neural networks in natural image processing, an increasing number of studies have been proposed to develop deep-learning-based frameworks for medical image segmentation. However, since the pixel-wise annotation of medical images is laborious and expensive, the amount of annotated data is usually deficient to well-train a neural network. In this paper, we propose a semi-supervised approach to train neural networks with limited labeled data and a large quantity of unlabeled images for medical image segmentation. A novel pseudo-label (namely self-loop uncertainty), generated by recurrently optimizing the neural network with a self-supervised task, is adopted as the ground-truth for the unlabeled images to augment the training set and boost the segmentation accuracy. The proposed self-loop uncertainty can be seen as an approximation of the uncertainty estimation yielded by ensembling multiple models with a significant reduction of inference time. Experimental results on two publicly available datasets demonstrate the effectiveness of our semi-supervied approach.

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

↑