A bayesian approach to generative adversarial imitation learning

W Jeon, S Seo, KE Kim - Advances in neural information …, 2018 - proceedings.neurips.cc
Advances in neural information processing systems, 2018proceedings.neurips.cc
Generative adversarial training for imitation learning has shown promising results on high-
dimensional and continuous control tasks. This paradigm is based on reducing the imitation
learning problem to the density matching problem, where the agent iteratively refines the
policy to match the empirical state-action visitation frequency of the expert demonstration.
Although this approach has shown to robustly learn to imitate even with scarce
demonstration, one must still address the inherent challenge that collecting trajectory …
Abstract
Generative adversarial training for imitation learning has shown promising results on high-dimensional and continuous control tasks. This paradigm is based on reducing the imitation learning problem to the density matching problem, where the agent iteratively refines the policy to match the empirical state-action visitation frequency of the expert demonstration. Although this approach has shown to robustly learn to imitate even with scarce demonstration, one must still address the inherent challenge that collecting trajectory samples in each iteration is a costly operation. To address this issue, we first propose a Bayesian formulation of generative adversarial imitation learning (GAIL), where the imitation policy and the cost function are represented as stochastic neural networks. Then, we show that we can significantly enhance the sample efficiency of GAIL leveraging the predictive density of the cost, on an extensive set of imitation learning tasks with high-dimensional states and actions.
proceedings.neurips.cc
以上显示的是最相近的搜索结果。 查看全部搜索结果