arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.00523cs.CV
收藏

SAGE:通过熵引导的自适应推测解码加速视觉-语言模型

SAGE: Accelerating Vision-Language Models via Entropy-Guided Adaptive Speculative Decoding

School of Computer Science(计算机科学学院);Artificial Intelligence, Wuhan University of Technology, Hubei 430070, China(人工智能,武汉科技大学,湖北430070,中国);School of Computer Science, Wuhan University, Hubei 430072, China(计算机科学学院,武汉大学,湖北430072,中国)

Yujia Tong, Tian Zhang, Yunyang Wan, Kaiwei Lin, Jingling Yuan, Chuang Hu

AI总结:

SAGE通过熵引导的自适应推测解码方法,动态调整树结构以提高视觉-语言模型的解码速度,实现高达3.36倍的加速效果。

摘要:

推测解码已逐渐成为加速视觉-语言模型(VLMs)推理的一种有前途的方法,通过允许多个草案标记的并行验证。然而,现有方法依赖于静态树结构,这些结构在整个解码过程中保持不变,无法适应生成步骤中预测难度的变化。这导致了次优的接受长度和有限的加速效果。在本文中,我们提出SAGE,一种新的框架,该框架根据实时预测不确定性动态调整推测树结构。我们的关键见解是输出熵作为自然的置信度指标,在解码步骤中具有强时间相关性。SAGE为高置信度预测构建更深更窄的树以最大化推测深度,并为不确定预测构建更浅更宽的树以多样化探索。SAGE提高了接受长度,并在静态树基线之上实现了更快的加速。在多个基准测试中的实验展示了SAGE的有效性:在不损失输出质量的情况下,它为LLaVA-OneVision-72B提供了高达3.36倍的解码加速,对于Qwen2.5-VL-72B则为3.18倍。

英文摘要:

Speculative decoding has emerged as a promising approach to accelerate inference in vision-language models (VLMs) by enabling parallel verification of multiple draft tokens. However, existing methods rely on static tree structures that remain fixed throughout the decoding process, failing to adapt to the varying prediction difficulty across generation steps. This leads to suboptimal acceptance lengths and limited speedup. In this paper, we propose SAGE, a novel framework that dynamically adjusts the speculation tree structure based on real-time prediction uncertainty. Our key insight is that output entropy serves as a natural confidence indicator with strong temporal correlation across decoding steps. SAGE constructs deeper-narrower trees for high-confidence predictions to maximize speculation depth, and shallower-wider trees for uncertain predictions to diversify exploration. SAGE improves acceptance lengths and achieves faster acceleration compared to static tree baselines. Experiments on multiple benchmarks demonstrate the effectiveness of SAGE: without any loss in output quality, it delivers up to $3.36\times$ decoding speedup for LLaVA-OneVision-72B and $3.18\times$ for Qwen2.5-VL-72B.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑