arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.04009cs.LG
收藏

探究基于模型的学习在探索和迁移中的作用

Investigating the role of model-based learning in exploration and transfer

DeepMind(深度思维);Google Research(谷歌研究院)

Jacob Walker, Eszter Vértes, Yazhe Li, Gabriel Dulac-Arnold, Ankesh Anand, Théophane Weber, Jessica B. Hamrick

摘要:

最先进的强化学习已经能够在日益复杂的任务上训练代理。然而,目前的范式往往有利于训练代理从头开始对每一个新的任务或任务的集合,以期推广到新的任务配置。前者的数据效率很低,而后者在测试任务分布不均匀时很困难。能够有效地传递他们关于世界的知识的代理人为这些问题提供了潜在的解决方案。在本文中,我们研究迁移学习的背景下,基于模型的代理。具体来说,我们的目标是了解环境模型何时具有优势以及为什么。我们发现,基于模型的方法优于受控的无模型基线的迁移学习。通过消融,我们表明,通过探索学习的政策和动力学模型的成功转移。我们展示了我们的结果在三个领域的不同,在他们的要求转移:在分布程序(工匠),在分布相同(RoboDesk),和出的分布(元世界)。我们的研究结果表明,内在的探索与环境模型相结合,提出了一个可行的方向代理,是自我监督,并能够推广到新的奖励功能。

英文摘要:

State of the art reinforcement learning has enabled training agents on tasks of ever increasing complexity. However, the current paradigm tends to favor training agents from scratch on every new task or on collections of tasks with a view towards generalizing to novel task configurations. The former suffers from poor data efficiency while the latter is difficult when test tasks are out-of-distribution. Agents that can effectively transfer their knowledge about the world pose a potential solution to these issues. In this paper, we investigate transfer learning in the context of model-based agents. Specifically, we aim to understand when exactly environment models have an advantage and why. We find that a model-based approach outperforms controlled model-free baselines for transfer learning. Through ablations, we show that both the policy and dynamics model learnt through exploration matter for successful transfer. We demonstrate our results across three domains which vary in their requirements for transfer: in-distribution procedural (Crafter), in-distribution identical (RoboDesk), and out-of-distribution (Meta-World). Our results show that intrinsic exploration combined with environment models present a viable direction towards agents that are self-supervised and able to generalize to novel reward functions.

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

↑