Horizon Imagination: Efficient On-Policy Rollout in Diffusion World Models
Lior Cohen, Ofir Nabati, Kaixin Wang, Navdeep Kumar, Shie Mannor · 2026
Read Horizon Imagination: Efficient On-Policy Rollout in Diffusion World Models by Lior Cohen, Ofir Nabati, Kaixin Wang, Navdeep Kumar, Shie Mannor (2026) in the Tower of Babel, a free online library of public-domain works preserved in clean, searchable text. No sign-up. The full text is served as a plain-text file at a predictable URL, where text is present: https://antarctic-labs.com/tower-of-babel/arxiv-2602-08032.txt.
Description
Diffusion-based world models can imagine future observations for reinforcement learning, but doing so step by step is prohibitively expensive. Lior Cohen and four co-authors propose Horizon Imagination, an on-policy rollout method that cleans up several imagined future observations at once for discrete stochastic policies.
A stabilization mechanism and a new sampling schedule decouple the denoising budget from the imagined horizon, even supporting sub-frame budgets. On Atari 100K and Craftium, the method holds control performance with half the usual denoising steps and produces better generations under varied schedules. Reinforcement-learning researchers gain a world model that imagines far ahead without paying step-by-step costs.
Catalog record
| CREATOR | Lior Cohen, Ofir Nabati, Kaixin Wang, Navdeep Kumar, Shie Mannor |
|---|---|
| YEAR | 2026 |
| COLLECTION | PAPERS |
| FORMAT | TXT |
| FILE SIZE | 59 KB |
| VERSION | 1.0 |
| LICENSE | http://creativecommons.org/licenses/by/4.0/ |
| RIGHTS | CC BY |
| SOURCE | arXiv |
Source: https://arxiv.org/abs/2602-08032