arXiv:2606.03075cs.CV2026-06中稿 · ICML

让视觉语言模型的缓存淘汰更懂文本,大幅节省内存同时保持高准确率。

TGV-KV: Text-Grounded KV Eviction for Vision-Language Models

论文配图:TGV-KV: Text-Grounded KV Eviction for Vision-Language Models
图 1 · 摘自论文原文
  • 基于文本引导决定视觉信息保留优先级,解决图文差异导致的缓存失效问题。
  • 在极端5%缓存预算下,仍保持99.2%的推理准确率,吞吐提升52.6%。
  • 适合需要高效推理的多模态大模型应用,如智能问答与图像理解系统。

视觉语言模型(VLMs)继承自回归生成范式,缓存所有先前令牌的键值对(KV)以加速推理,导致内存消耗随上下文长度线性增长。这一问题在VLM中尤为严重,因视觉模态存在大量冗余。尽管KV缓存淘汰方法能有效降低推理内存,但常导致性能显著下降,因其多数针对语言模型设计,忽视了文本与视觉之间的本质差异。本文系统分析了VLM中的模态差距,提出一种文本引导的KV淘汰方法(TGV-KV)。该方法包含三个模块:(1) 文本-视觉预算分配(TVB),根据互信息交互为各层分配预算;(2) 文本加权排序(TWR),基于加权文本-图像注意力评估文本优先级并排序视觉重要性;(3) 文本优先保留(TPR)策略,战略性保留文本KV以避免关键信息丢失。我们在五种不同规模和架构的模型上评估TGV-KV,结果显示,在VizWiz-VQA任务上,使用LLaVA-NeXT时,仅需5%缓存预算即可保持99.2%的全缓存准确率,并将端到端吞吐量提升52.6%。代码已开源。

原文摘要 · Abstract (English)

Vision-Language Models (VLMs) inherit the auto-regressive generation paradigm and cache the keys and values (KV) of all previous tokens to accelerate inference, resulting in memory consumption that scales linearly with context length. This issue is particularly pronounced in VLMs due to substantial redundancy in the visual modality. Although KV cache eviction approaches can effectively reduce inference memory, they often incur significant performance degradation in VLMs, as most are designed for language models and overlook the inherent gap between text and vision. By systematically analyzing the modality gap in VLMs in this work, we argue that the importance of visual information should be grounded in textual guidance and accordingly propose a Text-Grounded KV Eviction method for VLMs (TGV-KV). TGV-KV comprises three submodules: (1) Text-Vision Budgeting (TVB) assigns budget to each layer based on the mutual information interaction. (2) Text-Weighted Ranking (TWR) assesses the priority of text and ranks vision importance based on weighted text-image attention. (3) Text-Prioritised Retention (TPR) policy strategically preserves text KV to avoid acute information loss. We evaluate TGV-KV across five models with different sizes and architectures, showing that TGV-KV preserves 99.2% full-KV accuracy on the VizWiz-VQA task with LLaVA-NeXT and boosts end-to-end throughput by 52.6% with an extreme retention budget of 5%. Code is available at https://github.com/Danielement321/TGV-KV.

视觉语言模型缓存优化多模态推理

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