通过分层上下文感知剪枝,高效减少视觉语言模型的冗余图像令牌。
CoViPAL: Layer-wise Contextualized Visual Token Pruning for Large Vision-Language Models
- 分层设计剪枝模块,利用上下文信号识别并移除冗余视觉令牌。
- 在相同令牌预算下优于无训练剪枝方法,且性能超过需监督训练的方法。
- 轻量级插件式模块,兼容多种模型架构,适合部署优化。
大型视觉语言模型(LVLMs)处理包含文本和图像/视频提取的视觉令牌的多模态输入。由于视觉信息丰富,单张图像可生成数千个视觉令牌,导致预填充阶段计算成本高、解码阶段内存开销大。现有方法尝试剪枝冗余视觉令牌,揭示了视觉表征中的显著冗余。但这些方法在浅层因缺乏足够上下文信息而表现不佳。我们认为,即使在浅层,许多视觉令牌本质上冗余,可通过适当上下文信号安全有效地剪枝。为此,我们提出CoViPAL,一种分层上下文感知的视觉令牌剪枝方法,采用轻量级、模型无关的即插即用剪枝模块(PPM),在令牌进入LVLM前预测并移除冗余部分。大量实验表明,CoViPAL在等量令牌预算下超越无训练剪枝方法,并优于需监督训练的方法。该方法为提升LVLM推理效率提供了可扩展、高效的解决方案,且不损失精度。
原文摘要 · Abstract (English)
Large Vision-Language Models (LVLMs) process multimodal inputs consisting of text tokens and vision tokens extracted from images or videos. Due to the rich visual information, a single image can generate thousands of vision tokens, leading to high computational costs during the prefilling stage and significant memory overhead during decoding. Existing methods attempt to prune redundant vision tokens, revealing substantial redundancy in visual representations. However, these methods often struggle in shallow layers due to the lack of sufficient contextual information. We argue that many visual tokens are inherently redundant even in shallow layers and can be safely and effectively pruned with appropriate contextual signals. In this work, we propose CoViPAL, a layer-wise contextualized visual token pruning method that employs a Plug-and-Play Pruning Module (PPM) to predict and remove redundant vision tokens before they are processed by the LVLM. The PPM is lightweight, model-agnostic, and operates independently of the LVLM architecture, ensuring seamless integration with various models. Extensive experiments on multiple benchmarks demonstrate that CoViPAL outperforms training-free pruning methods under equal token budgets and surpasses training-based methods with comparable supervision. CoViPAL offers a scalable and efficient solution to improve inference efficiency in LVLMs without compromising accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。