GSANet: Semantic Segmentation with Global and Selective Attention
This work addresses the problem of accurate semantic segmentation for computer vision applications, offering incremental improvements with new attention modules.
The paper tackles semantic segmentation by proposing GSANet, a novel architecture that integrates global and selective attention mechanisms, achieving state-of-the-art accuracy improvements on ADE20k and Cityscapes datasets.
This paper proposes a novel deep learning architecture for semantic segmentation. The proposed Global and Selective Attention Network (GSANet) features Atrous Spatial Pyramid Pooling (ASPP) with a novel sparsemax global attention and a novel selective attention that deploys a condensation and diffusion mechanism to aggregate the multi-scale contextual information from the extracted deep features. A selective attention decoder is also proposed to process the GSA-ASPP outputs for optimizing the softmax volume. We are the first to benchmark the performance of semantic segmentation networks with the low-complexity feature extraction network (FXN) MobileNetEdge, that is optimized for low latency on edge devices. We show that GSANet can result in more accurate segmentation with MobileNetEdge, as well as with strong FXNs, such as Xception. GSANet improves the state-of-art semantic segmentation accuracy on both the ADE20k and the Cityscapes datasets.