2026
Horizon Imagination: Efficient On-Policy Rollout in Diffusion World Models
ICLR 2026poster
We study diffusion-based world models for reinforcement learning, which offer high generative fidelity but face critical efficiency challenges in control. Current methods either require heavyweight models at inference or rely on highly sequential imagination, both of which impose prohibitive comput…