用状态空间模型与局部注意力融合,实现高效多尺度语义分割
SegMAN: Omni-scale Context Modeling with State Space Models and Local Attention for Semantic Segmentation
- 混合滑动局部注意力与动态状态空间模型,兼顾全局上下文与细节保留
- 在ADE20K上达52.6% mIoU,比SegNeXt-L高1.6%且减少15%计算量
- 适合需要高效高精度分割的场景,如自动驾驶与医学图像分析
高质量语义分割依赖三个核心能力:全局上下文建模、局部细节编码和多尺度特征提取。然而,现有方法难以同时具备这三项能力。为此,我们提出SegMAN,一种线性时间复杂度的新模型,包含名为SegMAN Encoder的混合特征编码器和基于状态空间模型的解码器。SegMAN Encoder融合滑动局部注意力与动态状态空间模型,实现高效全局上下文建模并保持精细局部细节。解码器中的MMSCopE模块增强多尺度上下文特征提取,并随输入分辨率自适应调整。SegMAN-B Encoder在ImageNet-1k上达到85.1%准确率(较VMamba-S提升1.5%,参数更少)。搭配解码器后,完整模型在ADE20K上达52.6% mIoU(较SegNeXt-L高1.6%,节省15% GFLOPs),Cityscapes上为83.8% mIoU(较SegFormer-B3高2.1%,计算量减半),COCO-Stuff上优于VWFormer-B3 1.6%且更低计算开销。
原文摘要 · Abstract (English)
High-quality semantic segmentation relies on three key capabilities: global context modeling, local detail encoding, and multi-scale feature extraction. However, recent methods struggle to possess all these capabilities simultaneously. Hence, we aim to empower segmentation networks to simultaneously carry out efficient global context modeling, high-quality local detail encoding, and rich multi-scale feature representation for varying input resolutions. In this paper, we introduce SegMAN, a novel linear-time model comprising a hybrid feature encoder dubbed SegMAN Encoder, and a decoder based on state space models. Specifically, the SegMAN Encoder synergistically integrates sliding local attention with dynamic state space models, enabling highly efficient global context modeling while preserving fine-grained local details. Meanwhile, the MMSCopE module in our decoder enhances multi-scale context feature extraction and adaptively scales with the input resolution. Our SegMAN-B Encoder achieves 85.1% ImageNet-1k accuracy (+1.5% over VMamba-S with fewer parameters). When paired with our decoder, the full SegMAN-B model achieves 52.6% mIoU on ADE20K (+1.6% over SegNeXt-L with 15% fewer GFLOPs), 83.8% mIoU on Cityscapes (+2.1% over SegFormer-B3 with half the GFLOPs), and 1.6% higher mIoU than VWFormer-B3 on COCO-Stuff with lower GFLOPs. Our code is available at https://github.com/yunxiangfu2001/SegMAN.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。