通过任务感知的键值稀疏化,实现超长视频高效理解
Video-XL-2: Towards Very Long-Video Understanding Through Task-Aware KV Sparsification
- 分块预填充+双层键值解码,动态控制注意力计算范围
- 单卡处理超1万帧视频,数千帧仅需数秒完成推理
- 适合需要高精度长视频分析的研究与工业场景
多模态大语言模型在视频理解领域近年取得显著进展,但处理长视频仍面临高昂的内存与计算成本。为此,我们提出Video-XL-2,一种基于任务感知键值稀疏化的新型多模态大语言模型,显著提升长视频理解的性价比。该框架包含两个关键步骤:分块预填充将视觉标记序列分块,块内使用全注意力,块间采用稀疏注意力,大幅降低计算与内存开销;解码阶段采用双层键值解码,根据任务相关性选择性重载稠密或稀疏键值,进一步提升内存效率并增强细粒度信息捕捉能力。Video-XL-2在多个长视频理解基准上达到当前最优性能,优于现有开源轻量级模型。同时展现出卓越效率,可在单张NVIDIA A100(80GB)GPU上处理超过10,000帧视频,数千帧视频仅需数秒完成推理。
原文摘要 · Abstract (English)
Multi-modal large language models (MLLMs) models have made significant progress in video understanding over the past few years. However, processing long video inputs remains a major challenge due to high memory and computational costs. This makes it difficult for current models to achieve both strong performance and high efficiency in long video understanding. To address this challenge, we propose Video-XL-2, a novel MLLM that delivers superior cost-effectiveness for long-video understanding based on task-aware KV sparsification. The proposed framework operates with two key steps: chunk-based pre-filling and bi-level key-value decoding. Chunk-based pre-filling divides the visual token sequence into chunks, applying full attention within each chunk and sparse attention across chunks. This significantly reduces computational and memory overhead. During decoding, bi-level key-value decoding selectively reloads either dense or sparse key-values for each chunk based on its relevance to the task. This approach further improves memory efficiency and enhances the model's ability to capture fine-grained information. Video-XL-2 achieves state-of-the-art performance on various long video understanding benchmarks, outperforming existing open-source lightweight models. It also demonstrates exceptional efficiency, capable of processing over 10,000 frames on a single NVIDIA A100 (80GB) GPU and thousands of frames in just a few seconds.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。