arXiv:2411.14164cs.CVcs.AI2024-11被引 10

通过注意力机制筛选关键视觉标记,高效压缩大模型输入。

FoPru: Focal Pruning for Efficient Large Vision-Language Models

  • 基于视觉编码器的注意力得分,动态筛选重要视觉标记。
  • 在多个数据集上实现超70%的标记剪枝率,精度损失低于1%。
  • 无需训练,适合部署在资源受限的多模态应用中。

大型视觉语言模型(LVLMs)通过将图像转换为视觉标记并与文本标记对齐,使大语言模型具备理解视觉输入的能力。然而,大量冗余视觉标记限制了推理效率。为此,我们提出一种无需训练的焦点剪枝方法(FoPru),根据视觉编码器生成的注意力得分,选择性保留关键视觉标记。该方法包含两种策略:全局视角的排序策略,按重要性保留核心标记;局部视角的行策略,保持图像中连续的关键信息。被选中的标记按原始位置重排。在多种LVLM和多模态数据集上的实验表明,该方法可剪枝超过70%的冗余标记,同时保持精度损失低于1%,显著提升推理效率。

原文摘要 · Abstract (English)

Large Vision-Language Models (LVLMs) represent a significant advancement toward achieving superior multimodal capabilities by enabling powerful Large Language Models (LLMs) to understand visual input. Typically, LVLMs utilize visual encoders, such as CLIP, to transform images into visual tokens, which are then aligned with textual tokens through projection layers before being input into the LLM for inference. Although existing LVLMs have achieved significant success, their inference efficiency is still limited by the substantial number of visual tokens and the potential redundancy among them. To mitigate this issue, we propose Focal Pruning (FoPru), a training-free method that prunes visual tokens based on the attention-based token significance derived from the vision encoder. Specifically, we introduce two alternative pruning strategies: 1) the rank strategy, which leverages all token significance scores to retain more critical tokens in a global view; 2) the row strategy, which focuses on preserving continuous key information in images from a local perspective. Finally, the selected tokens are reordered to maintain their original positional relationships. Extensive experiments across various LVLMs and multimodal datasets demonstrate that our method can prune a large number of redundant tokens while maintaining high accuracy, leading to significant improvements in inference efficiency.

视觉语言模型剪枝效率优化注意力机制

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