titleLatent Spatial Memory for Video World Models
authorsWeijie Wang, Haoyu Zhao, Yifan Yang, Feng Chen, Zeyu Zhang, Yefei He, Zicheng Duan, Donny Y. Chen, Yuqing Yang, Bohan Zhuang
date2026-06-08
venuearXiv (preprint, not peer reviewed)
idarXiv:2606.09828
affiliationMicrosoft Research
upvotes65 (Hugging Face Daily Papers, 2026-11-01)
linksabs pdf hf code
primerWorld models
abstractVideo 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 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 10.57times faster end-to-end video generation and 55times 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.

cite

@misc{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={https://arxiv.org/abs/2606.09828}
}