论文标题
使用歧视性嵌入的全局相关网络几乎没有射击的医疗图像分割
Few-shot Medical Image Segmentation using a Global Correlation Network with Discriminative Embedding
论文作者
论文摘要
尽管深度卷积神经网络在医学图像计算和分析方面取得了令人印象深刻的进展,但其监督学习的范式仍需要大量注释以避免过度拟合并取得了令人鼓舞的结果。在临床实践中,在需要专门的生物医学专家知识的某些条件下,很难获得大量的语义注释,而且它也是只有少数注释类别的常见条件。在这项工作中,我们提出了一种新的方法,用于几次医学图像分割,该方法使分割模型能够快速概括为看不见的类,而训练图像很少。我们使用经过情节训练的深度卷积网络构建了几张图像分段。在医学图像中的空间一致性和规律性的启发下,我们开发了一个有效的全局相关模块,以捕获支持和查询图像之间的相关性,并将其整合到称为“全局相关网络”的深网络中。此外,我们增强了深层嵌入的可区分性,以鼓励同一类的特征域聚类,同时使不同器官的特征域保持较远。消融研究证明了拟议的全球相关模块和歧视性嵌入损失的有效性。对CT和MRI模式的解剖学腹部图像进行了广泛的实验,以证明我们提出的模型的最新性能。
Despite deep convolutional neural networks achieved impressive progress in medical image computing and analysis, its paradigm of supervised learning demands a large number of annotations for training to avoid overfitting and achieving promising results. In clinical practices, massive semantic annotations are difficult to acquire in some conditions where specialized biomedical expert knowledge is required, and it is also a common condition where only few annotated classes are available. In this work, we proposed a novel method for few-shot medical image segmentation, which enables a segmentation model to fast generalize to an unseen class with few training images. We construct our few-shot image segmentor using a deep convolutional network trained episodically. Motivated by the spatial consistency and regularity in medical images, we developed an efficient global correlation module to capture the correlation between a support and query image and incorporate it into the deep network called global correlation network. Moreover, we enhance discriminability of deep embedding to encourage clustering of the feature domains of the same class while keep the feature domains of different organs far apart. Ablation Study proved the effectiveness of the proposed global correlation module and discriminative embedding loss. Extensive experiments on anatomical abdomen images on both CT and MRI modalities are performed to demonstrate the state-of-the-art performance of our proposed model.