用时空压缩技术让大模型高效理解长视频,提速8倍还不丢关键信息。
STORM: Token-Efficient Long Video Understanding for Multimodal LLMs
- 引入Mamba模型做时序编码,融合帧间动态信息
- 长视频任务上性能提升超5%,计算量降低8倍
- 适合需要处理超长视频的AI研究与工业应用
近期基于视频的多模态大语言模型(Video-LLMs)通过将视频视为图像帧序列显著提升了视频理解能力。然而,许多现有方法在视觉主干网络中独立处理帧,缺乏显式的时序建模,限制了对动态模式的捕捉和长视频的高效处理。为此,我们提出STORM(Spatiotemporal TOken Reduction for Multimodal LLMs),一种在图像编码器与大语言模型之间加入专用时序编码器的新架构。该时序编码器采用Mamba状态空间模型,将时序信息融入图像标记,生成蕴含帧间动态关系的增强表示。这种丰富编码不仅提升视频推理能力,还支持高效的标记压缩策略,包括测试时采样和训练阶段的时空池化,大幅降低大模型的计算开销而不损失关键时序信息。结合这些技术,本方法在保持高性能的同时,显著降低训练与推理延迟,实现对长时序上下文的高效鲁棒理解。大量实验表明,STORM在多个长视频理解基准上达到顶尖水平(MLVU和LongVideoBench提升超5%),同时在固定输入帧数下计算成本降低达8倍,解码延迟减少2.4–2.9倍。
原文摘要 · Abstract (English)
Recent advances in video-based multimodal large language models (Video-LLMs) have significantly improved video understanding by processing videos as sequences of image frames. However, many existing methods treat frames independently in the vision backbone, lacking explicit temporal modeling, which limits their ability to capture dynamic patterns and efficiently handle long videos. To address these limitations, we introduce STORM (Spatiotemporal TOken Reduction for Multimodal LLMs), a novel architecture incorporating a dedicated temporal encoder between the image encoder and the LLM. Our temporal encoder leverages the Mamba State Space Model to integrate temporal information into image tokens, generating enriched representations that preserve inter-frame dynamics across the entire video sequence. This enriched encoding not only enhances video reasoning capabilities but also enables effective token reduction strategies, including test-time sampling and training-based temporal and spatial pooling, substantially reducing computational demands on the LLM without sacrificing key temporal information. By integrating these techniques, our approach simultaneously reduces training and inference latency while improving performance, enabling efficient and robust video understanding over extended temporal contexts. Extensive evaluations show that STORM achieves state-of-the-art results across various long video understanding benchmarks (more than 5% improvement on MLVU and LongVideoBench) while reducing the computation costs by up to $8\times$ and the decoding latency by 2.4-2.9$\times$ for the fixed numbers of input frames. Project page is available at https://research.nvidia.com/labs/lpr/storm
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。