Long-Context State-Space Video World Models
TLDR
Proposes a state-space model architecture for video world models that extends temporal memory efficiently, evaluated on Memory Maze and Minecraft.
Reasoning
The paper introduces a novel block-wise SSM scanning scheme to address long-term memory limitations in video diffusion world models, with strong empirical results on spatial retrieval tasks. However, the evaluation is limited to simulated environments, and the abstract does not provide details on real-world deployment or comparisons to model-based RL baselines.
Read-first score
Read-first score 70.8, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 53.
Field roles
Rank sensitivity
Stability: volatile; rank range: 53.
Keyword Scores
Deep Analysis
Innovations
- Novel architecture using state-space models (SSMs) for long-term temporal memory in video world models, overcoming computational cost of attention layers
- Block-wise SSM scanning scheme that strategically trades spatial consistency for extended temporal memory
- Dense local attention to ensure coherence between consecutive frames
Methodology
The paper proposes a causal video world model that combines state-space models (SSMs) with a block-wise scanning scheme to extend temporal memory efficiently, paired with dense local attention for frame-to-frame coherence. The model is trained on action-conditioned autoregressive frame prediction and evaluated on long-term spatial retrieval and reasoning tasks using Memory Maze and Minecraft datasets.
Key Results
The proposed approach surpasses baselines in preserving long-range memory on Memory Maze and Minecraft datasets while maintaining practical inference speeds suitable for interactive applications.
Limitations
- Trade-off between spatial consistency and extended temporal memory may degrade fine-grained spatial details
- Evaluation limited to Memory Maze and Minecraft datasets, leaving generalization to other domains unverified
- Inference speed is described as 'practical' but not quantitatively compared to baselines or real-time requirements