School of Mathematics and Statistics, Xi'an Jiaotong University, Xi'an, 710049, China.
Jarvis Lab, Tencent, Shenzhen, 518075, China.
Med Image Anal. 2021 Jan;67:101876. doi: 10.1016/j.media.2020.101876. Epub 2020 Oct 17.
Fully convolutional networks (FCNs) trained with abundant labeled data have been proven to be a powerful and efficient solution for medical image segmentation. However, FCNs often fail to achieve satisfactory results due to the lack of labelled data and significant variability of appearance in medical imaging. To address this challenging issue, this paper proposes a conjugate fully convolutional network (CFCN) where pairwise samples are input for capturing a rich context representation and guide each other with a fusion module. To avoid the overfitting problem introduced by intra-class heterogeneity and boundary ambiguity with a small number of training samples, we propose to explicitly exploit the prior information from the label space, termed as proxy supervision. We further extend the CFCN to a compact conjugate fully convolutional network (CFCN), which just has one head for fitting the proxy supervision without incurring two additional branches of decoders fitting ground truth of the input pairs compared to CFCN. In the test phase, the segmentation probability is inferred by the learned logical relation implied in the proxy supervision. Quantitative evaluation on the Liver Tumor Segmentation (LiTS) and Combined (CT-MR) Healthy Abdominal Organ Segmentation (CHAOS) datasets shows that the proposed framework achieves a significant performance improvement on both binary segmentation and multi-category segmentation, especially with a limited amount of training data. The source code is available at https://github.com/renzhenwang/pairwise_segmentation.
基于大量标注数据训练的全卷积网络(FCN)已被证明是一种强大而有效的医学图像分割方法。然而,由于缺乏标注数据以及医学成像中外观的显著可变性,FCN 常常无法取得令人满意的效果。为了解决这个具有挑战性的问题,本文提出了一种共轭全卷积网络(CFCN),它可以输入成对的样本,以捕捉丰富的上下文表示,并通过融合模块相互指导。为了避免由于训练样本数量少而导致的类内异质性和边界模糊引起的过拟合问题,我们提出了明确利用标签空间的先验信息,即代理监督。我们进一步将 CFCN 扩展到一个紧凑的共轭全卷积网络(CFCN),它只使用一个头部来拟合代理监督,而不需要与 CFCN 相比,为输入对的真实值拟合额外的两个解码器分支。在测试阶段,通过学习代理监督中隐含的逻辑关系来推断分割概率。在 Liver Tumor Segmentation (LiTS) 和 Combined (CT-MR) Healthy Abdominal Organ Segmentation (CHAOS) 数据集上的定量评估表明,所提出的框架在二进制分割和多类别分割方面都取得了显著的性能提升,尤其是在训练数据有限的情况下。代码可在 https://github.com/renzhenwang/pairwise_segmentation 上获取。