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

无标记数据概率分布建模forOne-shot医学图像分割

Modeling the Probabilistic Distribution of Unlabeled Data forOne-shot Medical Image Segmentation

Baidu Research(百度研究院);University of Technology Sydney(悉尼科技大学)

Yuhang Ding, Xin Yu, Yi Yang

摘要:

现有的图像分割网络主要利用大规模标记数据集来获得高精度。然而,标记医学图像是非常昂贵的,因为它需要复杂的专业知识。因此,在追求高分割性能时,更希望仅使用少量标记数据。在本文中,我们开发了一种数据增强方法,用于单次脑磁共振成像(MRI)图像分割,它只利用一个标记的MRI图像(命名为图集)和一些未标记的图像。特别是,我们建议通过3D变分自编码器(VAE)来学习不同未标记MRI图像相对于图谱的变形(包括形状和强度)的概率分布。通过这种方式,我们的方法能够利用学习到的图像变形分布来生成新的真实的大脑MRI图像,并且生成的样本的数量将足以训练深度分割网络。此外,我们引入了一个新的标准分割基准,通过交叉数据集设置(从不同来源收集)来评估分割网络的泛化性能。大量的实验表明,我们的方法优于国家的最先进的一次性医疗分割方法。我们的代码已在https://github.com/dyh127/Modeling-the-Probabilistic-Distribution-of-Unlabeled-Data上发布。

英文摘要:

Existing image segmentation networks mainly leverage large-scale labeled datasets to attain high accuracy. However, labeling medical images is very expensive since it requires sophisticated expert knowledge. Thus, it is more desirable to employ only a few labeled data in pursuing high segmentation performance. In this paper, we develop a data augmentation method for one-shot brain magnetic resonance imaging (MRI) image segmentation which exploits only one labeled MRI image (named atlas) and a few unlabeled images. In particular, we propose to learn the probability distributions of deformations (including shapes and intensities) of different unlabeled MRI images with respect to the atlas via 3D variational autoencoders (VAEs). In this manner, our method is able to exploit the learned distributions of image deformations to generate new authentic brain MRI images, and the number of generated samples will be sufficient to train a deep segmentation network. Furthermore, we introduce a new standard segmentation benchmark to evaluate the generalization performance of a segmentation network through a cross-dataset setting (collected from different sources). Extensive experiments demonstrate that our method outperforms the state-of-the-art one-shot medical segmentation methods. Our code has been released at https://github.com/dyh127/Modeling-the-Probabilistic-Distribution-of-Unlabeled-Data.

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

↑