Looped World Models
Abstract Overview
This paper introduces Looped World Models (LoopWM), presented as the first application of looped transformer architectures to world modeling. The core idea is to refine latent environment states iteratively using a parameter-shared transformer block, coupled with a spectrally constrained state-retention mechanism intended to keep recurrent updates bounded over long rollouts. The model also supports adaptive early exit, so computation depth can vary with transition difficulty, and a deferred decoding variant that rolls out in latent space and decodes only at the terminal step.
Novelty
The paper’s main novelty is framing world modeling with a looped, recurrent-depth transformer that reuses the same parameters across iterative latent refinement steps, rather than relying on a fixed-depth dynamics model. It also combines this with a spectral stability construction for bounded recurrent dynamics, adaptive early exit at inference, and a deferred decoding scheme for action-conditioned latent rollouts.
Results
On ScienceWorld, the reported overall scores for LoopWM are EM 68.4%, Token F1 85.3%, BLEU-4 80.7%, and Entity 83.9%, compared with 47.2%, 72.8%, 64.4%, and 72.3% for the claude-opus-4-6-max baseline in the paper’s comparison. On AlfWorld, LoopWM reports EM 51.6%, Token F1 80.4%, BLEU-4 71.6%, and Entity 81.1%, giving the best BLEU among the listed models and second-place EM and F1, while entity accuracy appears weaker than some baselines. The deferred decoding analysis further reports gains over external baselines across rollout steps and suggests benefits from accumulating latent computation over longer action sequences.
Key Points
- LoopWM uses a parameter-shared looped transformer to iteratively refine latent state transitions, making effective depth less tied to parameter count.
- The architecture adds a spectrally constrained recurrent update and adaptive early-exit mechanism to target stable long-horizon rollouts and variable test-time compute.
- Empirical results are reported on ScienceWorld and AlfWorld, with especially strong ScienceWorld gains over listed baselines and a deferred-decoding analysis indicating improved multi-step rollout performance.
References
- arXiv: https://arxiv.org/abs/2606.18208v1
- Fugu-MT: https://fugumt.com/fugumt/paper_check/2606.18208v1
- Hugging Face Papers: https://huggingface.co/papers/2606.18208