Planning / Foundations
verifiedWorld Model
Learn to predict what the environment does next, then practise inside that prediction instead of in the world. Real experience is slow, dangerous or expensive; imagined experience is none of those. The catch is that a policy trained in imagination is only as good as the imagination, and it will find and exploit whatever the model got wrong.
The usual shape is an encoder to a latent state, a dynamics model that rolls that state forward, and a reward head — so planning happens in latent space at a fraction of pixel cost. The exploitation problem is real and specific: an optimiser searching a learned model will steer toward its errors, which is why rollouts are kept short and why ensembles that disagree are used to say where the model should not be trusted.
Learn p(z sub t+1 | z sub t , a sub t ) and r̂(z sub t , a sub t ), then optimise a policy against rollouts of that model rather than the environment. Model error accumulates along a rollout rather than staying put: under the simulation lemma a per-step error ε gives a return gap growing roughly with the horizon h, and again with 1/(1−γ) through the discounting — the reason short rollouts from real states beat long ones from imagined starts.
imagined-steps holds 50% of the budget; rest holds the remaining 50%.
Steps taken inside the learned model against steps taken in the environment itself, in steps. Drag the imagination up to watch real experience become the minority — which is the saving, and also the exposure.
Reviewed by opendroid · 2026-08-18
- arXiv:1803.10122 — World Models
- arXiv:2301.04104 — Mastering Diverse Domains through World Models