arXiv:2503.14075cs.CVcs.CL2025-03被引 16

通过生长轻量模块加速视觉语言模型,兼顾精度与生成速度。

Growing a Multi-head Twig via Distillation and Reinforcement Learning to Accelerate Large Vision-Language Models

  • 在早期层添加多头轻量模块,结合蒸馏与强化学习优化剪枝。
  • 剪掉88.9%视觉令牌,保留96%原始性能,长文本生成提速154%。
  • 适合追求高效率部署的视觉语言模型应用开发者。

大型视觉语言模型(VLM)在开放世界多模态理解中表现卓越,但计算开销大,难以实际部署。现有方法通过早期层注意力图剪枝冗余视觉令牌,但仍存在两个问题:早期注意力信号不敏感导致性能下降,以及生成长文本时加速有限。为此,本文提出TwigVLM——在基础VLM早期层上生长一个轻量级模块(twig)。相比纯视觉令牌剪枝方法,TwigVLM采用基于twig引导的令牌剪枝(TTP)策略,在保留96%原始性能的同时剪除88.9%的视觉令牌;并引入自推测解码(SSD)策略,使长文本生成速度提升154%。进一步提出TwigVLM++,通过两阶段训练(蒸馏+剪枝导向强化学习)和树状结构的SSD策略,实现更高精度剪枝与推理加速。实验以LLaVA-1.5-7B为基线,结果表明其在准确率与速度上均优于当前最优加速方法。

原文摘要 · Abstract (English)

Large vision-language models (VLMs) have demonstrated remarkable capabilities in open-world multimodal understanding, yet their high computational overheads pose great challenges for practical deployment. Some recent works have proposed methods to accelerate VLMs by pruning redundant visual tokens guided by the attention maps of VLM's early layers. Despite the success of these token pruning methods, they still suffer from two major shortcomings: (i) considerable accuracy drop due to insensitive attention signals in early layers, and (ii) limited speedup when generating long responses (e.g., 30 tokens). To address the limitations above, we present TwigVLM -- a simple and general architecture by growing a lightweight module, named twig, upon an early layer of the base VLM. Compared with most existing VLM acceleration methods purely based on visual token pruning, our TwigVLM not only achieves better accuracy retention by employing a twig-guided token pruning (TTP) strategy, but also yields higher generation speed by utilizing a self-speculative decoding (SSD) strategy. Taking LLaVA-1.5-7B as the base VLM, experimental results show that TwigVLM preserves 96% of the original performance after pruning 88.9% of the visual tokens and achieves 154% speedup in generating long responses, delivering significantly better performance in terms of both accuracy and speed over the state-of-the-art VLM acceleration methods. Moreover, we extend TwigVLM to an improved TwigVLM++ variant by introducing a novel multi-head twig architecture with a specialized pruning head. TwigVLM++ improves pruning quality via a two-stage training paradigm combining a distillation learning stage and a pruning-oriented reinforcement learning stage, and further accelerates inference via a tree-based SSD strategy.

视觉语言模型模型加速剪枝强化学习

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