Latent Spatial Memory for Video World Models

Weijie Wang, Haoyu Zhao, Yifan Yang, Feng Chen, Zeyu Zhang +5 more
6/8/2026
cs.CV

No official code found. Best implementation: microsoft/LatentSpatialMemory (327 stars, MIT, last commit 2026-06-09).

Abstract

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.

View on arXivView PDF

Code Implementations(1)

MIT

Latent Spatial Memory for Video World Models

32711May 1, 20264 months agoMIT

Similar Papers

Cite this paper

@article{wang2026latent,
  title  = {Latent Spatial Memory for Video World Models},
  author = {Weijie Wang and Haoyu Zhao and Yifan Yang and Feng Chen and Zeyu Zhang and Yefei He and Zicheng Duan and Donny Y. Chen and Yuqing Yang and Bohan Zhuang},
  year   = {2026},
  eprint = {2606.09828},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2606.09828v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2606.09828.svg)](https://codeofpaper.com/paper/2606.09828)

Discussion