arXiv:2606.16067cs.CV2026-06被引 3

通过逐步选择视觉令牌提升多模态大模型推理效率

Stepwise Token Selection for Efficient Multimodal Large Language Models

论文配图:Stepwise Token Selection for Efficient Multimodal Large Language Models
图 1 · 摘自论文原文
  • 采用动态逐轮选择机制,基于已选令牌决定下一步
  • 压缩88.9%视觉令牌仍保持94.6%原始准确率,预填充速度提升1.88倍
  • 适用于追求高效推理的多模态模型部署场景

在多模态大语言模型中,推理开销主要来自视觉令牌前缀而非语言主干,因此令牌压缩是提升效率的关键。现有方法通常为视觉令牌独立分配重要性评分并保留固定数量的高分令牌,隐含假设令牌间独立且压缩比例恒定。本文将视觉令牌剪枝重构为序列决策过程,引入指针式选择机制,迭代选取信息量大的令牌,并基于已选令牌动态决定终止条件,实现对选中子集及其大小的联合优化。为在标准语言建模范式下实现端到端训练,设计基于方差保持噪声插值的可微松弛方案,使梯度能穿过离散选择过程。在LLaVA-v1.5-7B和Qwen2.5-VL-7B上的大量实验表明,该方法在不同压缩水平下均优于固定比率基线。在移除88.9%视觉令牌的激进剪枝下,仍保持94.6%原始准确率,预填充延迟提速1.88倍。

原文摘要 · Abstract (English)

In multimodal large language models (MLLMs), inference cost is largely dominated by the visual token prefix rather than the language backbone, making token reduction a key factor for improving efficiency. Existing approaches typically assign independent importance scores to visual tokens and retain a fixed number of top-ranked tokens, implicitly assuming token independence and a uniform compression ratio across inputs. In this work, we reformulate visual token pruning as a sequential decision-making process. Specifically, we introduce a pointer-style selection mechanism that iteratively chooses informative tokens, conditioning each decision on previously selected ones, and dynamically determines when to stop via a learned termination action. This enables joint optimization of both the selected subset and its size. To enable end-to-end training under standard language modeling objectives, we design a differentiable relaxation based on a variance-preserving noise interpolation scheme, allowing gradients to propagate through the discrete selection process. Extensive experiments on LLaVA-v1.5-7B and Qwen2.5-VL-7B demonstrate that our approach consistently outperforms fixed-ratio baselines across different compression levels. Under aggressive pruning that removes 88.9% of visual tokens, our method preserves 94.6% of the original accuracy while achieving a 1.88x speed-up in prefill latency.

多模态模型压缩推理加速令牌选择

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