AdaJEPA: An Adaptive Latent World Model
TLDR
AdaJEPA adapts a latent world model at test time via self-supervised updates within MPC, improving planning under distribution shift.
Reasoning
The paper introduces a novel test-time adaptation mechanism for latent world models, addressing a key limitation of frozen models. Its strength lies in the closed-loop self-supervised update that requires no extra demonstrations. However, the abstract lacks details on task diversity and does not mention real-world validation, limiting generalizability claims.
Read-first score
Read-first score 36.3, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 43.
Field roles
Rank sensitivity
Stability: volatile; rank range: 221.
Keyword Scores
Deep Analysis
Innovations
- Test-time adaptation of a latent world model within the closed loop of model predictive control (MPC)
- Self-supervised adaptation using observed next-state transitions as a signal, without additional expert demonstrations
- Continuous recalibration of the world model with as few as one gradient step per MPC replanning step
Methodology
AdaJEPA is a latent world model that, after training, performs MPC by planning action chunks, executing the first chunk, then using the observed next-state transition as a self-supervised adaptation signal to update the model (one gradient step) before replanning, all in a closed loop.
Key Results
Across a range of goal-reaching tasks, AdaJEPA substantially improves planning success with as few as one gradient step per MPC replanning step.