arXiv:2507.14042cs.CV2025-07被引 1

不训练的视觉Mamba token压缩方法,显著降计算量

Training-free Token Reduction for Vision Mamba

  • 基于Mamba结构设计重要性评分,无需训练
  • 在Vim-B上减少约40%计算量,性能仅降1.6%
  • 可直接嵌入各类Mamba模型,适配性强

Vision Mamba因其线性复杂度高效捕捉长程依赖而成为Vision Transformer(ViT)的有力竞争者。尽管在ViT中广泛使用的token压缩技术在Mamba中尚未被充分探索,我们发现直接将ViT的token压缩方法应用于Mamba会导致性能显著下降。这主要是因为Mamba是无注意力机制的序列模型,而多数ViT压缩方法依赖注意力机制衡量重要性,并忽略压缩后序列的顺序。本文提出一种Mamba结构感知的重要性评分方法,进而构建无需训练的Mamba Token Reduction(MTR)框架。该方法无需额外调参或训练,可无缝作为即插即用组件集成到多种Mamba模型中。大量实验表明,MTR在多个任务和骨干网络上显著降低计算负担,同时最小化性能损失。特别地,在Vim-B骨干上,计算量减少约40%,图像分类准确率仅下降1.6%,且无需重新训练。

原文摘要 · Abstract (English)

Vision Mamba has emerged as a strong competitor to Vision Transformers (ViTs) due to its ability to efficiently capture long-range dependencies with linear computational complexity. While token reduction, an effective compression technique in ViTs, has rarely been explored in Vision Mamba. Exploring Vision Mamba's efficiency is essential for enabling broader applications. However, we find that directly applying existing token reduction techniques for ViTs to Vision Mamba leads to significant performance degradation. This is primarily because Mamba is a sequence model without attention mechanisms, whereas most token reduction techniques for ViTs rely on attention mechanisms for importance measurement and overlook the order of compressed tokens. In this paper, we investigate a Mamba structure-aware importance score to evaluate token importance in a simple and effective manner. Building on this score, we further propose MTR, a training-free \textbf{M}amba \textbf{T}oken \textbf{R}eduction framework. Without the need for training or additional tuning parameters, our method can be seamlessly integrated as a plug-and-play component across various Mamba models. Extensive experiments demonstrate that our approach significantly reduces computational workload while minimizing performance impact across various tasks and multiple backbones. Notably, MTR reduces FLOPs by approximately 40\% on the Vim-B backbone, with only a 1.6\% drop in ImageNet performance without retraining.

Mambatoken压缩高效模型无训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。