提出自适应视觉令牌剪枝方法,提升大模型效率与准确性
AgilePruner: An Empirical Study of Attention and Diversity for Adaptive Visual Token Pruning in Large Vision-Language Models
- 结合注意力与多样性指标,分析剪枝策略优劣
- 复杂图像用多样性剪枝,简单图像用注意力剪枝更有效
- 新方法在主流基准和幻觉检测中表现稳定
大型视觉语言模型(LVLMs)采用视觉令牌剪枝以缓解海量视觉令牌序列带来的计算负担。以往研究多聚焦于基于注意力或多样性的剪枝方法,但对其特性与局限的深入分析仍不充分。本文通过有效秩(erank)衡量特征多样性、注意力分数熵分析视觉令牌处理机制,揭示两大发现:(1) erank定量分析显示,多数多样性导向剪枝方法实际保留的特征多样性远低于预期;在CHAIR数据集上,其保留的多样性与幻觉频率显著相关,高于注意力剪枝;(2) 注意力方法在视觉信息集中的简单图像上更有效,而多样性方法在特征分布复杂的图像上表现更优。基于此,我们证明在现有混合剪枝策略中引入图像感知调整可持续提升性能。此外,我们设计了一种简洁的自适应剪枝机制,其在标准基准及幻觉专项评估中均表现出强且稳定的性能。
原文摘要 · Abstract (English)
Large Vision-Language Models (LVLMs) have adopted visual token pruning strategies to mitigate substantial computational overhead incurred by extensive visual token sequences. While prior works primarily focus on either attention-based or diversity-based pruning methods, in-depth analysis of these approaches' characteristics and limitations remains largely unexplored. In this work, we conduct thorough empirical analysis using effective rank (erank) as a measure of feature diversity and attention score entropy to investigate visual token processing mechanisms and analyze the strengths and weaknesses of each approach. Our analysis reveals two insights: (1) Our erank-based quantitative analysis shows that many diversity-oriented pruning methods preserve substantially less feature diversity than intended; moreover, analysis using the CHAIR dataset reveals that the diversity they do retain is closely tied to increased hallucination frequency compared to attention-based pruning. (2) We further observe that attention-based approaches are more effective on simple images where visual evidence is concentrated, while diversity-based methods better handle complex images with distributed features. Building on these empirical insights, we show that incorporating image-aware adjustments into existing hybrid pruning strategies consistently improves their performance. We also provide a minimal instantiation of our empirical findings through a simple adaptive pruning mechanism, which achieves strong and reliable performance across standard benchmarks as well as hallucination-specific evaluations. Our project page available at https://cvsp-lab.github.io/AgilePruner.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。