Benchmarking World Models
for Continual Learning on Compositional Tasks

Applied Artificial Intelligence (A2I) Lab, Oxford Robotics Institute
Department of Engineering Science, University of Oxford
*Corresponding author

Abstract

A desirable property of a world model is the ability to learn continually across tasks, adapting to new environments without forgetting what it has already learnt. In particular, the ability to retain and reuse knowledge obtained from prior experiences underpins an agent's ability to efficiently adapt to novel environments, as the dynamics of the physical world can often be described in reoccurring mechanisms. Their measure of adaptation, however, entangles two abilities: the speed and capacity to learn unseen tasks, and the reuse of knowledge already acquired, because incoming tasks always carry novel content alongside what recurs. In order to isolate knowledge reuse from prior experiences, we propose a compositional continual learning benchmark for world models in robot manipulation. Specifically, we design each task curriculum to end with a compositional task built by recombining all primitives that precede it in the sequence. We further factorise this composition along the axes of action and perception to better understand how different input modalities bottleneck knowledge reuse. We evaluate state-of-the-art world models under canonical continual learning methods, alongside a modular world model whose dynamics backbone contains explicitly reusable components. Results show that modularity balances reuse against forgetting better than conventional methods, but none solve the problem fully, leaving clear room for continual world models built to reuse without forgetting.

Overview of the compositional continual learning benchmark for world models.

Overview of our compositional continual learning benchmark for world models in robot manipulation. Top: Each task suite presents a sequence of primitive manipulation tasks, ending with a composition task that recombines them across action, perception, or full composition axes. Bottom: We evaluate the continual learning ability of world models to efficiently adapt to the composition tasks through knowledge reuse, without forgetting previously learnt primitives along the sequence.

Motivation: Reuse Without Forgetting

A robot should draw on familiar objects, motions, and mechanisms when learning a new task, while keeping the skills it already has. World models learn how the environment responds to actions, making them a natural foundation for this reuse. Retaining and recombining that knowledge can reduce both relearning and the need to keep adding model capacity.

The benchmark gap. Existing continual learning benchmarks introduce unseen interactions with each new task, even when tasks are related. Faster adaptation therefore mixes two abilities: reusing prior knowledge and learning new interactions quickly.

Our approach. We build task sequences that end with a composition of the primitives learnt earlier. Forward transfer on this final task gives a more direct measure of reuse, while evaluation of earlier tasks tracks forgetting.

Compositional Benchmark

Six robot manipulation suites built on Meta-World assets share the same interface: three camera views and proprioception as observations, with continuous end-effector and gripper control. Each suite presents two or three primitive tasks followed by their composition.

Composition Along Three Axes

We vary action and perception, the two inputs to a world model, to locate where reuse becomes difficult.

  • Action: combine familiar motions while keeping the scene and objects fixed.
  • Perception: recombine familiar visual components while keeping the manipulation behaviour approximately fixed.
  • Full: combine both motions and visual components.
Suite Primitive task curriculum Composition task
Action composition
Reach reach in xy → reach in xz → reach in yz reach in xyz
Grasp pick cube → place cube pick and place cube
Perception composition
BinPnP pick and place cube from red bin to blue bin → yellow to blue → red to purple yellow bin to purple bin
PnPBlock pick and place red block → pick and place blue block stack the two blocks
Full composition
DrawerPnP pick and place cube → open drawer open drawer then pick and place cube inside
PnPBoxClose pick and place cube → close box pick and place cube inside box then close it

Model Separation for Continual Learning

A shared backbone learns continually. The encoder, dynamics model, and decoder (if present) capture the environment and carry knowledge across tasks.

Task-specific heads need not. Reward, actor, and critic heads reflect each task's objective. We train fresh heads for each task and then freeze and preserve them. Earlier tasks are evaluated using their stored heads with the current backbone.

Experimental Setup

Baselines. We evaluate DreamerV3 and TD-MPC2 with fine-tuning (FT), experience replay (ER), elastic weight consolidation (EWC), and PackNet. Our continual variant of Prismatic World Model (PWM) adds dynamics experts for each task and learns to combine them with frozen prior experts.

Frozen-encoder diagnostic. To examine dynamics reuse without encoder drift, we also compare PWM and TD-MPC2 with the same encoder pretrained on demonstrations from all benchmark tasks and then frozen; this diagnostic has access to the full task distribution.

Forward transfer (FWT↑) measures learning on the final composition task relative to learning from scratch. Backward transfer (BWT↓) measures forgetting on earlier tasks. See metric definitions, full scores, and learning curves below.

Results

Baseline Performance

Backward and forward transfer across six suites for DreamerV3, TD-MPC2, and PWM.

Left: conventional methods. Right: modular PWM versus TD-MPC2 with scratch or pretrained frozen encoders. Means over three seeds; whiskers show the minimum and maximum.

DreamerV3 outperforms TD-MPC2 on average. Its stronger reuse and retention suggest that reconstructing observations may help preserve useful representations.

Difficulty rises from action to perception and full composition. Visual changes make retaining and reusing knowledge harder, with full composition challenging both abilities together.

Conventional methods trade reuse against retention. Replay gives the strongest forward transfer, while PackNet comes closest to preventing forgetting.

Explicit modularity pushes the frontier further. With the shared pretrained frozen encoder, PWM nearly eliminates forgetting while retaining forward transfer comparable to TD-MPC2, at the cost of growing expert capacity.

Probing Reuse in the Modular World Model

Probing PWM with the pretrained frozen encoder reveals how its experts contribute:

Routing reflects task-level reuse in several curricula. Prior experts receive weight consistent with recurring components in Reach, Grasp, and BinPnP.

Task-level routing may limit temporal composition. A fixed expert mixture cannot switch between primitives as a task progresses, which may explain weaker correspondence in stacking and full composition.

Expert preferences survive composition learning. The highest-weighted expert from each primitive remains top-ranked in 13 of 14 cases.

Ablations confirm functional reuse of prior experts. Prior experts retain useful performance, while new experts alone and uniform routing fall short of the full learnt mixture.

Reuse without forgetting remains open: the results point towards stable representations learnt continually and more effective combinations of prior skills, especially across stages of a task.

Appendix A

Metric Definitions

The two metrics on a representative run.

The two metrics on a representative run. Solid traces show each task while it is being trained, dashed traces show the evaluation afterwards, and the dotted grey trace shows the final composition task learnt from scratch. Each shaded region is the numerator of one metric. Backward transfer: for every earlier task, the region between its end-of-training level $C_i$ and its later evaluation curve, which we express as a fraction of the rectangle of width $b_N - b_i$ and height $C_i - f$ that encloses it. Forward transfer: on the final task only, the region between the sequential curve and the scratch curve, positive where the curriculum helps and negative where it hurts.

Let a curriculum contain $N$ tasks trained sequentially, and let $b_i$ be the global step at which training of task $i$ ends. Let $R_i(s)$ be the expert-normalised return of task $i$ at step $s$, where $f = 0$ is the failure floor and $100$ is a privileged, state-based scripted oracle policy reference. Scores are not capped above, so a model that outperforms the reference policy takes values above 100. Define $C_i = R_i(b_i)$ as the task's performance when its own training ends. Evaluation is run at a fixed checkpoint interval, so each integral is computed by the trapezoidal rule over those checkpoints, and all curves are averaged over three random seeds before a metric is applied.

Backward transfer. For every non-final task, we take the shaded region between $C_i$ and the task's evaluation curve, and express it as a fraction of the rectangle enclosing it:

$$ D_i = \frac{\displaystyle\int_{b_i}^{b_N} \max\!\left(0,\, C_i - R_i(s)\right) \mathrm{d}s} {\left(b_N - b_i\right)\left(C_i - f\right)} \times 100. $$

The clipping at 0 means later gains do not offset earlier losses, so $D_i$ lies in $[0, 100]$: 0 if the curve never drops after training, 100 if it collapses to the floor immediately and stays there. The normalisation requires $C_i > f$, which means a task must end with non-zero performance. If a task is never learnt, it holds nothing that could be forgotten and backward transfer calculation is excluded. Averaging over the tasks that were learnt yields the backward transfer score for this task suite, lower is better:

$$ \mathcal{I} = \left\{\, i < N : C_i > f \,\right\}, \qquad \mathrm{BWT} = \frac{1}{|\mathcal{I}|} \sum_{i \in \mathcal{I}} D_i. $$

Note that our BWT is a forgetting score in $[0, 100]$, and so is not the same quantity as the backward transfer introduced in Lopez-Paz and Ranzato (2017), which subtracts a task's original performance from its final performance and is higher-is-better. Theirs compares two data points; ours integrates over the whole remaining curriculum.

Forward transfer. Forward transfer is measured only on the final composition task to examine the reusing ability of the models. Let $R_N^{\mathrm{seq}}(t)$ be its learning curve during its position in the curriculum, rebased to $t = 0$, and let $R_N^{\mathrm{scr}}(t)$ be the curve obtained when learning the same task from scratch with no preceding primitives. With a shared training budget of $T$ steps, FWT is the shaded region between them divided by the learning length:

$$ \mathrm{FWT} = \frac{1}{T} \int_{0}^{T} \left( R_N^{\mathrm{seq}}(t) - R_N^{\mathrm{scr}}(t) \right) \mathrm{d}t. $$

This is higher-is-better. Because the curves are compared over the whole budget rather than at a fixed threshold, FWT rewards learning the composition task faster and learning it better.

Appendix B

Complete Numerical Results

The tables below give the per-suite values underlying the aggregate discussion above. Suites are listed in the order of the task suite table, grouped by composition type.

Suite FT ER EWC PackNet
BWT↓FWT↑ BWT↓FWT↑ BWT↓FWT↑ BWT↓FWT↑
DreamerV3
Reach 11.7743.91 5.6849.42 7.22−2.25 0.6736.95
Grasp 10.0721.28 5.0726.71 4.62−20.26 2.4316.71
BinPnP 29.2742.09 6.4649.26 8.360.90 1.5738.00
PnPBlock 31.8011.64 11.8036.37 4.368.74 0.719.20
DrawerPnP 56.157.06 21.9035.52 13.92−1.65 0.2013.45
PnPBoxClose 65.7531.12 43.223.99 39.206.01 0.6328.86
Average 34.1426.18 15.6933.55 12.95−1.42 1.0423.86
TD-MPC2
Reach 38.1410.47 25.3812.07 23.0910.35 0.7712.94
Grasp 62.6227.17 46.7024.74 51.5417.27 8.9819.04
BinPnP 77.1610.50 44.6119.69 64.465.18 3.849.73
PnPBlock 94.892.51 82.792.39 90.20−54.28 4.56−10.73
DrawerPnP 89.591.07 88.511.81 91.65−3.53 4.10−0.17
PnPBoxClose 93.07−6.27 85.483.19 90.163.21 6.9013.04
Average 75.917.57 62.2510.65 68.52−3.63 4.867.31

Continual learning performance of monolithic world models. Underline marks the better backbone (DreamerV3 or TD-MPC2) for that method and suite. Bold marks the best method within a backbone. Values are means over three random seeds.

Suite Scratch encoder Pretrained frozen encoder
PWM TD-MPC2 PWM TD-MPC2
BWT↓FWT↑ BWT↓FWT↑ BWT↓FWT↑ BWT↓FWT↑
Reach 44.459.35 38.1410.47 2.475.84 35.0710.64
Grasp 90.7314.49 62.6227.17 5.6948.29 13.8049.44
BinPnP 66.971.88 77.1610.50 6.6534.30 10.2737.50
PnPBlock 92.36−3.21 94.892.51 3.9950.02 45.3658.38
DrawerPnP 90.812.01 89.591.07 1.6441.02 59.0640.22
PnPBoxClose 89.95−10.09 93.07−6.27 2.6837.62 46.2432.51
Average 79.212.41 75.917.57 3.8536.18 34.9738.11

Modular PWM against monolithic TD-MPC2. Bold marks the better backbone (PWM or TD-MPC2) for that suite under the same encoder choice. Values are means over three random seeds.

Appendix C

Complete Learning Curves

The figures below give the learning curves for every suite and method. Solid coloured traces show sequential training, dashed traces show task evaluation after each switch, and grey shows final-task evaluation from scratch. Shading with different colours corresponds to the FWT and BWT regions defined in Appendix A. Vertical dotted lines mark task switches. The training budget per task is set by the steps a from-scratch run needs to converge, and therefore differs across backbones. Both metrics are normalised over the interval they integrate; these comparisons do not impose equal parameter or compute budgets.

Complete DreamerV3 learning curves.

Complete DreamerV3 learning curves across six task suites and four continual-learning methods.

Complete TD-MPC2 learning curves.

Complete TD-MPC2 learning curves across six task suites and four continual-learning methods.

The frozen-encoder comparison across all six task suites.

The frozen-encoder comparison across all six task suites: PWM with a scratch encoder, PWM with the pretrained frozen encoder, and monolithic TD-MPC2 with the same pretrained frozen encoder. The middle and right panels share the same pretrained frozen encoder and compare modular PWM with monolithic TD-MPC2. PWM also grows its expert capacity across tasks, so this is not a parameter-matched comparison. TD-MPC2 with a scratch encoder can be found in the first column of the TD-MPC2 figure above.