通过关键帧导向的视觉令牌剪枝,提升大模型处理长视频效率
Keyframe-oriented Vision Token Pruning: Enhancing Efficiency of Large Vision Language Models on Long-Form Video Processing
- 根据帧与查询的相关性自适应分配剪枝率,保留上下文连续性
- 在不损失时空一致性前提下,减少80%视觉令牌使用量
- 适合需要高效处理长视频的大模型应用开发者
视觉语言模型(VLMs)在联合处理视觉与文本数据方面表现出强大能力,但在长视频场景中常因冗余视觉信息导致显著计算开销。现有方法或侧重视觉令牌剪枝(可能忽略时空依赖),或依赖关键帧选择(识别重要帧但丢弃其余帧,破坏上下文连续性)。本文提出KVTP(Keyframe-oriented Vision Token Pruning)框架,克服上述局限。通过基于帧与查询相关性的自适应剪枝率分配,有效保留关键上下文信息,同时大幅减少冗余计算。为全面评估VLM在长视频理解上的能力,我们从VideoMME、EgoSchema和NextQA中提取并重组子集,构建统一基准SparseKV-QA,突出现实世界中稀疏但关键事件的场景。多尺度VLM实验表明,KVTP可在不牺牲时空与上下文一致性的前提下,将令牌使用量降低80%,显著降低计算成本并保持性能。结果验证了该方法在高效长视频处理中的有效性,推动VLM更广泛部署。
原文摘要 · Abstract (English)
Vision language models (VLMs) demonstrate strong capabilities in jointly processing visual and textual data. However, they often incur substantial computational overhead due to redundant visual information, particularly in long-form video scenarios. Existing approaches predominantly focus on either vision token pruning, which may overlook spatio-temporal dependencies, or keyframe selection, which identifies informative frames but discards others, thus disrupting contextual continuity. In this work, we propose KVTP (Keyframe-oriented Vision Token Pruning), a novel framework that overcomes the drawbacks of token pruning and keyframe selection. By adaptively assigning pruning rates based on frame relevance to the query, KVTP effectively retains essential contextual information while significantly reducing redundant computation. To thoroughly evaluate the long-form video understanding capacities of VLMs, we curated and reorganized subsets from VideoMME, EgoSchema, and NextQA into a unified benchmark named SparseKV-QA that highlights real-world scenarios with sparse but crucial events. Our experiments with VLMs of various scales show that KVTP can reduce token usage by 80% without compromising spatiotemporal and contextual consistency, significantly cutting computation while maintaining the performance. These results demonstrate our approach's effectiveness in efficient long-video processing, facilitating more scalable VLM deployment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。