arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.07853cs.LGcs.AIstat.ML
收藏

具有进度曲线的主动世界模型学习

Active World Model Learning with Progress Curiosity

Stanford University(斯坦福大学);Harvard University(哈佛大学)

Kuno Kim, Megumi Sano, Julian De Freitas, Nick Haber, Daniel Yamins

摘要:

世界模型是世界如何演变的自我监督预测模型。人类通过好奇地探索他们的环境来学习世界模型,在这个过程中获得高带宽感官输入的紧凑抽象,跨长时间范围计划的能力,以及对其他代理人行为模式的理解。在这项工作中,我们研究如何设计这样一个好奇心驱动的主动世界模型学习(AWML)系统。要做到这一点,我们构建了一个好奇的代理建设世界模型,同时在视觉上探索一个3D的物理环境,丰富的蒸馏代表现实世界的代理。我们提出了一个AWML系统驱动的$γ$ -进步:一个可扩展的和有效的学习进度为基础的好奇心信号。我们表明,$γ$ -进展自然会产生一个探索政策,将注意力集中在复杂的,但学习的动态平衡的方式,从而克服了“白噪声问题”。因此,我们的$γ$ -Progress-driven控制器实现了比配备最先进的探索策略(如随机网络蒸馏和模型不一致)的基线控制器更高的AWML性能。

英文摘要:

World models are self-supervised predictive models of how the world evolves. Humans learn world models by curiously exploring their environment, in the process acquiring compact abstractions of high bandwidth sensory inputs, the ability to plan across long temporal horizons, and an understanding of the behavioral patterns of other agents. In this work, we study how to design such a curiosity-driven Active World Model Learning (AWML) system. To do so, we construct a curious agent building world models while visually exploring a 3D physical environment rich with distillations of representative real-world agents. We propose an AWML system driven by $γ$-Progress: a scalable and effective learning progress-based curiosity signal. We show that $γ$-Progress naturally gives rise to an exploration policy that directs attention to complex but learnable dynamics in a balanced manner, thus overcoming the "white noise problem". As a result, our $γ$-Progress-driven controller achieves significantly higher AWML performance than baseline controllers equipped with state-of-the-art exploration strategies such as Random Network Distillation and Model Disagreement.

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

↑