Imitation Learning via Kernel Mean Embedding

By Kee-Eung Kim, Hyun Soo Park. 2018

44 citations recorded by the source corpus.

Imitation learning refers to the problem where an agent learns a policy that mimics the demonstration provided by the expert, without any information on the cost function of the environment. Classical approaches to imitation learning usually rely on a restrictive class of cost functions that best explains the expert's demonstration, exemplified by linear functions of pre-defined features on states and actions. We show that the kernelization of a classical algorithm naturally reduces the imitation learning to a distribution learning problem, where the imitation policy tries to match the state-action visitation distribution of the expert. Closely related to our approach is the recent work on leveraging generative adversarial networks (GANs) for imitation learning, but our reduction to distribution learning is much simpler, robust to scarce expert demonstration, and sample efficient. We demonstrate the effectiveness of our approach on a wide range of high-dimensional control tasks.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Statistical and Nonlinear Physics

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Sergey Levine
  2. Olivier Pietquin
  3. Matthieu Geist
  4. L'eonard Hussenot
  5. Robert Dadashi
  6. Ilya Kostrikov
  7. Jonathan Tompson
  8. Pieter Abbeel

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Dacheng Tao
  3. Andrew Zisserman
  4. Sergey Levine
  5. Trevor Darrell
  6. Xiaogang Wang
  7. Jitendra Malik
  8. Abhinav Kumar Gupta

In Statistical and Nonlinear Physics

  1. Ilya Kostrikov
  2. Minkai Xu
  3. Tairan He
  4. Weinan Zhang
  5. Minghuan Liu
  6. Sergey Levine
  7. Debidatta Dwibedi
  8. Jonathan Tompson

Citations in the corpus