mlsanityData to 14 Oct 2026

Adaptive Compute in Latent World Models: When Depth Helps, Hurts, or Doesn't Matter

arXiv 2607.10203 ↗First posted 11 Jul 2026cs.LG cs.AIcited by 1

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

  1. 2024Revisiting Feature Prediction for Learning Visual Representations from Video15 Feb
  2. 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecture19 Jan
  3. 2022Temporal Difference Learning for Model Predictive Control9 Mar
  4. 2018Learning Latent Dynamics for Planning from Pixels12 Nov

Other references · 30

  1. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  2. 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,750
  3. 2014Going Deeper with Convolutionscited by 4,241
  4. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
  5. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
  6. 2020Exploring Simple Siamese Representation Learningcited by 2,866
  7. 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
  8. 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,353
  9. 2023Mastering Diverse Domains through World Modelscited by 1,253
  10. 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,057
  11. 2018DeepMind Control Suitecited by 956
  12. 2021Deep Reinforcement Learning at the Edge of the Statistical Precipicecited by 807
18 more
  1. 2018Recurrent World Models Facilitate Policy Evolutioncited by 785
  2. 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Modelscited by 774
  3. 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 773
  4. 2017Deep Reinforcement Learning that Matterscited by 754
  5. 2018Universal Transformerscited by 647
  6. 2017BranchyNet: Fast Inference via Early Exiting from Deep Neural Networkscited by 596
  7. 2023TD-MPC2: Scalable, Robust World Models for Continuous Controlcited by 512
  8. 2016Adaptive Computation Time for Recurrent Neural Networkscited by 506
  9. 2021Dynamic Neural Networks: A Surveycited by 408
  10. 2014Deeply-Supervised Netscited by 367
  11. 2020Data-Efficient Reinforcement Learning with Self-Predictive Representationscited by 318
  12. 2022Confident Adaptive Language Modelingcited by 310
  13. 2024Mixture-of-Depths: Dynamically allocating compute in transformer-based language modelscited by 258
  14. 2019Depth-Adaptive Transformercited by 236
  15. 2019DeepMDP: Learning Continuous Latent Space Models for Representation Learningcited by 207
  16. 2021PonderNet: Learning to Pondercited by 168
  17. 2017Adaptive Neural Networks for Efficient Inferencecited by 163
  18. 2016Spatially Adaptive Computation Time for Residual Networkscited by 118

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026DeepJEPA: Scaling World Models from Within
  2. 2026A Control Theory of Predictability in Latent World Models
  3. 2026One-Step Prediction Fidelity Is a Poor Proxy for Planning Performance in JEPA World Models
  4. 2025Better World Models Can Lead to Better Post-Training Performance
  5. 2026Compact but Moving: Intervention-Relevant Geometry in Recurrent World Models
  6. 2026Sandwich-Residuals: Parameter-Efficient Test-time Adaptation of World Models
  7. 2026The Objective Is the Bottleneck: Latent World Models Encode What Their Planners Cannot Use
  8. 2026HaM-World: Soft-Hamiltonian World Models with Selective Memory for Planning