arXiv:2608.04132cs.CV2026-08

用可学习的分配策略,大幅减少视觉令牌数量而不损失性能。

RUTA: Principled Visual Token Allocation via Rate-Utility Optimization

论文配图:RUTA: Principled Visual Token Allocation via Rate-Utility Optimization
图 1 · 摘自论文原文
  • 根据查询动态选择保留哪些视觉令牌,按需分配数量。
  • 仅用2%~4%令牌,仍保持88%~94%的任务性能。
  • 适合需要高效推理的多模态大模型应用者参考。

高分辨率图像和长视频为视觉语言模型提供了丰富的上下文以支持多模态推理与细粒度感知,但由此产生的长视觉令牌序列使大语言模型端的计算与内存开销巨大。现有视觉令牌压缩方法通常采用预设缩减率,而近期方法则通过特定学习阈值或重要性预测器自适应调整令牌数。本文提出RUTA,一种基于速率-效用优化的原理性视觉令牌分配方法,通过联合学习每个图像-查询对应保留哪些令牌及其数量,在大语言模型前进行预处理缩减。RUTA构建查询相关的候选令牌,并预测每个候选的保留概率。训练时,这些概率参数化独立的伯努利门控,其总和提供每个对在训练时的可微令牌数估计。保留的令牌作为锚点,根据语义相似性和空间邻近性聚合未保留令牌的信息。RUTA以惩罚性速率-效用目标优化,平衡下游任务损失与预期令牌使用量。在五个基准上平均测试,相对于各骨干模型的全令牌基线,RUTA在LLaVA-NeXT-7B和Qwen3-VL-8B上分别仅使用2.0%和4.2%的视觉令牌,同时保留88.2%和94.4%的任务性能。

原文摘要 · Abstract (English)

High-resolution images and long videos provide vision-language models with rich context for multimodal reasoning and fine-grained perception, but the resulting long visual token sequences make large language model-side computation and memory costly. Existing visual token reducers often operate at prescribed rates, while recent methods adapt token counts across inputs using method-specific learned thresholds or importance predictors. We introduce RUTA, a principled Rate-Utility Token Allocation method that performs pre-LLM reduction by jointly learning which tokens to retain and how many to allocate to each image-query pair. RUTA constructs query-conditioned candidate tokens and predicts a retention probability for each candidate. During training, these probabilities parameterize independent Bernoulli gates, while their sum provides a differentiable training-time estimate of the token count for each pair. Retained tokens serve as anchors that aggregate information from non-retained tokens according to semantic affinity and spatial proximity. RUTA is optimized with a penalized rate-utility objective that balances downstream task loss against expected token usage. Averaged across five benchmarks and measured relative to each backbone's full-token baseline, RUTA uses only $2.0\%$ and $4.2\%$ of visual tokens while preserving $88.2\%$ and $94.4\%$ of task performance on LLaVA-NeXT-7B and Qwen3-VL-8B, respectively.

视觉令牌效率优化多模态

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