arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.03570cs.LGcs.AI
收藏

Diffusion World Model:面向离线强化学习的超越逐步展开的未来建模

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

Princeton University(普林斯顿大学);University of Texas at Austin(德克萨斯大学奥斯汀分校);Meta

Zihan Ding, Amy Zhang, Yuandong Tian, Qinqing Zheng

AI总结:

提出Diffusion World Model(DWM),通过单次前向传播同时预测多步未来状态与奖励,集成于离线强化学习的值估计,在D4RL上较一步模型提升44%性能。

摘要:

我们引入了Diffusion World Model(DWM),这是一种能够同时预测多步未来状态和奖励的条件扩散模型。与传统的一步动力学模型不同,DWM在单次前向传播中提供长程预测,消除了递归查询的需要。我们将DWM集成到基于模型的值估计中,其中短期回报由从DWM采样的未来轨迹模拟。在离线强化学习背景下,DWM可以被视为通过生成建模实现的保守值正则化。或者,它也可以被看作是使带有合成数据的离线Q-learning成为可能的数据源。我们在D4RL数据集上的实验证实了DWM对长程模拟的鲁棒性。在绝对性能方面,DWM以44%的性能提升显著超越了一步动力学模型,并且与无模型对应方法相当或略胜一筹。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑