arXiv:2604.07812cs.CV2026-04被引 3

通过关注注意力头重要性,高效剪裁视觉令牌,提升多模态模型推理速度。

HAWK: Head Importance-Aware Visual Token Pruning in Multimodal Models

  • 基于注意力头重要性动态评估视觉令牌价值,实现精准剪枝。
  • 在保留96.0%准确率的前提下,剪掉80.2%的视觉令牌。
  • 无需训练,兼容多种多模态模型,适合实时与低资源场景。

在多模态大语言模型(MLLMs)中,视觉令牌数量激增导致推理时间与计算开销显著上升,限制了其在实时或资源受限场景的应用。视觉令牌剪枝是一种降低推理成本的有效策略。现有方法通常假设所有注意力头对视觉理解贡献均等,但本研究发现不同头可能捕捉不同的视觉语义并承担不同角色。为此,我们提出HAWK——一种头重要性感知的视觉令牌剪枝方法,通过评估注意力头在视觉任务中的重要性,最大化保留关键令牌。HAWK利用头重要性权重与文本引导的注意力来判断视觉令牌的重要性,有效保留任务相关令牌,剔除冗余信息。该方法完全无需训练,可无缝适配多种MLLM。在多个主流视觉-语言基准上的实验表明,HAWK达到当前最优性能。应用于Qwen2.5-VL时,剪除80.2%视觉令牌后仍保持96.0%原始准确率,端到端延迟降至74.4%,并显著降低各模型的GPU内存占用。

原文摘要 · Abstract (English)

In multimodal large language models (MLLMs), the surge of visual tokens significantly increases the inference time and computational overhead, making them impractical for real-time or resource-constrained applications. Visual token pruning is a promising strategy for reducing the cost of MLLM inference by removing redundant visual tokens. Existing research usually assumes that all attention heads contribute equally to the visual interpretation. However, our study reveals that different heads may capture distinct visual semantics and inherently play distinct roles in visual processing. In light of this observation, we propose HAWK, a head importance-aware visual token pruning method that perceives the varying importance of attention heads in visual tasks to maximize the retention of crucial tokens. By leveraging head importance weights and text-guided attention to assess visual token significance, HAWK effectively retains task-relevant visual tokens while removing redundant ones. The proposed HAWK is entirely training-free and can be seamlessly applied to various MLLMs. Extensive experiments on multiple mainstream vision-language benchmarks demonstrate that HAWK achieves state-of-the-art accuracy. When applied to Qwen2.5-VL, HAWK retains 96.0% of the original accuracy after pruning 80.2% of the visual tokens. Additionally, it reduces end-to-end latency to 74.4% of the original and further decreases GPU memory usage across the tested models. The code is available at https://github.com/peppery77/HAWK.git.

多模态视觉剪枝注意力机制推理加速

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