PRISM: PRior-guided Imagination Sampling in world Models
TLDR
PRISM extracts action priors from a world model's frozen encoder to guide sampling in model-based planning, improving success rates with minimal architectural overhead.
Reasoning
The paper presents a simple, task-agnostic method for guiding action sampling using a prior derived from the world model's own representations, avoiding complex external models. However, the abstract cuts off the quantitative results, and there is no mention of real-world experiments or benchmarks, limiting the assessment of practical impact.
Read-first score
Read-first score 58, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 43.
Field roles
Rank sensitivity
Stability: volatile; rank range: 478.
Keyword Scores
Deep Analysis
Innovations
- Proposes a task-agnostic framework that extracts action prior from the same dataset and learned representations of the world model, avoiding architectural bloat from independent visual encoders or large-scale VLMs.
- Attaches a lightweight MLP directly to a frozen JEPA-style encoder to predict a state-conditioned Gaussian prior.
- Fuses the prior into the planner's sampling distribution via a precision-weighted Product-of-Gaussians update, a parameter-free closed-form integration.
- Improves success rates by 35 percentage points on Cube and 32 percentage points on PushT over vanilla world-model-based MPC without significant inference overhead.
Methodology
PRISM builds on a standard JEPA-style latent world model. It attaches a lightweight MLP to the frozen encoder to predict a state-conditioned Gaussian prior. At plan time, this prior is fused into the planner's sampling distribution via a precision-weighted Product-of-Gaussians update, steering the sampling process. The framework is task-agnostic and uses the same dataset as the world model.
Key Results
PRISM improves success rates by 35 percentage points on Cube and 32 percentage points on PushT over vanilla world-model-based MPC, without introducing significant inference overhead.