为Mamba模型设计适配器,提升视觉任务表现
Mamba-Adaptor: State Space Model Adaptor for Visual Recognition
- 引入Adaptor-T和Adaptor-S模块增强上下文记忆与空间建模
- 在ImageNet和COCO上达到当前最佳性能
- 适用于通用视觉骨干、模型增强与高效微调
近期状态空间模型(SSM),尤其是Mamba,在视觉建模中表现出色且效率优异。然而,其在视觉任务中的应用受限于三个主要问题:1)因果计算无法获取全局上下文;2)计算当前隐藏状态时存在长程遗忘;3)输入序列化导致空间结构建模能力弱。为此,本文提出一种简单而强大的Mamba视觉适配器(Mamba-Adaptor),包含两个功能模块:Adaptor-T通过轻量级预测模块选择可学习位置作为记忆增强,缓解长程遗忘;Adaptor-S采用多尺度空洞卷积核增强空间建模并引入图像先验。两者均在因果计算中扩展上下文建模能力,输出由不可访问特征增强。我们探索了三种使用方式:通用视觉骨干、提升预训练骨干性能的增强模块、高效微调模块。大量实验验证其在三种设置下的有效性。尤其在ImageNet和COCO基准上达到当前最优表现。
原文摘要 · Abstract (English)
Recent State Space Models (SSM), especially Mamba, have demonstrated impressive performance in visual modeling and possess superior model efficiency. However, the application of Mamba to visual tasks suffers inferior performance due to three main constraints existing in the sequential model: 1) Casual computing is incapable of accessing global context; 2) Long-range forgetting when computing the current hidden states; 3) Weak spatial structural modeling due to the transformed sequential input. To address these issues, we investigate a simple yet powerful vision task Adaptor for Mamba models, which consists of two functional modules: Adaptor-T and Adaptor-S. When solving the hidden states for SSM, we apply a lightweight prediction module Adaptor-T to select a set of learnable locations as memory augmentations to ease long-range forgetting issues. Moreover, we leverage Adapator-S, composed of multi-scale dilated convolutional kernels, to enhance the spatial modeling and introduce the image inductive bias into the feature output. Both modules can enlarge the context modeling in casual computing, as the output is enhanced by the inaccessible features. We explore three usages of Mamba-Adaptor: A general visual backbone for various vision tasks; A booster module to raise the performance of pretrained backbones; A highly efficient fine-tuning module that adapts the base model for transfer learning tasks. Extensive experiments verify the effectiveness of Mamba-Adaptor in three settings. Notably, our Mamba-Adaptor achieves state-of the-art performance on the ImageNet and COCO benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。