arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2502.14504cs.CVcs.AI

PLPHP:面向高效大型视觉-语言模型的逐层逐头视觉 Token 剪枝

PLPHP: Per-Layer Per-Head Vision Token Pruning for Efficient Large Vision-Language Models

Shenzhen International Graduate School, Tsinghua University(清华大学深圳国际研究生院);Tsinghua University(清华大学);Tongji University(同济大学)

Yu Meng, Kaiyuan Li, Chenran Huang, Chen Gao, Xinlei Chen, Yong Li, Xiaoping Zhang

AI总结:

针对大型视觉-语言模型解码时视觉 token 过多导致的效率瓶颈,提出 PLPHP 逐层分配保留率并逐头剪枝,显著加速、减小 KV Cache 且性能损失极小。

摘要:

大型视觉-语言模型(LVLMs)已在一系列多模态任务中展现出卓越能力。然而,其推理效率受到解码阶段处理的大量视觉 token 的限制。为应对这一挑战,我们提出逐层逐头视觉 Token 剪枝(Per-Layer Per-Head Vision Token Pruning,PLPHP),这是一种两级细粒度剪枝方法,包括层级保留率分配和头级视觉 token 剪枝。受解码器各层之间视觉 token 再注意现象的启发,我们逐层动态调整 token 保留率:对视觉信息表现出更强注意的层保留更多视觉 token,而视觉注意较低的层则进行激进剪枝。此外,PLPHP 在注意头层面执行剪枝,使同一层内的不同头能够独立保留关键上下文。在多个基准上的实验表明,PLPHP 使解码速度提升 18%,并将 Key-Value Cache(KV Cache,键值缓存)大小减少 50% 以上,而代价仅为平均性能下降 0.46%,同时在多图像任务中还取得了显著的性能提升。这些结果凸显了细粒度 token 剪枝的有效性,并有助于推动 LVLMs 的效率和可扩展性发展。我们的源代码将公开发布。

英文摘要:

Large Vision-Language Models (LVLMs) have demonstrated remarkable capabilities across a range of multimodal tasks. However, their inference efficiency is constrained by the large number of visual tokens processed during decoding. To address this challenge, we propose Per-Layer Per-Head Vision Token Pruning (PLPHP), a two-level fine-grained pruning method including Layer-Level Retention Rate Allocation and Head-Level Vision Token Pruning. Motivated by the Vision Token Re-attention phenomenon across decoder layers, we dynamically adjust token retention rates layer by layer. Layers that exhibit stronger attention to visual information preserve more vision tokens, while layers with lower vision attention are aggressively pruned. Furthermore, PLPHP applies pruning at the attention head level, enabling different heads within the same layer to independently retain critical context. Experiments on multiple benchmarks demonstrate that PLPHP delivers an 18% faster decoding speed and reduces the Key-Value Cache (KV Cache) size by over 50%, all at the cost of 0.46% average performance drop, while also achieving notable performance improvements in multi-image tasks. These results highlight the effectiveness of fine-grained token pruning and contribute to advancing the efficiency and scalability of LVLMs. Our source code will be made publicly available.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑