TransPrune:面向高效大型视觉语言模型的词元变化剪枝
TransPrune: Token Transition Pruning for Efficient Large Vision-Language Model
对做 LVLM 视觉侧推理加速的人,可直接复用「词元表征在模块间传播时的幅值与方向变化」作为重要性信号,替代有位置偏置的注意力/相似度准则,且免训练即可插到 LLaVA、Qwen2.5-VL 等架构中。
原文依据 we propose TransPrune, a training-free and efficient token pruning method
摘要译文
由模型翻译 · 术语统一 · 请对照原文核对关键结论大型视觉语言模型(LVLM)推动了多模态学习的发展,但由于输入大量视觉词元,面临高昂的计算开销问题,这促使人们采用词元剪枝来提升推理效率。关键挑战在于识别哪些词元真正重要。现有方法大多依赖基于注意力或相似度的准则来估计词元重要性,但它们本身存在一定局限,例如与任务无关以及存在位置偏置。在本工作中,我们从 LVLM 中词元变化的新视角来探讨词元重要性的分配,其中词元变化被定义为词元表示在模型各模块中传播时所发生的变化。我们观察到,词元表示的变化提供了有意义的语义信息信号。基于这一洞见,我们提出 TransPrune,一种免训练且高效的词元剪枝方法。具体而言,TransPrune 通过结合词元变化幅度(TTV)与指令引导注意力(IGA)来评估词元重要性,从而渐进式地剪枝词元;前者衡量词元表示在幅值和方向上的变化,后者通过注意力衡量指令对视觉词元的关注强度。在 LLaVA-v1.5、LLaVA-Next 和 Qwen2.5-VL 等多种 LVLM 架构上的大量实验表明,TransPrune 在保持可比的多模态性能的同时,将推理 TFLOPs 降低一半以上。代码见 https://github.com/liaolea/TransPrune。
英文原文摘要 1601 字符
Large Vision-Language Models (LVLMs) have advanced multimodal learning but face high computational cost issues due to the input of large number of visual tokens, motivating token pruning to improve inference efficiency.The key challenge lies in identifying which tokens are truly important.Most existing approaches rely on attention- or similarity-based criteria to estimate token importance.However, they inherently suffer from certain limitations, such as being task-agnostic and exhibiting positional bias.In this work, we explore a new perspective on token importance assignment based on token transitions in LVLMs, where token transitions are defined as the changes in token representations occurring as they propagate through the model's modules.We observe that the transition of token representations provides a meaningful signal of semantic information.Based on this insight, we propose TransPrune, a training-free and efficient token pruning method.Specifically, TransPrune progressively prunes tokens by assessing their importance through a combination of Token Transition Variation (TTV), which measures changes in both the magnitude and direction of token representations; as well as Instruction-Guided Attention (IGA), which measures how strongly the instruction attends to visual tokens via attention.Extensive experiments on various LVLM architectures, such as LLaVA-v1.5, LLaVA-Next and Qwen2.5-VL, demonstrate that TransPrune maintains comparable multimodal performance while reducing inference TFLOPs by more than half. The code is available at https://github.com/liaolea/TransPrune.