Latent Spatial Memory for Video World Models

AK's Picks Embodied AI and Robotics World Models GenAI Diffusion
2026年06月08日
通常,能够保持生成帧之间三维空间一致性的视频世界模型,依赖于在RGB空间中显式构建的点云记忆。这种设计不仅计算开销巨大——需要反复进行渲染和变分自编码器(VAE)编码——而且本质上存在信息损失:像素空间的往返重构过程会丢弃扩散模型所学习到的潜在表征中蕴含的丰富特征。本文提出面向视频世界模型的“潜在空间空间记忆”(latent spatial memory),即一种持久化的三维缓存机制,可直接在扩散模型的潜在空间中存储场景信息,从而完全规避像素空间的重建过程。基于该思想,我们进一步提出了Mirage框架——一种基于潜在空间的空间记忆架构:它通过深度引导的反向投影,将潜在token提升至三维空间以构建记忆;再通过直接在潜在空间中进行形变(warping)来合成新视角,实现对记忆的查询。这一统一建模方式,既消除了像素空间重建带来的信息损失,也避免了反复编码与渲染所导致的巨大计算负担。实验表明,相较于显式的三维基线方法,潜在空间空间记忆可实现端到端视频生成速度最高提升**10.57倍**,内存占用降低**55倍**。此外,Mirage充分利用扩散模型所隐含的几何先验,在WorldScore基准上达到当前最优性能,并在RealEstate10K数据集上展现出优异的场景重建质量。
Video world models that maintain 3D spatial consistency across generated frames typically rely on explicit point cloud memory constructed in RGB space. This design is both computationally expensive, requiring repeated rendering and VAE encoding, and inherently lossy, as the round trip through pixel space discards rich features of the learned latent representation. In this paper, we introduce \emph{latent spatial memory} for video world models, a persistent 3D cache that stores scene information directly in the diffusion latent space, avoiding pixel-space reconstruction. Building on this, we propose Mirage, a latent-space spatial memory framework that constructs the memory by lifting latent tokens into 3D via depth-guided back-projection and queries it by synthesizing novel views through direct latent-space warping. This unified formulation eliminates both the information loss of pixel-space reconstruction and the computational burden of repeated encoding and rendering. Experiments show that latent spatial memory achieves up to \textbf{10.57}$\times$ faster end-to-end video generation and \textbf{55}$\times$ reduction in memory footprint relative to explicit 3D baselines. Leveraging the geometric prior of the diffusion model, Mirage attains state-of-the-art performance on WorldScore and strong reconstruction quality on RealEstate10K.
许愿