ReTaKe通过减少时间与知识冗余,让视频大模型能处理8倍长的视频。
ReTaKe: Reducing Temporal and Knowledge Redundancy for Long Video Understanding
- 基于帧间距离峰值选关键帧,结合大模型先验知识压缩非关键帧。
- 可处理最多2048帧视频,性能比同类模型高3-5%,接近更大模型。
- 无训练开销,推理延迟降低20%,预填充额外开销仅10%。
视频大语言模型(VideoLLMs)在视频理解上取得进展,但受限于骨干语言模型,在长视频处理上表现不佳。现有方法依赖长度外推(受内存限制)或视觉标记压缩(主要利用低层时间冗余,忽略更有效的高层知识冗余)。为此,我们提出无需训练的ReTaKe方法,包含两个新模块:DPSelect和PivotKV,协同减少时间视觉冗余与知识冗余以实现视频压缩。为模拟人类时间感知,DPSelect基于帧间距离峰值识别关键帧;为利用大模型学习到的先验知识,PivotKV将关键帧设为枢纽,并通过剪枝其键值缓存中的低注意力标记来压缩非枢纽帧。ReTaKe使视频大模型可处理长达2048帧的视频(较以往提升8倍),在VideoMME、MLVU、LongVideoBench和LVBench上性能优于同规模模型3-5%,甚至媲美更大模型。此外,通过重叠压缩操作与预填充阶段,仅引入约10%预填充延迟,同时降低约20%解码延迟。代码已开源。
原文摘要 · Abstract (English)
Video Large Language Models (VideoLLMs) have made significant strides in video understanding but struggle with long videos due to the limitations of their backbone LLMs. Existing solutions rely on length extrapolation, which is memory-constrained, or visual token compression, which primarily leverages low-level temporal redundancy while overlooking the more effective high-level knowledge redundancy. To address this, we propose $\textbf{ReTaKe}$, a training-free method with two novel modules DPSelect and PivotKV, to jointly reduce both temporal visual redundancy and knowledge redundancy for video compression. To align with the way of human temporal perception, DPSelect identifies keyframes based on inter-frame distance peaks. To leverage LLMs' learned prior knowledge, PivotKV marks the keyframes as pivots and compress non-pivot frames by pruning low-attention tokens in their KV cache. ReTaKe enables VideoLLMs to process 8 times longer frames (up to 2048), outperforming similar-sized models by 3-5% and even rivaling much larger ones on VideoMME, MLVU, LongVideoBench, and LVBench. Moreover, by overlapping compression operations with prefilling, ReTaKe introduces only ~10% prefilling latency overhead while reducing decoding latency by ~20%. Our code is available at https://github.com/SCZwangxiao/video-ReTaKe.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。