arXiv:2412.01818cs.CVcs.AI2024-12ICCV被引 145

用视觉线索替代注意力分数,高效剪裁视觉令牌,大幅降低大模型计算开销。

Beyond Text-Visual Attention: Exploiting Visual Cues for Effective Token Pruning in VLMs

  • 基于视觉注意力选关键帧,再去重保留多样性
  • 无需训练即减91%算力,延迟降75%仍保性能
  • 适配多种模型,插件式部署,实测优于传统方法

大型视觉语言模型(LVLM)中视觉令牌远多于文本令牌,带来巨大计算负担。现有方法多依赖文本-视觉注意力分数判断视觉令牌重要性,但本研究分析发现该分数并非理想剪裁指标。为此提出VisPruner,一种利用视觉线索进行更有效令牌剪裁的即插即用方法:先通过视觉注意力筛选关键令牌,再基于相似性去除重复项,保留输入图像的丰富视觉信息。实验表明,该方法在多种模型架构与压缩比下均保持优异性能,显著优于基于文本-视觉注意力的方法。尤其值得注意的是,无需训练即可使LLaVA-1.5-7B的FLOPs减少91%,推理延迟降低75%,性能基本不变。代码已开源。

原文摘要 · Abstract (English)

Large vision-language models (LVLMs) generally contain significantly more visual tokens than their textual counterparts, resulting in a considerable computational burden. Recent efforts have been made to tackle this issue by pruning visual tokens early within the language model. Most existing works use attention scores between text and visual tokens to assess the importance of visual tokens. However, in this study, we first analyze the text-visual attention in the language model and find that this score is not an ideal indicator for token pruning. Based on the analysis, We propose VisPruner, a plug-and-play method that utilizes visual cues for more effective token pruning in LVLMs. Specifically, we first use visual attention to select a limited number of significant tokens. Then, we remove duplicate tokens from the remaining ones based on their similarity. By retaining diverse tokens alongside the initially selected important tokens, we maximally preserve the visual information of the input image. Experimental results demonstrate that our VisPruner sustains strong performance across various VLM architectures and reduction ratios, significantly outperforming existing methods based on text-visual attention. Notably, without any training, VisPruner can reduce the FLOPs of LLaVA-1.5-7B by 91% and inference latency by 75%, while maintaining comparable performance. Our code is available at https://github.com/Theia-4869/VisPruner.

视觉语言模型令牌剪裁效率优化无训练

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