SCFI-ESeg: spatial and content feature integration for efficient semantic segmentation
摘要
Semantic segmentation plays a crucial role in various computer vision applications. However, improving the accuracy of existing models without significantly increasing computational cost remains a challenging task. Furthermore, current models often fall short in effectively utilizing spatial and contextual information. In this paper, we propose SCFI-ESeg, a competitive and efficient segmentation framework based on Transformer, which strengthens spatial information and aggregates multi-stage contextual information. We introduce a compact Multi-Stage Attention module and a Continuous Atrous Spatial Pyramid Pooling to enhance the integration of features at different levels and improve the model’s understanding of multi-scale information. Additionally, we present a Spatial Feature Enhancement Module, which leverages query features from the encoder to specifically enhance spatial information, thus improving the model’s ability to perceive fine image details. Experimental results demonstrate that SCFI-ESeg performs exceptionally well on public datasets, ADE20K and Pascal Context and Cityscapes, especially in complex scenarios. Compared to the baseline Segmenter, our framework achieves an average performance improvement of 1.6% on the ADE20K dataset, with a 2.5% improvement under the ViT-tiny configuration. Notably, this gain is obtained with a small absolute increase of 1.7M parameters and 0.6G GFLOPs. While the relative parameter growth is more noticeable for lightweight models such as ViT-tiny, the overall computational overhead remains modest, and the trade-off between accuracy and efficiency is favorable. Source code and models are available at Github: https://github.com/Eadeath/SCFI-ESeg