Taxonomies of World Models › Reading the Results

Michael BrenndoerferJune 4, 20267 min read
World Models HandbookAll chapters
Part 1 of 12The World-Model Idea
Chapter 4Taxonomies of World Models › Reading the Results
Reading section 9 of 12 · Reading the Results · 7 minSection 9 of 12 · 7 min

Book contents

Section 9 of 12 · 7 min read

Taxonomies of World Models › Reading the ResultsOne section · Sections in this chapter9 / 12 ↓
Reading tools Definitions & inline links

Choose your expertise level to adjust how many terms are explained. Beginners see more tooltips, experts see fewer to maintain reading flow. Hover over underlined terms for instant definitions.

Article links

Make inline references clickable

The headline result is the growth of multi-step error with horizon, plotted on a logarithmic scale so that the early errors remain visible beside the much larger long-horizon errors.

Out[17]:
Visualization
Log-scale line chart of prediction error rising with horizon, steepest for the observation-space model.
Angle prediction RMSE versus rollout horizon for the three contestants. The observation-space linear model compounds error fastest, while the learned latent model remains close to the oracle-feature reference throughout the sixty-step rollout.

The pattern confirms the taxonomy's central claim: a representation is useful only together with dynamics that fit its coordinates. All three contestants see the same episodes, action sequences, and evaluation cases. The observation-space baseline asks one linear map to express nonlinear dynamics in sixteen sensor coordinates, so its structural error compounds. The other two contestants use the pendulum's structured transition in three dynamics-relevant coordinates. Their difference is initialization: the learned model estimates those coordinates from an observation, while the oracle receives them directly.

The horizon changes the ranking. At one step, the observation-space baseline has about 0.008 radians of angle RMSE and appears better than the learned latent model at about 0.021. At sixty steps, the baseline reaches about 1.55 radians while the learned latent model remains near 0.15, a reversal of roughly a factor of ten. The oracle-feature reference is slightly lower at about 0.14. A next-step score therefore answers a different question from the one a planner cares about: whether repeated applications remain useful. The bar chart makes that reversal explicit.

Out[19]:
Visualization
Grouped bar chart comparing one-step and sixty-step RMSE across the three models.
Angle RMSE at one step versus sixty steps for the three contestants. The one-step ranking favors the observation-space baseline over the learned latent model, but the sixty-step ranking reverses by roughly a factor of ten; the oracle-feature model provides a noise-limited reference.

To see what the aggregate curves mean at the level of individual imaginations, we overlay one predicted trajectory against ground truth for the two learned models.

Out[20]:
Visualization
Line chart of predicted versus true angle, prediction drifting off quickly.
Sixty-step imagined trajectory from the observation-space linear model versus the true pendulum angle. The prediction drifts away from reality within a dozen steps.
Line chart of predicted versus true angle, prediction tracking longer.
The same rollout from the learned latent model. The prediction tracks the true trajectory substantially longer before diverging.

Two details deserve attention beyond the obvious difference in tracking quality. First, the true trajectory is visibly irregular while both predictions are smooth: the process noise that agitates the real pendulum is absent from deterministic imaginations, exactly the mean-seeking behavior described earlier. Second, the latent model's errors are not random jitter but slow drift, the signature of a slightly wrong but structurally reasonable model. Drift of this kind is what model-based controllers can sometimes correct through feedback, whereas the observation-space model's errors are too violent and too early for any planner to exploit its imagination productively. This distinction between error modes, drift versus chaos, matters practically: slow drift can be diagnosed, corrected through feedback, and often traced to specific neglected effects, while chaotic divergence tells you only that the model class was wrong and gives no hint of how to fix it.

Last, the stochastic variant shows what the deterministic/stochastic axis contributes independently of representation quality.

Out[21]:
Visualization
Fan chart of sampled trajectory percentiles widening over time, with deterministic prediction and true trajectories overlaid.
Two hundred sampled rollouts from the stochastic feature-space model, summarized by percentile bands, alongside the deterministic prediction and five true noisy trajectories. For this seed, the reference trajectories mostly lie inside the sampled cloud; the deterministic model commits to one path.

Read this figure as a portrait of honest ignorance. The deterministic model draws one line and asserts it. The stochastic model draws a cloud whose width grows with horizon, reflecting accumulating uncertainty, and the true trajectories travel inside the cloud rather than along its center. Nothing about the cloud tells a planner what to do; converting sampled futures into risk-sensitive decisions is the business of Part VII. But the cloud is the raw material that risk-sensitive planning consumes, and a deterministic model cannot manufacture it after the fact. Note also what the figure does not show: any guarantee that the cloud's width is correct. With only five reference trajectories overlaid, we can check bracketing informally, but rigorous calibration assessment would demand many more trials and a proper scoring rule, which is exactly the methodology Part XI develops.

It is worth being precise about what this toy demonstrates and what it does not. The feature-space model enjoys an unfair advantage we handed it: access to the sufficient statistics. The learned encoder also receives supervised state targets, a privileged diagnostic that isolates state-estimation error from transition-model error rather than a general recipe for representation learning. Part III develops methods that try to recover useful state from observations without such labels, using reconstruction, self-supervised dynamics, rewards, or combinations of objectives. Token-based and object-centered variants make still different tradeoffs. The pendulum fixes the shape of the argument, not the engineering frontier. What transfers from toy to frontier is the logic: identify which taxonomic position a design occupies, predict its characteristic failure from that position, and check the prediction against rollout behavior rather than one-step accuracy.