about
Diffusion World Model (arxiv.org)
3 points by beefman on Feb 7, 2024 | hide | past | pdf | discuss on HN

In plain words: Instead of predicting one step at a time and chaining guesses, this model imagines whole future states and rewards in one go, then uses those rollouts to score actions from past data. It beat one-step simulators by 44% and matched methods that skip simulation.

Abstract · Diffusion World Model: Future Modeling Beyond Step-by-Step Rollout for Offline Reinforcement Learning

We introduce Diffusion World Model (DWM), a conditional diffusion model capable of predicting multistep future states and rewards concurrently. As opposed to traditional one-step dynamics models, DWM offers long-horizon predictions in a single forward pass, eliminating the need for recursive queries. We integrate DWM into model-based value estimation, where the short-term return is simulated by future trajectories sampled from DWM. In the context of offline reinforcement learning, DWM can be viewed as a conservative value regularization through generative modeling. Alternatively, it can be seen as a data source that enables offline Q-learning with synthetic data. Our experiments on the D4RL dataset confirm the robustness of DWM to long-horizon simulation. In terms of absolute performance, DWM significantly surpasses one-step dynamics models with a $44\%$ performance gain, and is comparable to or slightly surpassing their model-free counterparts.

Zihan Ding, Amy Zhang, Yuandong Tian, Qinqing Zheng
arXiv:2402.03570 · cs.LG, cs.AI · submitted Feb 5, 2024 · updated Oct 15, 2024
abstract · pdf · html

add comment on HN