BAGNet: Branch Attention Guided Decoder for Semantic Segmentation

Zhengtan Wang,Yuan Zheng
DOI: https://doi.org/10.1145/3577530.3577547
2022-01-01
Abstract:The effectiveness of encoder-decoder structures for semantic segmentation has been demonstrated. Given the importance of decoders, we propose a branch attention guided decoder for semantic segmentation, called BAGNet. The proposed decoder exploits the multi-scale feature maps generated by the backbone and adaptively selects the important feature information. Attention mechanism and multipath representation are important for semantic segmentation. The existing methods mostly introduce the attention mechanisms into the encoder, while we observe that decoder also need the attention module to select important information when using lateral connections. To solve this problem, we present a Branch Attention Unit(BAU), which applies the channel-wise attention on different branches to reduce the semantic gap and leverage their success in capturing cross-feature interactions and learning diverse representations to select important information in branches. To validate the performance of the proposed BAU and BAGNet, we conduct extensive experiments and achieve consistently improved performances on popular segmentation benchmarks, namely 80.33%, 42.30% and 44.66% mIoUs on Cityscapes, ADE20K and Pascal Context validation sets with ResNet-50 backbone, respectively. Furthermore, we verify their generalization ability using different baseline models.
What problem does this paper attempt to address?