arXiv:2508.18264cs.CV2025-08中稿 · ICLR被引 27

用图文联合信息选关键视觉令牌,提速近2倍且几乎不丢性能

MMTok: Multimodal Coverage Maximization for Efficient Inference of VLMs

  • 基于图文覆盖度优化选择视觉令牌,兼顾文本与图像信息
  • 在POPE数据集上实现1.87倍加速,保留98.7%原模型性能
  • 仅用4个视觉令牌即可保持87.7%性能,适合高效部署场景

视觉语言模型(VLMs)通过将视觉输入转换为视觉令牌来理解视觉内容,但视觉令牌存在冗余,导致推理效率下降。现有方法多仅依赖单一模态信息进行剪枝,忽视了视觉语言任务的内在多模态特性,且缺乏通用的筛选标准。为此,本文提出MMTok方法,利用视觉与文本令牌共同构建覆盖准则,以选择更具信息量的视觉令牌。我们将子集选择问题建模为最大覆盖问题,并优化视觉令牌子集以同时覆盖原始视觉令牌集和文本令牌。在多个基准数据集和不同VLM模型上进行评估,结果表明视觉与文本信息具有互补性,融合多模态信息可显著优于单模态基线。在POPE数据集上,该方法使LLaVA-NeXT-13B达到1.87倍加速,性能保留98.7%;在LLaVA-1.5-7B上,仅使用4个视觉令牌仍可保持87.7%的原始性能。这些结果验证了覆盖准则在令牌选择中的有效性。代码已开源。

原文摘要 · Abstract (English)

Vision-Language Models (VLMs) demonstrate impressive performance in understanding visual content with language instruction by converting visual inputs to vision tokens. However, redundancy in vision tokens results in the degraded inference efficiency of VLMs. While many algorithms have been proposed to reduce the number of vision tokens, most of them apply only unimodal information (i.e., vision/text) for pruning and ignore the inherent multimodal property of vision-language tasks. Moreover, it lacks a generic criterion that can be applied to different modalities. To mitigate this limitation, in this work, we propose to leverage both vision and text tokens to select informative vision tokens by the coverage criterion. We first formulate the subset selection problem as a maximum coverage problem. Afterwards, a subset of vision tokens is optimized to cover the text tokens and the original set of vision tokens, simultaneously. The proposed method MMTok is extensively evaluated on benchmark datasets with different VLMs. The comparison illustrates that vision and text information are complementary, and combining multimodal information can surpass the unimodal baseline with a clear margin. Moreover, under the maximum coverage criterion on the POPE dataset, our method achieves a 1.87x speedup while maintaining 98.7% of the original performance on LLaVA-NeXT-13B. Finally, with only four vision tokens, 87.7% of the original performance is still preserved on LLaVA-1.5-7B. These results highlight the effectiveness of coverage in token selection. The code is available at https://github.com/Ironieser/mmtok

多模态视觉推理高效模型令牌压缩

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