解决视频大模型注意力偏差,高效压缩视觉令牌
AdaTP: Attention-Debiased Token Pruning for Video Large Language Models
- 引入去偏模块,分别纠正全局与局部注意力偏差
- 仅用27.3%计算量,保持原模型性能不下降
- 无需额外训练,适合追求效率的视频理解应用
视频大语言模型在视频理解任务中表现优异,但因多帧生成大量视觉令牌而带来沉重计算负担。现有压缩方法依赖语言模型的注意力分数引导,但该分数存在固有偏差:全局偏差导致模型偏向序列两端,局部偏差使模型过度关注各帧相同空间位置。为此,我们提出AdaTP——一种针对视频大模型的注意力去偏令牌剪枝方法。该方法在剪枝流程中集成两个专用去偏模块,分别缓解全局与局部注意力偏差。无需额外训练,显著降低计算开销并保持原始模型性能。大量实验表明,AdaTP在多个主流视频理解基准上达到领先水平。例如,在LLaVA-OneVision-7B上,仅需27.3%的浮点运算量(FLOPs),即可维持原模型性能。代码即将开源。
原文摘要 · Abstract (English)
Video Large Language Models (Video LLMs) have achieved remarkable results in video understanding tasks. However, they often suffer from heavy computational overhead due to the large number of visual tokens generated from multiple video frames. Existing visual token compression methods often rely on attention scores from language models as guidance. However, these scores exhibit inherent biases: global bias reflects a tendency to focus on the two ends of the visual token sequence, while local bias leads to an over-concentration on the same spatial positions across different frames. To address the issue of attention bias, we propose $\textbf{A}$ttention-$\textbf{D}$ebi$\textbf{a}$sed $\textbf{T}$oken $\textbf{P}$runing for Video Large Language Models ($\textbf{AdaTP}$), a novel token pruning pipeline for Video LLMs. AdaTP integrates two dedicated debiasing modules into the pipeline, targeting global attention bias and local attention bias, respectively. Without the need for additional training, our method significantly reduces the computational overhead of Video LLMs while retaining the performance of vanilla models. Extensive evaluation shows that AdaTP achieves state-of-the-art performance in various commonly used video understanding benchmarks. In particular, on LLaVA-OneVision-7B, AdaTP maintains performance without degradation while using only up to $27.3\%$ FLOPs compared to the vanilla model. Our code will be released soon.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。