arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2606.16067cs.CV

逐步令牌选择用于高效多模态大语言模型

Stepwise Token Selection for Efficient Multimodal Large Language Models

Shenzhen University of Advanced Technology(深圳先进技术大学)

Landi He, Shawn Young, Lijian Xu

AI总结:

提出一种基于指针机制的逐步视觉令牌选择方法,通过可微松弛实现端到端训练,动态决定保留令牌数量,在去除88.9%令牌时保持94.6%准确率并加速1.88倍。

摘要:

在多模态大语言模型(MLLMs)中,推理成本主要由视觉令牌前缀而非语言骨干网络决定,因此令牌减少成为提高效率的关键因素。现有方法通常为视觉令牌分配独立的的重要性分数,并保留固定数量的排名靠前的令牌,这隐含地假设令牌独立且输入间压缩比均匀。在这项工作中,我们将视觉令牌剪枝重新表述为序列决策过程。具体来说,我们引入了一种指针式的选择机制,该机制迭代地选择信息丰富的令牌,每次决策都基于先前选择的令牌,并通过学习到的终止动作动态决定何时停止。这使得所选子集及其大小能够联合优化。为了实现标准语言建模目标下的端到端训练,我们设计了一种基于方差保持噪声插值方案的可微松弛,允许梯度通过离散选择过程传播。在LLaVA-v1.5-7B和Qwen2.5-VL-7B上的大量实验表明,我们的方法在不同压缩水平下始终优于固定比例基线。在去除88.9%视觉令牌的激进剪枝下,我们的方法保持了94.6%的原始准确率,同时实现了1.88倍的预填充延迟加速。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑