← 返回 2026-03-30

视线之外但记忆犹存:面向动态视频世界模型的混合记忆 Out of Sight but Not Out of Mind: Hybrid Memory for Dynamic Video World Models

Kaijin Chen, Dingkang Liang, Xin Zhou, Yikang Ding, Xiaoqiang Liu, Pengfei Wan, Xiang Bai 📅 2026-03-26 👍 157 2026-07-13 08:36
3D场景 世界模型 时空一致性 视频生成 记忆机制

提出混合记忆范式,让视频世界模型追踪离屏动态主体

前置知识

Video World Model

视频世界模型是一类能够模拟真实物理世界的生成模型,通过视频生成的方式预测未来帧。这类模型通常基于扩散模型(Diffusion Model)架构,结合3D VAE进行时空压缩,在自动驾驶、具身智能等领域有重要应用。其核心挑战在于如何在长时间生成过程中保持场景的空间和时间一致性,即生成的视频要符合物理规律且前后连贯。

本文的核心问题——动态主体在视野外消失后重新出现时的不一致性——正是视频世界模型面临的关键挑战之一,理解这一背景才能把握本文的研究动机。

Memory Mechanism in Video Generation

视频生成中的记忆机制是指模型在生成过程中如何存储和利用历史上下文信息的方法。常见的实现包括:3D surfel索引记忆(VMem)、视野重叠记忆(Context-as-Memory)、可更新语义包(MemoryPack)等。这些机制的核心目标是在长时间视频生成中保持场景一致性,避免出现物体消失、背景突变等问题。

本文提出的混合记忆范式正是对现有记忆机制的根本性改进,理解传统记忆机制的工作方式才能理解本文的创新所在。

Diffusion Transformer (DiT)

DiT是将Transformer架构与扩散模型结合的生成模型架构。在视频生成中,DiT通过3D注意力机制处理时空特征,逐步将噪声去噪为清晰的视频帧。本文的基线模型Wan2.1-T2V-1.3B就是基于DiT架构构建的,其中每个DiT块包含自注意力、交叉注意力和前馈网络。

本文的HyDRA方法通过替换DiT中的标准3D自注意力层来实现动态检索注意力,理解DiT架构才能理解这一技术改造的细节。

Top-K Retrieval

Top-K检索是一种从候选集合中选择最相关K个元素的策略。在本文中,模型首先计算目标查询与所有记忆token之间的时空亲和度分数,然后选择分数最高的K个记忆token用于注意力计算。这种方法避免了使用全部记忆信息带来的计算开销和噪声干扰,是一种高效的信息筛选机制。

HyDRA的动态检索注意力核心就是基于Top-K策略从记忆中筛选最相关的运动和外观线索,这是本文方法的关键技术组件。

Unreal Engine 5 数据渲染

Unreal Engine 5(UE5)是Epic Games开发的游戏引擎,具有高度逼真的实时渲染能力。本文使用UE5构建数据渲染流水线,通过组合3D场景、主体、主体轨迹和相机轨迹四个维度来生成包含进出画面事件的视频数据。这种方法可以精确控制相机和主体的运动,产生自然界中难以捕获的特定动态场景。

HM-World数据集的构建完全依赖UE5渲染流水线,理解这一技术才能理解数据集的设计理念和质量保证。

研究动机

现有视频世界模型的记忆机制存在一个根本性盲点:将世界视为静态画布。以VMem、Context-as-Memory、MemoryPack等代表性方法为例,它们在处理静态背景(建筑、道路等不动元素)时表现出色,能够通过3D surfel索引、视野重叠或语义压缩等技术保持场景一致性。然而,真实物理世界是由大量动态主体构成的——行走的行人、奔跑的动物等,这些主体遵循各自独立的运动逻辑。当动态主体隐藏到相机视野之外时,现有模型会彻底丢失对它们的追踪,导致主体在重新进入画面时呈现为冻结的雕像、扭曲的幻影,或者直接消失不见。这一问题在涉及相机运动(如平移、旋转)的场景中尤为严重,因为相机运动会频繁导致主体进出画面。现有的世界模型在处理这种动态进出事件时几乎完全失效,成为制约视频世界模型实际应用的关键瓶颈。

本文的目标是本文的核心目标是提出一种全新的记忆范式——混合记忆(Hybrid Memory),要求模型同时扮演精确的静态背景档案管理员和警觉的动态主体追踪者。具体而言,混合记忆需要在两个层面同时工作:第一,在静态层面,模型需要像传统方法一样精确记忆和重建不动的背景环境,确保相机回到之前位置时场景完全一致;第二,在动态层面,模型需要持续追踪和预测动态主体的运动,即使主体暂时离开视野也要保持对其身份和运动状态的记忆。当主体重新进入画面时,模型不仅要保持其原始视觉身份,还要在合理的位置以一致的运动状态出现。为实现这一目标,本文同时构建了专门的数据集HM-World和专门的架构HyDRA,从数据和方法两个维度系统性地解决混合记忆问题。

与已有工作不同的是,本文的独特切入角度体现在三个层面的创新:首先是范式层面的突破,现有研究都在静态记忆的框架内优化,而本文首次明确提出了混合记忆这一新范式,将动态主体追踪纳入记忆机制的核心要求;其次是数据层面的创新,现有数据集要么只包含静态场景(如Context-as-Memory数据集),要么动态主体只在原地动作(如Multi-Cam Video),要么相机固定不动(如360°-Motion),缺乏专门用于研究混合记忆的大规模数据集,本文构建的HM-World填补了这一空白;最后是方法层面的创新,现有方法直接在原始潜在表示上进行检索,缺乏对运动信息的显式建模,本文提出的记忆tokenizer通过3D卷积实现时空交互,生成具有更丰富运动感知的token表示,并通过动态亲和度检索实现针对性的信息召回。这种从数据到方法的系统性创新是本文区别于已有工作的核心所在。

核心方法

本文的方法整体思路可以分为三个层次理解。从直觉层面看,混合记忆的核心是让模型具备'视线之外的心理模拟能力'——当主体离开画面时,模型需要在潜空间中持续想象和预测主体的运动轨迹,而非简单地遗忘。技术路线上,本文构建了一个完整的系统:首先通过HM-World数据集提供训练和评估的数据基础,该数据集通过UE5渲染流水线生成包含丰富进出画面事件的视频;然后在Wan2.1-T2V-1.3B基线模型上进行架构改造,引入三个核心组件:记忆tokenizer将历史帧压缩为具有运动感知能力的token、动态检索注意力机制实现针对性的信息召回、相机编码器实现精确的空间控制。整个流程是:上下文帧经过3D VAE编码后送入记忆tokenizer生成记忆token,同时目标帧的去噪潜在表示作为查询,通过计算查询与记忆token之间的时空亲和度来选择最相关的历史信息,最终将检索到的记忆信息与局部时序信息融合后用于生成。

本文的核心创新点在于提出了一种'压缩-检索'的记忆架构,与已有方法存在本质区别。传统方法如Context-as-Memory直接基于视野重叠进行记忆检索,这种方法仅利用静态几何关系,无法捕捉动态主体的运动信息;MemoryPack虽然引入了语义压缩,但其更新机制是全局性的,缺乏对特定主体的针对性记忆。本文的关键洞察是:原始潜在表示中的静态背景特征和动态主体特征高度耦合,直接检索会导致主体冻结或失真。因此,本文引入记忆tokenizer这一创新组件,通过3D卷积在时空维度上同时进行交互,将耦合的潜在表示解耦为具有运动感知能力的记忆token。这种token表示不仅包含外观信息,还编码了时序运动模式。检索时,本文采用动态亲和度方法,通过计算查询与记忆token之间的元素级内积来量化时空对应关系,而非依赖静态的视野重叠。这种基于注意力机制的细粒度检索能够更准确地定位与当前目标最相关的历史运动线索。

方法步骤详情

本文方法的完整流程包含以下步骤:第一步是相机条件注入,将相机轨迹序列 $P = \{(R_i, t_i)\}_{i=1}^f$ 展平拼接为统一的相机条件 $c_{cam} \in \mathbb{R}^{f \times 12}$,通过MLP相机编码器 $\mathcal{E}_{cam}(\cdot)$ 编码后广播到空间维度,与潜在特征进行逐元素相加:$H_{out} = H_{in} + \mathcal{E}_{cam}(c_{cam})$。第二步是记忆token化,将编码后的记忆潜在表示 $Z_{mem}$ 送入基于3D卷积的记忆tokenizer $\mathcal{T}_{mem}$,通过核大小为 $2 \times 4 \times 4$ 的3D卷积同时处理时空维度,生成紧凑的记忆token $M \in \mathbb{R}^{C' \times f'_{mem} \times h \times w}$。第三步是动态检索注意力,对于每个目标潜在查询 $q_i$,首先通过空间池化下采样到与记忆token相同的空间分辨率,然后计算与每个记忆token切片的时空亲和度 $S_{i,j} = \frac{1}{\sqrt{d}} \sum_{y=1}^{h} \sum_{x=1}^{w} \tilde{q}_i(x,y) \cdot k_{mem,j}(x,y)$,选择Top-K个最相关的记忆token。第四步是注意力计算,将检索到的记忆key/value与局部时序窗口的key/value拼接后,执行标准注意力计算:$\text{Attention}(q_i, K'_i, V'_i) = \text{Softmax}\left(\frac{q_i (K'_i)^T}{\sqrt{d}}\right) V'_i$。

技术新颖性

本文的技术新颖性体现在多个维度。首先,混合记忆范式的提出本身就是一种概念创新,它突破了现有视频生成记忆机制的静态假设,首次将动态主体追踪纳入记忆的核心要求。这一范式转变不仅定义了新问题,还为后续研究指明了方向。其次,记忆tokenizer的设计具有技术创新性,通过3D卷积实现时空维度的联合处理,生成具有运动感知能力的token表示,这种'压缩即理解'的思想与传统的简单下采样或语义聚类有本质区别。第三,动态检索注意力机制是一种全新的注意力变体,它将全局检索与局部时序保持结合,通过Top-K选择实现高效的信息筛选。第四,DSC(动态主体一致性)指标的提出填补了评估动态主体一致性的方法学空白,该指标通过CLIP特征提取和时空对齐,能够定量评估主体在进出画面事件中的身份和运动保持能力。最后,HM-World数据集的构建方法——通过解耦相机轨迹和主体运动来系统性地生成进出画面事件——是一种创新的数据工程方法,为混合记忆研究提供了基础设施。

进出画面的相机运动实例
Fig. 2: 进出画面的相机运动实例
HM-World数据集的构建流程
Fig. 3: HM-World数据集的构建流程
模型架构总览
Fig. 4: 模型架构总览
HyDRA详细结构:(a) 记忆token化模块,(b) 动态检索注意力
Fig. 5: HyDRA详细结构:(a) 记忆token化模块,(b) 动态检索注意力

实验结果

本文在多个维度验证了HyDRA方法的有效性。在定量比较中,HyDRA在所有评估指标上均显著超越基线和现有方法:相比基线,PSNR从18.696提升至20.357(提升1.661),SSIM从0.517提升至0.606(提升17.2%),LPIPS从0.356降至0.289(降低18.8%),这表明HyDRA在帧重建精度上具有明显优势。在动态主体一致性方面,DSCctx从0.812提升至0.827,DSCGT从0.837提升至0.849,证明了方法在追踪离屏主体方面的有效性。与DFoT(ICML 2025)相比,HyDRA的PSNR高出2.664;与Context-as-Memory(SIGGRAPH Asia 2025)相比,PSNR高出1.436,Subject Consistency从0.911提升至0.926。更值得注意的是,即使与商业模型WorldPlay进行零样本比较,HyDRA也全面胜出,PSNR差距高达5.502,尽管WorldPlay在上下文相关指标上表现出色(DSCctx为0.822),这间接验证了DSC指标的有效性。消融实验进一步证实了设计的鲁棒性:记忆tokenizer的时间维度至关重要(从2减到1导致PSNR下降1.281),而空间维度的变化影响较小;检索token数量为10-15时性能稳定,过少(5个)会导致严重信息损失;动态亲和度检索相比视野重叠检索在Subject Consistency上提升2个百分点。

现有数据集与HM-World数据集的对比
Table 1: 现有数据集与HM-World数据集的对比
与其他方法的定量比较
Table 2: 与其他方法的定量比较
记忆tokenizer核大小的消融实验
Table 4: 记忆tokenizer核大小的消融实验
检索token数量的消融实验
Table 5: 检索token数量的消融实验
token检索方法的比较
Table 6: token检索方法的比较
与其他方法的定性比较
Fig. 6: 与其他方法的定性比较
查看结构化数据
任务指标本文基线提升
视频世界模型生成 PSNR (dB) 20.357 18.696 +1.661 (8.9%)
视频世界模型生成 SSIM 0.606 0.517 +0.089 (17.2%)
视频世界模型生成 LPIPS 0.289 0.356 -0.067 (18.8%)
动态主体一致性 DSCctx 0.827 0.812 +0.015 (1.8%)
动态主体一致性 DSCGT 0.849 0.837 +0.012 (1.4%)
视觉一致性 Subject Consistency 0.926 0.903 +0.023 (2.5%)
视觉一致性 Background Consistency 0.932 0.925 +0.007 (0.8%)

局限与改进

作者在论文中明确指出了本文的局限性:HyDRA在处理涉及三个或更多主体的复杂场景或存在严重遮挡的情况时,保持一致性生成的能力会下降。这一局限性是可以理解的,因为当多个主体同时进出画面时,记忆token中编码的运动信息会相互干扰,动态检索注意力的选择难度也会显著增加。从我自己的观察来看,还存在以下潜在局限:首先,数据集的构建完全依赖UE5渲染,虽然视觉质量高,但与真实世界视频存在域差异,模型在真实场景中的泛化能力有待验证;其次,本文的评估是在HM-World测试集上进行的,该测试集虽然包含未见过的场景和主体,但仍然遵循相同的渲染逻辑,对真实复杂场景的覆盖可能不足;第三,DSC指标依赖YOLOv11进行主体检测和裁剪,如果检测失败会影响指标的准确性;第四,方法基于Wan2.1-T2V-1.3B这一特定基线模型,其在更大规模模型上的适用性需要进一步验证。

独立分析的弱点

本文存在几个可以改进的弱点:第一,记忆tokenizer的3D卷积核大小固定为 $2 \times 4 \times 4$,虽然消融实验表明空间维度变化影响不大,但这种固定配置可能无法适应不同运动速度和尺度的主体,一种改进方向是设计自适应核大小或多尺度tokenizer;第二,Top-K检索策略虽然高效,但K值需要预先设定,在主体数量变化的场景中可能不够灵活,可以探索基于阈值的动态选择机制或引入可学习的门控机制;第三,当前方法对所有主体采用统一的记忆处理策略,没有区分不同主体的运动模式和重要程度,未来可以引入主体级别的记忆管理;第四,局部时序窗口大小固定为5帧,在快速运动场景中可能不足以保持局部连贯性,可以设计自适应窗口机制;第五,相机编码器采用简单的MLP结构,对于复杂的相机运动模式可能表达能力不足,可以考虑使用更复杂的编码架构如Transformer。

未来方向

作者提出未来工作将探索更先进、更鲁棒的记忆机制来处理复杂的多主体动态,并将方法扩展到无约束的真实世界环境。基于本文的成果,可以延伸出多个研究方向:首先,可以将混合记忆范式应用于更长视频生成,探索如何在数百帧甚至数千帧的生成中保持动态一致性;其次,可以将HyDRA与其他视频生成架构(如自回归模型)结合,探索混合记忆在不同生成范式中的适用性;第三,可以研究混合记忆在交互式场景中的应用,如用户控制的相机运动和主体行为;第四,可以探索将本文的记忆token化思想应用于其他需要长期一致性的生成任务,如3D场景生成、音频生成等;第五,可以研究如何将真实世界视频纳入混合记忆的训练,通过域适应或自监督学习缩小渲染数据与真实数据之间的差距;第六,可以探索更精细的主体表示方法,如3D感知的主体编码,以更好地处理遮挡和视角变化。

复现评估

本文在可复现性方面表现良好:代码已在GitHub开源(https://github.com/H-EmbodVis/HyDRA),为复现提供了基础。数据集HM-World包含59K视频片段,规模适中,但构建依赖UE5渲染流水线,这需要一定的游戏引擎使用经验。模型基于开源的Wan2.1-T2V-1.3B构建,训练使用32块GPU、10K迭代,总batch size为32,这一算力需求对学术机构是可接受的。不过,UE5场景和主体资源的具体来源、渲染参数的详细配置等信息在论文中未完全公开,可能影响数据集的完全复现。评估协议中使用的YOLOv11检测器和CLIP模型均为开源资源,评估流程可以复现。总体而言,核心方法和评估流程的复现难度中等,数据集构建的复现难度较高。