Adaptive Compute in Latent World Models: When Depth Helps, Hurts, or Doesn't Matter
On the map: World models → Latent predictive world models
Abstract (arXiv)
Adaptive-compute world models run fewer predictor blocks on some rollout steps than on others. This rests on two assumptions: that a deeper predictor gives a more accurate rollout, and that the steps which need depth can be identified before the predictor runs. We test both on nine DeepMind Control tasks with a twelve-block early-exit latent predictor, eight seeds per task. The first assumption fails on three tasks, where a shallower exit rolls out more accurately than the full stack on every seed, and composing predictions over ten steps moves the depth-error relation away from its one-step shape in both directions. In hindsight, choosing depth per step helps: a greedy oracle that sees the true next latent has 10-37% lower rollout error than the best mixture of fixed depths at equal average compute, and 8-28% lower than the best step-index schedule, on all nine tasks. No router we trained captures this. The difficulty router these models carry stays within 5% of fixed-depth mixtures; a router trained to predict each exit's error beats them by up to 12% but no more than a fixed schedule does; and in closed-loop planning at equal compute the routed planner matches a fixed middle depth. On cheetah, where depth 2 rolls out more accurately than the full stack (an inversion), supervising the early exits only at the first rollout step removes the inversion, while a control with the same loss weight that supervises them at one random step keeps most of it (rho 0.91 vs 0.85 standard and 1.14 first-step-only, three seeds): the inversion depends on supervising exits from composed latents. These results change with the encoder, backbone, horizon and training data. The fixed-depth campaign was pre-registered; the routing analyses, planner runs, ablation control and one-step decomposition are exploratory.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Latent predictive world models, one of 11 directions in World models. Builds on 34 of its 37 references that are in the map. First posted on arXiv 11 Jul 2026.
Builds on
34 of 37 references.
Closest ancestors · 4
Same direction, newest first
- 2024Revisiting Feature Prediction for Learning Visual Representations from Video15 Feb
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecture19 Jan
- 2022Temporal Difference Learning for Model Predictive Control9 Mar
- 2018Learning Latent Dynamics for Planning from Pixels12 Nov
Other references · 30
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,750
- 2014Going Deeper with Convolutionscited by 4,241
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
- 2020Exploring Simple Siamese Representation Learningcited by 2,866
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
- 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,353
- 2023Mastering Diverse Domains through World Modelscited by 1,253
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,057
- 2018DeepMind Control Suitecited by 956
- 2021Deep Reinforcement Learning at the Edge of the Statistical Precipicecited by 807
18 more
- 2018Recurrent World Models Facilitate Policy Evolutioncited by 785
- 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Modelscited by 774
- 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 773
- 2017Deep Reinforcement Learning that Matterscited by 754
- 2018Universal Transformerscited by 647
- 2017BranchyNet: Fast Inference via Early Exiting from Deep Neural Networkscited by 596
- 2023TD-MPC2: Scalable, Robust World Models for Continuous Controlcited by 512
- 2016Adaptive Computation Time for Recurrent Neural Networkscited by 506
- 2021Dynamic Neural Networks: A Surveycited by 408
- 2014Deeply-Supervised Netscited by 367
- 2020Data-Efficient Reinforcement Learning with Self-Predictive Representationscited by 318
- 2022Confident Adaptive Language Modelingcited by 310
- 2024Mixture-of-Depths: Dynamically allocating compute in transformer-based language modelscited by 258
- 2019Depth-Adaptive Transformercited by 236
- 2019DeepMDP: Learning Continuous Latent Space Models for Representation Learningcited by 207
- 2021PonderNet: Learning to Pondercited by 168
- 2017Adaptive Neural Networks for Efficient Inferencecited by 163
- 2016Spatially Adaptive Computation Time for Residual Networkscited by 118
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026DeepJEPA: Scaling World Models from Within
- 2026A Control Theory of Predictability in Latent World Models
- 2026One-Step Prediction Fidelity Is a Poor Proxy for Planning Performance in JEPA World Models
- 2025Better World Models Can Lead to Better Post-Training Performance
- 2026Compact but Moving: Intervention-Relevant Geometry in Recurrent World Models
- 2026Sandwich-Residuals: Parameter-Efficient Test-time Adaptation of World Models
- 2026The Objective Is the Bottleneck: Latent World Models Encode What Their Planners Cannot Use
- 2026HaM-World: Soft-Hamiltonian World Models with Selective Memory for Planning