Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

MosaicMem: Hybrid Spatial Memory for Controllable Video World Models

About

Video diffusion models are moving beyond short, plausible clips toward world simulators that must remain consistent under camera motion, revisits, and intervention. Yet spatial memory remains a key bottleneck: explicit 3D structures can improve reprojection-based consistency but struggle to depict moving objects, while implicit memory often produces inaccurate camera motion even with correct poses. We propose Mosaic Memory (MosaicMem), a hybrid spatial memory that lifts patches into 3D for reliable localization and targeted retrieval, while exploiting the model's native conditioning to preserve prompt-following generation. MosaicMem composes spatially aligned patches in the queried view via a patch-and-compose interface, preserving what should persist while allowing the model to inpaint what should evolve. With PRoPE camera conditioning and two new memory alignment methods, experiments show improved pose adherence compared to implicit memory and stronger dynamic modeling than explicit baselines. MosaicMem further enables minute-level navigation, memory-based scene editing, and autoregressive rollout.

Wei Yu, Runjia Qian, Yumeng Li, Liquan Wang, Songheng Yin, Sri Siddarth Chakaravarthy P, Dennis Anthony, Yang Ye, Yidi Li, Weiwei Wan, Animesh Garg• 2026

Related benchmarks

TaskDatasetResultRank
Camera Motion ControlMosaicMem Camera Control (Dedicated Evaluation Set)
Rotational Error0.51
12
Memory Retrieval ConsistencyMosaicMem Memory Retrieval (test)
SSIM75
12
Motion Dynamics ModelingMosaicMem Motion Dynamics (Dedicated Evaluation Set)
Dynamic Score2.58
12
Video GenerationMosaicMem Dedicated Evaluation Set Overall Generation
FID65.67
12
Autoregressive video generationVBench
Total Quality Score81.11
4
Showing 5 of 5 rows

Other info

Follow for update