arXiv:2509.15235cs.CVcs.CL2025-09NeurIPS被引 22

让视觉语言模型更快生成,速度提升超1.5倍。

ViSpec: Accelerating Vision-Language Models with Vision-Aware Speculative Decoding

  • 用轻量视觉适配器压缩图像令牌,保持位置信息。
  • 在文本中注入全局图像特征,提升多模态一致性。
  • 自建长输出数据集,避免模型偷懒学习。

推测解码是加速大语言模型推理的常用技术,但在视觉语言模型(VLMs)中的应用仍不充分,现有方法速度提升不足1.5倍。随着多模态能力成为大规模模型的核心,这一差距日益显著。我们提出,大型VLM可逐层过滤冗余图像信息而不影响文本理解,而小型草稿模型则难以做到。为此,我们提出面向视觉语言模型的推测解码框架(ViSpec)。ViSpec引入轻量级视觉适配器模块,将图像令牌压缩为紧凑表示,并无缝集成到草稿模型的注意力机制中,同时保留原始图像位置信息。此外,我们为每张输入图像提取全局特征向量,并将其增强至所有后续文本令牌,以提升多模态连贯性。针对缺乏包含长助理回复的多模态数据集的问题,我们通过重构现有数据集并使用目标VLM生成扩展输出来构建专用训练数据集。我们的训练策略降低了草稿模型利用目标模型隐藏状态直接获取答案的风险,避免了快捷学习。大量实验验证了ViSpec的有效性,据我们所知,首次实现了对VLM推测解码的显著加速。代码已开源:https://github.com/KangJialiang/ViSpec。

原文摘要 · Abstract (English)

Speculative decoding is a widely adopted technique for accelerating inference in large language models (LLMs), yet its application to vision-language models (VLMs) remains underexplored, with existing methods achieving only modest speedups (<1.5x). This gap is increasingly significant as multimodal capabilities become central to large-scale models. We hypothesize that large VLMs can effectively filter redundant image information layer by layer without compromising textual comprehension, whereas smaller draft models struggle to do so. To address this, we introduce Vision-Aware Speculative Decoding (ViSpec), a novel framework tailored for VLMs. ViSpec employs a lightweight vision adaptor module to compress image tokens into a compact representation, which is seamlessly integrated into the draft model's attention mechanism while preserving original image positional information. Additionally, we extract a global feature vector for each input image and augment all subsequent text tokens with this feature to enhance multimodal coherence. To overcome the scarcity of multimodal datasets with long assistant responses, we curate a specialized training dataset by repurposing existing datasets and generating extended outputs using the target VLM with modified prompts. Our training strategy mitigates the risk of the draft model exploiting direct access to the target model's hidden states, which could otherwise lead to shortcut learning when training solely on target model outputs. Extensive experiments validate ViSpec, achieving, to our knowledge, the first substantial speedup in VLM speculative decoding. Code is available at https://github.com/KangJialiang/ViSpec.

视觉语言模型推测解码多模态加速

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