The World Model Remembers, the Actor Forgets: Dream Rehearsal for Continual Model-Based RL

Gurp Nijjer
7/22/2026
cs.LGcs.AI

No official code found. Best implementation: gurpnijjer/dream-rehearsal (2 stars, Apache-2.0, last commit 2026-07-23).

Abstract

Model-based reinforcement-learning agents of the DreamerV3 family forget catastrophically when trained on task sequences, even when an unbounded replay buffer preserves every earlier experience. We ask a question the continual-RL literature has assumed an answer to but never measured: which component forgets? Under never-clear replay, pre-registered component-level probes (n=3 seeds throughout) show that the world model retains essentially everything measurable about old tasks -- reward discrimination (retention ratio ~1.0), value estimates, and termination structure -- while the actor's behavior collapses. Forgetting in this regime is a channel problem, not a memory problem. We demonstrate this by intervention: with the world model frozen and identical imagined rollouts, reinforcement learning in imagination fails to recover a lost skill (0/3 seeds), while supervised self-imitation on the world model's own graded dreams recovers it on 3/3 seeds with zero environment interaction. Interleaved during training, this graded dream rehearsal yields a task-label-free, parameter-constant continual learner: 3/3 four-task chains retained where plain replay passes 0/3, 3/3 eight-task chains, and consistent gains over matched real-episode cloning (paired difference +0.13, bootstrap 95% CI [0.07, 0.24], complete seed separation). The dream-grading step is load-bearing: we characterize two scoring failure modes, provide an offline selection gauge that caught both before they contaminated results, and give a realized-first grading rule that closes them. All experiments were pre-registered with committed protocols; every refuted hypothesis is reported.

View on arXivView PDF

Code Implementations(1)

Apache-2.0

The World Model Remembers, the Actor Forgets: Dream Rehearsal for Continual Model-Based RL — code, pre-registration trail, and run data

20Jul 20, 20262 months agoApache-2.0
catastrophic-forgettingcontinual-learningdreamerv3model-based-rlreinforcement-learning+2 more

Similar Papers

Cite this paper

@article{nijjer2026world,
  title  = {The World Model Remembers, the Actor Forgets: Dream Rehearsal for Continual Model-Based RL},
  author = {Gurp Nijjer},
  year   = {2026},
  eprint = {2607.19749},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2607.19749v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2607.19749.svg)](https://codeofpaper.com/paper/2607.19749)

Discussion