MemLearner: Learning to Query Context memory for Video World Models
TLDR
MemLearner learns to query context memory adaptively for video world models, improving scene consistency under occlusions and dynamics.
Reasoning
The paper directly addresses a key limitation (memory) in video world models with a novel learning-based query method, supported by a new dataset and multi-dataset training. Strengths include leveraging pre-trained priors and strong empirical results; weaknesses are not evident from the abstract alone but the method's scalability and real-world generalization remain to be fully assessed.
Read-first score
Read-first score 42.5, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 46.
Field roles
Rank sensitivity
Stability: volatile; rank range: 321.
Keyword Scores
Deep Analysis
Innovations
- Learning-based adaptive context query method using query tokens to bridge context and predicted tokens for memory in video world models
- Leveraging the video generation model itself for context querying, exploiting pre-trained visual priors without training additional modules from scratch
- Efficient training and inference strategies for the query mechanism
- Collection of a new dataset of long videos with scene occlusions and dynamic objects, paired with camera pose annotations
- Multi-dataset training strategy combining annotated rendered videos and unannotated real-world videos
Methodology
MemLearner introduces a learning-based adaptive context query approach that uses query tokens within a video generation model to retrieve relevant context frames from memory. It avoids training new modules from scratch by repurposing the generation model's own pre-trained visual priors for querying, and employs efficient training and inference strategies. The method is trained on a newly collected dataset of long videos featuring occlusions and dynamic objects with camera poses, using a multi-dataset strategy that leverages both annotated rendered and unannotated real-world videos.
Key Results
MemLearner significantly outperforms prior video world models in scene consistency and memory, particularly in challenging scenarios with occlusions and dynamic objects.