Learning Goal Embeddings via Self-Play for Hierarchical Reinforcement Learning
Learning Goal Embeddings via Self-Play for Hierarchical Reinforcement Learning
复制标题
通过分层强化学习的自我对弈学习目标嵌入
DOI:
--
复制
发表时间:
2018
期刊:
影响因子:
--
通讯作者:
R. Fergus
中科院分区:
文献类型:
--
作者:
Sainbayar Sukhbaatar;Emily L. Denton;Arthur Szlam;R. Fergus
In hierarchical reinforcement learning a major challenge is determining appropriate low-level policies. We propose an unsupervised learning scheme, based on asymmetric self-play from Sukhbaatar et al. (2018), that automatically learns a good representation of sub-goals in the environment and a low-level policy that can execute them. A high-level policy can then direct the lower one by generating a sequence of continuous sub-goal vectors. We evaluate our model using Mazebase and Mujoco environments, including the challenging AntGather task. Visualizations of the sub-goal embeddings reveal a logical decomposition of tasks within the environment. Quantitatively, our approach obtains compelling performance gains over non-hierarchical approaches.