arXiv:2602.04657cs.CV2026-02

通过推理目标引导压缩视觉令牌,显著提升视觉语言模型效率。

TRIO: Token Reduction via Inference-Objective Guidance for Efficient Vision-Language Models

  • 基于推理目标设计梯度重要性准则,动态筛选关键视觉令牌。
  • 仅保留11.1%视觉令牌,性能保持97.2%,推理速度提升2.14倍。
  • 无需训练、兼容FlashAttention,适合实际部署与集成应用。

近期,为加速视觉语言模型(VLM)推理而减少冗余视觉令牌成为热点。然而,现有方法多依赖视觉令牌间或跨模态相似性的启发式规则,存在压缩性能和部署实用性局限。本文提出TRIO,从推理目标出发,将视觉令牌压缩转化为保持输出结果不变性,并依据对这一目标的重要性选择令牌。具体而言,利用设计的层内代理损失生成令牌级梯度显著性,对视觉令牌进行重排序,再按非极大值抑制(NMS)原则选取最具价值的令牌。TRIO无需训练,兼容FlashAttention,可独立作为编码器无关方法部署,也可与VisionZip等编码器压缩方法结合使用。在LLaVA-Next-7B上,仅保留11.1%视觉令牌即可维持97.2%原性能,预填充速度提升2.75倍,推理速度提升2.14倍,计算量降低6.22倍,KV缓存开销减少6.05倍。

原文摘要 · Abstract (English)

Recently, reducing redundant visual tokens in vision-language models (VLMs) to accelerate VLM inference has emerged as a hot topic. However, most existing methods rely on heuristics constructed based on inter-visual-token similarity or cross-modal visual-text similarity, which gives rise to certain limitations in compression performance and practical deployment. In contrast, we propose TRIO from the perspective of inference objectives, which transforms visual token compression into preserving output result invariance and selects tokens primarily by their importance to this goal. Specifically, vision tokens are reordered with the guidance of token-level gradient saliency generated by our designed layer-local proxy loss, a coarse constraint from the current layer to the final result. Then the most valuable vision tokens are selected following the non-maximum suppression (NMS) principle.The proposed TRIO is training-free and compatible with FlashAttention, friendly to practical application and deployment. It can be deployed independently as an encoder-free method, or combined with encoder compression approaches like VisionZip for use as an encoder-involved method. On LLaVA-Next-7B, TRIO retains just 11.1\% of visual tokens but maintains 97.2\% of the original performance, with a 2.75$\times$ prefill speedup, 2.14$\times$ inference speedup, 6.22$\times$ lower FLOPs, and 6.05$\times$ reduced KV Cache overhead.Our code is available at https://github.com/ocy1/TRIO.

视觉语言模型令牌压缩推理加速

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