Théo Cachet

Demonstration-Conditioned Reinforcement Learning for Few-Shot Imitation

In few-shot imitation, an agent is given a few demonstrations of a previously unseen task, and must then successfully perform that task. We propose a novel approach to learning few-shot-imitation agents that we call demonstration-conditioned reinforcement learning (DCRL). Given a training set consis…