让大模型像人一样看长视频,突破帧数限制。
Scaling the Long Video Understanding of Multimodal Large Language Models via Visual Memory Mechanism
- 用视觉记忆机制分步处理视频,不一次加载全部内容。
- 单张3090显卡下可处理超1000帧视频,性能超越现有方法。
- 无需训练,适配长视频与直播类任务,适合研究者快速部署。
长视频理解是制约多模态大语言模型(MLLMs)发展的关键挑战。本文从视觉记忆机制出发,提出一种新颖且无需训练的方法——柔性记忆(FlexMem)。该方法模拟人类观看视频时持续接收信息并调用相关记忆的模式,使MLLM能实现无限长度的视频理解,突破传统方法一次性处理全部视频内容的输入上限。具体而言,FlexMem将视觉键值缓存作为记忆源,通过双路径压缩设计实现高效的记忆传递与写入;同时针对不同任务探索多种记忆读取策略,包括主流的流式处理方式。我们在两个主流视频-MLLM上验证了FlexMem,涵盖五个长视频任务和一个流式视频任务。实验表明,在单张3090显卡下,该方法显著优于现有高效视频理解方案,可处理超过1000帧视频,并使基础模型在部分基准上达到甚至超越GPT-4o、Gemini-1.5 Pro等顶级模型的性能。
原文摘要 · Abstract (English)
Long video understanding is a key challenge that plagues the advancement of \emph{Multimodal Large language Models} (MLLMs). In this paper, we study this problem from the perspective of visual memory mechanism, and proposed a novel and training-free approach, termed \emph{Flexible Memory} (\textbf{FlexMem}). In principle, FlexMem aims to mimic human behavior of video watching, \emph{i.e.}, continually watching video content and recalling the most relevant memory fragments to answer the question. In this way, FlexMem can help MLLMs achieve video understanding of infinite lengths, unlike previous methods that process all video information at once and have input upper-limit. Concretely, FlexMem first consider the visual KV caches as the memory sources, and realize the effective memory transfer and writing via a dual-pathway compression design. Afterwards, FlexMem also explores different memory reading strategies for the diverse video understanding tasks, including the popular streaming one. To validate FlexMem, we apply it to two popular video-MLLMs, and conduct extensive experiments on five long video and one streaming video task. The experimental results show that on \textbf{a single 3090 GPU}, our FlexMem can achieve obvious improvements than existing efficient video understanding methods and process more than \textbf{1k frames}, which also helps the base MLLMs achieve comparable or even better performance than SOTA MLLMs on some benchmarks, \emph{e.g.} , GPT-4o and Gemini-1.5 Pro.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。