arXiv:2510.16598cs.CV2025-10中稿 · ACM Multimedia 202…被引 10

提出可端到端学习的视觉令牌压缩方法,提升多模态大模型效率与性能。

VisionSelector: End-to-End Learnable Visual Token Compression for Efficient Multimodal LLMs

  • 设计可学习的VisionSelector模块,实现自适应视觉令牌筛选。
  • 在10%保留率下比启发式方法高12.14个百分点,30%保留率下保持100%性能。
  • 轻量级结构仅12.85M参数,适合部署于各类多模态大模型中。

多模态大语言模型(MLLMs)因高分辨率图像或多图输入生成大量视觉令牌,面临显著的计算与内存瓶颈。以往的令牌压缩方法常受启发式规则限制,易丢弃关键信息,且存在注意力集中等偏差,导致在高压缩比下性能骤降。为此,本文将令牌压缩重构为轻量级、可插拔的端到端可学习框架。提出VisionSelector,一个解耦于MLLM主干的评分模块,结合可微分Top-K机制与课程退火策略,弥合训练与推理间的差距,实现不同压缩率下的高效自适应选择。该模块仅含12.85M可训练参数,展现出对多种压缩率的良好泛化能力,能精准识别关键令牌。实验表明,在30%保留率下,其在MME数据集上保持100%性能;在10%保留率下,相比代表性启发式基线提升12.14个百分点,并使预填充速度翻倍。代码已开源。

原文摘要 · Abstract (English)

Multimodal Large Language Models (MLLMs) encounter significant computational and memory bottlenecks from the massive number of visual tokens generated by high-resolution images or multi-image inputs. Previous token compression techniques are often constrained by heuristic rules that risk discarding critical information. They may suffer from biases, such as attention sinks, that lead to sharp performance drops under aggressive compression ratios. To address these limitations, we reformulate token compression as a lightweight plug-and-play framework and turn it into an end-to-end learnable decision process. Specifically, we propose VisionSelector, a scorer module decoupled from the MLLM backbone that incorporates a differentiable Top-K mechanism and a curriculum annealing strategy to bridge the training-inference gap, enabling efficient and adaptive token selection across various compression rates. Remarkably lightweight with only 12.85M trainable parameters, VisionSelector demonstrates generalization across various compression rates and adaptively identifies critical tokens. This leads to superior performance across the evaluated compression budgets, evidenced by preserving 100% performance on MME with a 30% retention budget, outperforming representative heuristic baselines by 12.14 percentage points at a 10% retention budget, and doubling prefill speed. Our code is available at https://github.com/JulietChoo/VisionSelector .

多模态压缩轻量化可学习

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