动态调整视觉令牌修剪比例,显著降低大模型计算开销。
ATP-LLaVA: Adaptive Token Pruning for Large Vision Language Models
- 根据输入实例自适应计算每层的令牌重要性与修剪阈值
- 平均减少75%视觉令牌,仅损失1.9%性能
- 适合移动端或资源受限设备部署的多模态模型
大型视觉语言模型在多模态任务中表现卓越,但长序列视觉令牌的处理成本在资源受限设备上过高。现有方法采用固定比例修剪视觉令牌以降低计算负担,但未考虑不同模型层与图像-提示对之间的差异。我们发现修剪比例的影响在不同层和实例间存在显著变化,因此提出ATP-LLaVA:一种分层且实例自适应的视觉令牌修剪方法。核心是引入自适应令牌修剪(ATP)模块,基于输入实例动态计算重要性分数与修剪阈值,可无缝嵌入任意两层LLM之间,几乎无额外开销。同时设计空间增强修剪(SAP)策略,从冗余与空间建模双重角度进行修剪。实验表明,该方法将平均令牌数减少75%,在七个主流基准上仅造成1.9%性能下降。
原文摘要 · Abstract (English)
Large Vision Language Models (LVLMs) have achieved significant success across multi-modal tasks. However, the computational cost of processing long visual tokens can be prohibitively expensive on resource-limited devices. Previous methods have identified redundancy in visual tokens within the Large Language Model (LLM) decoder layers and have mitigated this by pruning tokens using a pre-defined or fixed ratio, thereby reducing computational overhead. Nonetheless, we observe that the impact of pruning ratio varies across different LLM layers and instances (image-prompt pairs). Therefore, it is essential to develop a layer-wise and instance-wise vision token pruning strategy to balance computational cost and model performance effectively. We propose ATP-LLaVA, a novel approach that adaptively determines instance-specific token pruning ratios for each LLM layer. Specifically, we introduce an Adaptive Token Pruning (ATP) module, which computes the importance score and pruning threshold based on input instance adaptively. The ATP module can be seamlessly integrated between any two LLM layers with negligible computational overhead. Additionally, we develop a Spatial Augmented Pruning (SAP) strategy that prunes visual tokens with both token redundancy and spatial modeling perspectives. Our approach reduces the average token count by 75% while maintaining performance, with only a minimal 1.9% degradation across seven widely used benchmarks. The project page can be accessed via https://yxxxb.github.io/ATP-LLaVA-page/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。