解决多模态大模型长序列处理中的令牌剪枝失效问题
D2Pruner: Debiased Importance and Structural Diversity for MLLM Token Pruning
- 用去偏注意力得分选关键令牌,再结合空间与语义图进行多样性剪枝
- 在保持99.2%性能的同时降低74.2%计算量,定位任务性能提升63.53%
- 适合需要高精度视觉定位的多模态应用开发者
长视觉令牌序列处理对多模态大语言模型(MLLMs)带来显著计算负担。尽管令牌剪枝可加速推理,但现有方法在细粒度定位任务中表现严重退化。我们发现,重要性方法存在强位置偏差,而多样性方法忽视结构信息和用户提示。为此,提出D2Pruner框架:首先基于去偏注意力得分选取关键令牌作为枢纽;随后在融合空间邻近性与语义相似性的混合图上执行最大独立集(MIS)选择,迭代保留最重要且未被覆盖的令牌,移除其邻居。实验表明,该方法兼具高效与高保真性:应用于LLaVA-1.5-7B时,减少74.2% FLOPs,性能保留99.2%;在InternVL-2.5-8B的定位任务中,以90%令牌压缩率维持85.7%性能,相比现有方法最高提升63.53%。
原文摘要 · Abstract (English)
Processing long visual token sequences poses a significant computational burden on Multimodal Large Language Models (MLLMs). While token pruning offers a path to acceleration, we find that current methods, while adequate for general understanding, catastrophically fail on fine-grained localization tasks. We attribute this failure to the inherent flaws of the two prevailing strategies: importance-based methods suffer from a strong positional bias, an inherent model artifact that distracts from semantic content, while diversity-based methods exhibit structural blindness, disregarding the user's prompt and spatial redundancy. To address this, we introduce D2Pruner, a framework that rectifies these issues by uniquely combining debiased importance with a structural pruning mechanism. Our method first secures a core set of the most critical tokens as pivots based on a debiased attention score. It then performs a Maximal Independent Set (MIS) selection on the remaining tokens, which are modeled on a hybrid graph where edges signify spatial proximity and semantic similarity. This process iteratively preserves the most important and available token while removing its neighbors, ensuring that the supplementary tokens are chosen to maximize importance and diversity. Extensive experiments demonstrate that D2Pruner has exceptional efficiency and fidelity. Applied to LLaVA-1.5-7B for general understanding tasks, it reduces FLOPs by 74.2\% while retaining 99.2\% of its original performance. Furthermore, in challenging localization benchmarks with InternVL-2.5-8B, it maintains 85.7\% performance at a 90\% token reduction rate, marking a significant advancement with up to 63. 53\% improvement over existing methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。