← 返回 2026-03-16

边看边想:面向多模态大语言模型多轮视频推理的在线流式分段级记忆机制 Think While Watching: Online Streaming Segment-Level Memory for Multi-Turn Video Reasoning in Multimodal Large Language Models

Lu Wang, Zhuoran Jin, Yupu Hao, Yubo Chen, Kang Liu, Yulong Ao, Jun Zhao 📅 2026-03-12 👍 10 2026-07-13 08:36
在线学习 多模态大语言模型 流式推理 视频理解 记忆机制

提出分段级记忆机制实现流式视频多轮问答,解耦感知与生成

前置知识

多模态大语言模型 (MLLMs)

多模态大语言模型是指能够同时处理多种模态信息(如文本、图像、视频)的大规模语言模型。这类模型通常基于Transformer架构,通过视觉编码器将图像或视频帧转换为token序列,然后与文本token一起输入到语言模型中进行联合推理。在视频理解场景中,MLLMs需要处理时间序列的视觉信息,并结合自然语言指令进行理解和推理,是实现视频问答、视频描述等任务的核心技术。

理解MLLMs的工作原理是理解本文提出方法的基础,因为整个系统都建立在MLLMs之上进行视觉和语言的联合理解

流式视频处理

流式视频处理是指在视频内容持续到达的场景下进行实时处理,而不是等待整个视频加载完毕后再进行分析。这种处理方式适用于直播监控、机器人助手、实时视频会议等场景。流式处理的关键挑战在于系统必须在有限的历史信息下做出决策,同时保持对长期依赖关系的建模能力,并且要保证低延迟的响应速度。

本文的核心问题就是如何在流式场景下进行有效的多轮视频推理,理解流式处理的特点和挑战是理解问题动机的关键

KV缓存

KV缓存是Transformer模型推理时的一种优化技术,它存储已计算过的键(Key)和值(Value)向量,避免在自回归生成过程中重复计算历史token的信息。在标准的Transformer解码中,每生成一个新token都需要计算该token与所有历史token的注意力分数,KV缓存通过存储历史计算结果使得新token只需要计算与自己的注意力,大大提高了推理效率。在多模态场景中,KV缓存的管理变得更加复杂,因为需要处理不同模态的token序列。

本文提出的双KV缓存机制是实现并行推理的关键技术,理解KV缓存的工作原理有助于理解推理优化的核心思想

注意力掩码 (Attention Mask)

注意力掩码是Transformer模型中用于控制注意力机制可见性的技术。在自回归生成任务中,通常使用因果掩码确保每个位置只能关注其之前的token,防止信息从未来泄露到过去。在标准的因果掩码中,位置i可以关注所有位置j不超过i,形成一个下三角矩阵。本文提出的流式掩码则更加复杂,需要同时处理输入序列和输出序列之间的可见性关系,确保在流式场景下每个生成的token只能看到已经到达的输入片段。

本文提出的分段级流式因果掩码是保证流式推理严格因果性的核心技术,理解标准因果掩码有助于理解本文的创新

链式思维 (Chain-of-Thought, CoT)

链式思维是一种提示技术,要求模型在给出最终答案之前先展示中间推理步骤。通过让模型逐步分解问题、逐步推理,可以提高复杂任务的准确性。在视频理解场景中,CoT可以帮助模型逐步分析视频内容、提取关键信息、建立时间关系,最终得出答案。本文将CoT引入流式视频推理,设计了分段级的记忆笔记作为中间推理产物,使模型能够在线地记录和利用推理过程中的关键信息。

本文构建了三阶段CoT数据集用于训练,理解CoT的概念有助于理解训练策略的设计

研究动机

当前多模态大语言模型在离线视频理解任务上表现出色,但在实际应用中,许多高价值场景本质上是流式的:直播监控、机器人助手、实时视频会议等。在这些场景中,用户可能在任何时间提问,助手必须实时回答同时忠实于已观察到的视觉证据。现有的流式方法采用交替感知-生成范式(interleaved perception-generation paradigm),这种方法存在两个根本性问题。首先是记忆侵蚀(Memory Erosion):多轮对话中后续问题经常引用早期问题或早期视觉线索,但与生成交替进行的方式往往会削弱长期记忆能力。实验数据显示,对于Qwen3-VL-4B Thinking模型,在线多轮设置的准确率相比离线设置下降了40.39%,这凸显了维持长期时间一致性的严峻挑战。其次是序列化瓶颈(Serialization Bottleneck):自回归模型使用统一的位置编码,新输入必须与长度未知的生成输出对齐,迫使输入暂停,造成序列化瓶颈。随着轮次累积,输入不断堆积,导致端到端延迟不断增加。

本文的目标是本文旨在设计一个能够在持续到达的视频流上进行在线多轮交互的流式视频推理框架。具体目标包括:第一,维持分段级记忆作为持久状态,通过隐式检索和整合相关记忆笔记来回答每个查询,提高多轮一致性;第二,解耦感知和生成,通过独立的位置编码实现输入输出并行化,从而降低延迟;第三,设计实用的训练和推理流程,包括三阶段训练、流式分段掩码、流式位置编码以及双KV缓存推理,支持感知和生成之间的并行化。

与已有工作不同的是,本文的独特切入角度在于将记忆写入作为显式的在线行为:对于每个观察到的片段,模型写入一个记忆笔记并将其添加到记忆库;当问题到达时,模型通过注意力机制隐式整合相关笔记进行回答。这种方法与现有的交替处理方式本质不同,它将记忆维护从被动的信息保留转变为主动的知识提取和存储。同时,通过为输入和输出流分配独立的位置编码,打破了自回归模型中输入必须与输出对齐的限制,实现了真正的并行处理。这种解耦设计借鉴了操作系统中的CPU进程调度思想,将推理组织为多阶段流水线,实现了感知和生成的并行化。

核心方法

Think While Watching是一个基于记忆锚定的流式视频推理框架,其核心思想是在观看视频的同时持续维护分段级记忆,并通过注意力机制隐式检索相关记忆来回答问题。整体架构包含三个关键组件:分段级记忆笔记系统用于在线记录视频内容的关键信息;流式注意力掩码和位置编码确保严格的因果性;双KV缓存流水线实现感知和生成的并行化。与传统的交替处理方法不同,本方法将视频处理分解为连续的片段,每个片段到达时模型立即生成一个记忆笔记,记录关键实体、属性、动作和交互等信息。当用户提问时,模型通过注意力机制从记忆库中隐式选择最相关的笔记进行回答,同时继续处理新到达的视频片段。这种设计使得系统能够在持续观看视频的同时进行思考和回答,大大降低了响应延迟。

本文的核心创新在于两个方面。第一,分段级记忆笔记作为持久状态:不同于现有方法将视频信息直接传递给生成器,本方法让模型为每个视频片段生成一个紧凑的记忆笔记,这些笔记存储在记忆库中作为跨轮次的持久状态。记忆笔记通过注意力机制隐式参与后续的问答过程,使得模型能够在长期交互中保持对早期信息的记忆。第二,流式位置编码实现输入输出解耦:通过为输入流和输出流分配独立的位置索引,打破了自回归模型中输入必须与输出对齐的限制。输入流遵循标准的累积偏移方案,而输出流独立从0开始位置编码,这样即使输出长度未知,新到达的输入片段也能被分配正确的位置。这种设计借鉴了操作系统中的进程调度思想,实现了感知和生成的真正并行化。

方法步骤详情

方法包含三个主要步骤。第一步是分段级记忆生成:将到达的视频流分割为连续片段 $S_1, S_2, ..., S_T$,每个片段 $S_t$ 到达时,模型生成一个记忆笔记 $m_t = \text{Mem}_\theta[S_t]$,记录关键实体、动作、场景变化等信息,这些笔记存储在记忆库 $M_t = \{⟨i, m_i⟩\}_{i=1}^t$ 中。第二步是流式注意力掩码设计:定义分段级掩码 $M^{seg}[A, B]$,确保生成单元 $C_u$ 只能关注到达索引不超过 $u$ 的接收单元和生成单元,实现严格的流式因果性。第三步是流式位置编码:使用MRoPE的扩展版本,为输入流计算累积偏移 $B^{S_k}$ 和 $B^{Q_k}$,为输出流独立计算偏移 $B^{C_k}$,实现输入输出的位置解耦。推理时采用双KV缓存和自适应注意力后端选择,当 $q_{len} = k_{len}$ 时使用Flash Attention,当 $1 < q_{len} < k_{len}$ 时使用内存高效注意力。

技术新颖性

本文的技术新颖性体现在多个层面。首先,在架构层面提出了分段级记忆笔记的概念,将传统的被动信息保留转变为主动的知识提取和存储,这在流式视频理解中是首创。其次,在位置编码层面,提出了流式MRoPE,通过独立的输入输出位置索引实现了自回归模型中前所未有的输入输出解耦能力。第三,在训练层面设计了三阶段训练策略:阶段1学习写记忆笔记和单轮问答;阶段2扩展到多轮对话;阶段3针对长期能力进行训练,包括长期记忆、不确定性和干扰片段学习。第四,在推理层面,借鉴操作系统进程调度的思想,将推理组织为多阶段流水线,通过双KV缓存实现感知和生成的并行化。第五,构建了首个三阶段流式CoT数据集,填补了多轮流式视频CoT数据的空白。

训练组件设计
Figure 2: 训练组件设计

实验结果

实验结果验证了Think While Watching框架的有效性。在StreamingBench上,基于Qwen3-VL-4B的单轮准确率从58.52%提升到60.04%,提升了1.52个百分点;在OVO-Bench上从50.70%提升到55.02%,提升了4.32个百分点。更重要的是,在多轮协议下,框架在保持竞争力准确率的同时大幅减少了输出token数量:在StreamingBench上token使用量从689.22减少到302.56,减少了56.10%;在OVO-Bench上从472.18减少到255.91,减少了45.80%。在8B模型上,单轮准确率提升更显著:StreamingBench提升3.83个百分点达到62.04%,OVO-Bench提升3.25个百分点达到57.07%。记忆消融实验表明,移除记忆笔记会导致准确率从57.40%下降到52.35%,证实了记忆笔记作为持久状态的有效性。注意力分析显示,阶段3训练后模型的注意力分布从近历史段转移到更远的历史段,特别是在记忆token上更加明显。TTFT分析表明,与批量Thinking相比,流式流水线将TTFT从31203.69个token减少到2304.28个token,减少了92.6%。

三阶段训练数据集统计
Table 1: 三阶段训练数据集统计
StreamingBench评估结果
Table 2: StreamingBench评估结果
OVO-Bench评估结果
Table 3: OVO-Bench评估结果
离线视频理解结果
Table 4: 离线视频理解结果
记忆和片段粒度消融实验
Table 5: 记忆和片段粒度消融实验
TTFT分析结果
Table 6: TTFT分析结果
答案注意力与片段距离的关系
Figure 3: 答案注意力与片段距离的关系
帧掩码消融实验
Figure 4: 帧掩码消融实验
查看结构化数据
任务指标本文基线提升
StreamingBench单轮视频理解 Overall Accuracy TWWsingle-turn,S3: 60.04% Thinking: 58.52% +1.52%
OVO-Bench单轮视频理解 Overall Accuracy TWWsingle-turn,S3: 55.02% Thinking: 50.70% +4.32%
StreamingBench多轮流式推理 Avg Tokens TWWmulti-turn,S3: 302.56 Thinking: 689.22 Token Reduce: 56.10%
OVO-Bench多轮流式推理 Overall Accuracy TWWmulti-turn,S3: 51.80% Thinkingonline: 16.21% +35.59%
Video-MME离线视频理解 Overall Accuracy TWWsingle-turn,S3: 73.41% Thinking: 68.89% +4.52%
LV-Bench离线视频理解 Overall Accuracy TWWsingle-turn,S3: 57.68% Thinking: 53.47% +4.21%

局限与改进

尽管本文取得了显著成果,但仍存在一些局限性。首先,分段粒度的选择需要权衡准确率和效率:较长的片段(120s/60s)会降低准确率2.07%,而较短的片段(30s/15s)会增加25.8%的解码长度。其次,当前的记忆笔记生成依赖于GPT-5.2进行数据合成,这可能限制了数据的多样性和可控性。第三,实验主要在Qwen3-VL系列模型上进行验证,对于其他架构的泛化能力还需要进一步研究。第四,虽然框架支持多轮交互,但在极长视频(超过1小时)上的性能还需要更全面的评估。第五,当前的干扰片段学习只在阶段3训练中引入,对于动态场景中实时出现的干扰物的鲁棒性可能有限。此外,框架的实时性能还受限于硬件加速能力和模型大小,在资源受限的边缘设备上部署可能面临挑战。

独立分析的弱点

本文存在几个可以改进的弱点。首先,记忆笔记的质量直接影响后续问答的准确性,但当前方法没有对记忆笔记的质量进行显式评估或优化,可能存在信息丢失或关键信息遗漏的问题。可以考虑引入记忆质量评估机制或自适应的记忆粒度调整。其次,分段策略采用固定时长的硬分割,没有考虑语义边界,可能导致语义完整性被破坏。可以探索基于场景变化或动作边界的自适应分割方法。第三,当前的记忆机制是纯文本的,丢失了视觉细节信息,对于需要精确视觉推理的任务可能不足。可以考虑引入多模态记忆表示。第四,训练数据合成依赖外部大模型,成本较高且可能引入偏差。可以探索更高效的数据增强或自监督学习方法。第五,推理时的自适应注意力后端选择增加了系统复杂性,可能存在次优选择的情况。可以考虑学习一个轻量级的选择策略。

未来方向

基于本文的成果,未来研究可以在多个方向展开。首先,可以探索更精细的记忆粒度,如对象级或事件级记忆,而不是当前的片段级记忆,以支持更细粒度的推理。其次,可以将框架扩展到多模态输入,如结合音频信息进行视听联合理解。第三,可以研究记忆的主动遗忘机制,通过重要性评估动态删除过时或不相关的信息,提高长期交互的效率。第四,可以探索分布式记忆架构,将记忆分布在多个位置或多个模型实例中,支持更大规模的并发交互。第五,可以研究记忆的可解释性,让用户能够理解和查询模型的记忆内容,提高系统的透明度和可信度。此外,可以将流式推理框架应用于其他领域,如实时语音理解、流式文档分析等。最后,可以探索与强化学习的结合,通过在线学习持续优化记忆策略和推理能力。

复现评估

本文在可复现性方面提供了较好的支持。作者开源了代码实现,包括完整的训练和推理流程。三阶段训练数据集基于公开可用的VideoChatOnline-IT和YouTube数据构建,并提供了详细的数据生成流程和提示词模板。实验设置清晰,使用了标准的评估基准(StreamingBench、OVO-Bench)和明确的评估协议。然而,数据合成依赖GPT-5.2等闭源模型,可能影响完全复现的成本。算力需求方面,训练在多个Qwen3-VL规模(2B、4B、8B)上进行,需要中等规模的GPU集群。推理时的双KV缓存机制需要额外的内存开销,但相比传统方法有明显的延迟优势。总体而言,对于有一定资源的研究团队,本文的结果应该是可复现的。