arXiv:2509.23663cs.CV2025-09被引 2

不需训练,通过分层剪枝提升视觉语言模型推理速度

HIVTP: A Training-Free Method to Improve VLMs Efficiency via Hierarchical Visual Token Pruning Using Middle-Layer-Based Importance Score

  • 用中间层注意力图评估视觉标记重要性,更精准定位关键区域
  • 分两阶段剪枝:全局保留区域高重要性标记,局部保留窗口内最优标记
  • 在不降精度前提下,提速超50%,适合部署在资源受限场景

视觉语言模型(VLMs)在多模态任务中表现强大,但视觉编码器输出的大量视觉标记严重拖慢推理效率。已有研究指出其中许多标记可安全剪枝。本文提出HIVTP,一种无需训练的方法,通过基于中间层的重要性评分实现分层视觉标记剪枝。具体地,利用视觉编码器中间层提取的注意力图,更准确反映细粒度和对象级关注区域,以评估标记重要性。在此基础上,提出分层剪枝策略:将一维视觉标记序列重构为二维空间布局,在全局保留阶段,将图像划分为区域,保留各区域中重要性高的标记;在局部保留阶段,进一步划分小窗口,保留每个窗口内的最高重要性标记。实验表明,HIVTP可使LLaVA-v1.5-7B和LLaVA-Next-7B的时间到首个标记(TTFT)分别降低50.0%和55.1%,生成吞吐量提升60.9%和47.3%,且不损失准确率,甚至在部分基准上有所提升。相比先前方法,HIVTP在保持更高精度的同时实现了更高的推理效率。

原文摘要 · Abstract (English)

Vision-Language Models (VLMs) have shown strong capabilities on diverse multimodal tasks. However, the large number of visual tokens output by the vision encoder severely hinders inference efficiency, and prior studies have shown that many of these tokens are not important and can therefore be safely pruned. In this work, we propose HIVTP, a training-free method to improve VLMs efficiency via hierarchical visual token pruning using a novel middle-layer-based importance score. Specifically, we utilize attention maps extracted from the middle layers of the vision encoder, which better reflect fine-grained and object-level attention, to estimate visual token importance. Based on this, we propose a hierarchical visual token pruning method to retain both globally and locally important visual tokens. Specifically, we reshape the 1-D visual token sequence output by the vision encoder into a 2-D spatial layout. In the global retaining stage, we divide the image into regions and retain tokens with higher importance scores in each region; in the local retaining stage, we then divide the image into small windows and retain the most important token in each local window. Experimental results show that our proposed method, HIVTP, can reduce the time-to-first-token (TTFT) of LLaVA-v1.5-7B and LLaVA-Next-7B by up to 50.0% and 55.1%, respectively, and improve the token generation throughput by up to 60.9% and 47.3%, without sacrificing accuracy, and even achieving improvements on certain benchmarks. Compared with prior works, HIVTP achieves better accuracy while offering higher inference efficiency.

视觉语言模型标记剪枝推理加速无训练

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