Adaptive Feature Recombination and Recalibration for Semantic Segmentation With Fully Convolutional Networks

被引:62
作者
Pereira, Sergio [1 ,2 ]
Pinto, Adriano [1 ,2 ]
Amorim, Joana [1 ]
Ribeiro, Alexandrine [1 ]
Alves, Victor [2 ]
Silva, Carlos A. [1 ]
机构
[1] Univ Minho, CMEMS UMinho Res Unit, P-4800058 Guimaraes, Portugal
[2] Univ Minho, Ctr Algoritmi, P-4710057 Braga, Portugal
关键词
Segmentation; deep learning; fully convolutional network; recalibration; recombination; adaptive; BRAIN-TUMOR SEGMENTATION; MODEL;
D O I
10.1109/TMI.2019.2918096
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Fully convolutional networks have been achieving remarkable results in image semantic segmentation, while being efficient. Such efficiency results from the capability of segmenting several voxels in a single forward pass. So, there is a direct spatial correspondence between a unit in a feature map and the voxel in the same location. In a convolutional layer, the kernel spans over all channels and extracts information from them. We observe that linear recombination of feature maps by increasing the number of channels followed by compression may enhance their discriminative power. Moreover, not all feature maps have the same relevance for the classes being predicted. In order to learn the inter-channel relationships and recalibrate the channels to suppress the less relevant ones, squeeze and excitation blocks were proposed in the context of image classification with convolutional neural networks. However, this is not well adapted for segmentation with fully convolutional networks since they segment several objects simultaneously, hence a feature map may contain relevant information only in some locations. In this paper, we propose recombination of features and a spatially adaptive recalibration block that is adapted for semantic segmentation with fully convolutional networks- the SegSE block. Feature maps are recalibrated by considering the cross-channel information together with spatial relevance. The experimental results indicate that recombination and recalibration improve the results of a competitive baseline, and generalize across three different problems: brain tumor segmentation, stroke penumbra estimation, and ischemic stroke lesion outcome prediction. The obtained results are competitive or outperform the state of the art in the three applications.
引用
收藏
页码:2914 / 2925
页数:12
相关论文
共 43 条
  • [1] [Anonymous], P 3 INT C LEARNING R
  • [2] [Anonymous], P INT MICCAI BRAINL
  • [3] [Anonymous], P INT MICCAI BRAINL
  • [4] [Anonymous], 2016, DEEP LEARNING
  • [5] [Anonymous], 2015, PROC CVPR IEEE
  • [6] [Anonymous], P IEEE C COMP VIS PA
  • [7] [Anonymous], ADV NEURAL INFORM PR
  • [8] [Anonymous], 2017, INT MICCAI BRAINL WO
  • [9] [Anonymous], 2014, P 2 INT C LEARNING R
  • [10] [Anonymous], 1999, ADV COURSE ARTIFICIA