Integrating Behavior Cloning and Reinforcement Learning for Improved Performance in Sparse Reward Environments

By Vinicius G. Goecks, Gregory M. Gremillion, Vernon J. Lawhern, J. Valasek, Nicholas R. Waytowich. 2019

34 citations recorded by the source corpus.

This paper investigates how to efficiently transition and update policies, trained initially with demonstrations, using off-policy actor-critic reinforcement learning. It is well-known that techniques based on Learning from Demonstrations, for example behavior cloning, can lead to proficient policies given limited data. However, it is currently unclear how to efficiently update that policy using reinforcement learning as these approaches are inherently optimizing different objective functions. Previous works have used loss functions, which combine behavior cloning losses with reinforcement learning losses to enable this update. However, the components of these loss functions are often set anecdotally, and their individual contributions are not well understood. In this work, we propose the Cycle-of-Learning (CoL) framework that uses an actor-critic architecture with a loss function that combines behavior cloning and 1-step Q-learning losses with an off-policy pre-training step from human demonstrations. This enables transition from behavior cloning to reinforcement learning without performance degradation and improves reinforcement learning in terms of overall performance and training time. Additionally, we carefully study the composition of these combined losses and their impact on overall policy learning. We show that our approach outperforms state-of-the-art techniques for combining behavior cloning and reinforcement learning for both dense and sparse reward scenarios. Our results also suggest that directly including the behavior cloning loss on demonstration data helps to ensure stable learning and ground future policy updates.

Subfields: Artificial Intelligence, Information Systems

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. Shixiang Shane Gu
  2. Sergey Levine
  3. Scott Fujimoto
  4. Jonathan Kelly
  5. Filip Mari'c
  6. Trevor Ablett
  7. Pieter Abbeel
  8. Ofir Nachum

In Information Systems

  1. Scott Fujimoto
  2. Shixiang Shane Gu

Citations in the corpus