An Attention Enhanced Graph Convolutional Network for Semantic Segmentation

Ao Chen,Yue Zhou
DOI: https://doi.org/10.1007/978-3-030-60633-6_61
2020-01-01
Abstract:Global modeling and integrating over feature relations are beneficial for semantic segmentation. Previous researches excel at modeling features through convolution operations but are inefficient at associating features between distant regions. Consequently, key information within one semantic category can be neglected if the features between distant regions are not bridged. To solve this issue, we propose a novel Attention Enhanced Graph Convolutional Network to explore relational information and preserve details in semantic segmentation tasks. We first build a graph over feature maps, where each node is represented by features in spatial and channel dimensions. Then the interdependent semantic information is obtained through applying the GCN to globally bridge co-occurrence features regardless of their distance. Finally, we introduce the self-attention mechanism in spatial and channel dimensions respectively to project the semantic-aware information into feature maps so as to capture the discriminative information. We examine the merit of our model on three challenging scene segmentation datasets: Cityscapes, PASCAL VOC2012 and COCO Stuff. Experimental results show that our model boosts the performance over other state-of-the-arts. Code will be available.
What problem does this paper attempt to address?