ml-agents/docs/Training-Imitation-Learning.md

Imitation Learning

It is often more intuitive to simply demonstrate the behavior we want an agent to perform, rather than attempting to have it learn via trial-and-error methods. Consider our running example of training a medic NPC. Instead of indirectly training a medic with the help of a reward function, we can give the medic real world examples of observations from the game and actions from a game controller to guide the medic's behavior. Imitation Learning uses pairs of observations and actions from from a demonstration to learn a policy. Video Link.

Imitation learning can also be used to help reinforcement learning. Especially in environments with sparse (i.e., infrequent or rare) rewards, the agent may never see the reward and thus not learn from it. Curiosity helps the agent explore, but in some cases it is easier to just show the agent how to achieve the reward. In these cases, imitation learning can dramatically reduce the time it takes to solve the environment. For instance, on the Pyramids environment, just 6 episodes of demonstrations can reduce training steps by more than 4 times.

ML-Agents provides several ways to learn from demonstrations. For most situations, GAIL is the preferred approach.

• To train using GAIL (Generative Adversarial Imitaiton Learning) you can add the GAIL reward signal. GAIL can be used with or without environment rewards, and works well when there are a limited number of demonstrations.
• To help bootstrap reinforcement learning, you can enable pretraining on the PPO trainer, in addition to using a small GAIL reward signal.
• To train an agent to exactly mimic demonstrations, you can use the Behavioral Cloning trainer. Behavioral Cloning can be used offline and online (in-editor), and learns very quickly. However, it usually is ineffective on more complex environments without a large number of demonstrations.

Recording Demonstrations

It is possible to record demonstrations of agent behavior from the Unity Editor, and save them as assets. These demonstrations contain information on the observations, actions, and rewards for a given agent during the recording session. They can be managed from the Editor, as well as used for training with Offline Behavioral Cloning (see below).

In order to record demonstrations from an agent, add the Demonstration Recorder component to a GameObject in the scene which contains an Agent component. Once added, it is possible to name the demonstration that will be recorded from the agent.

When Record is checked, a demonstration will be created whenever the scene is played from the Editor. Depending on the complexity of the task, anywhere from a few minutes or a few hours of demonstration data may be necessary to be useful for imitation learning. When you have recorded enough data, end the Editor play session, and a .demo file will be created in the Assets/Demonstrations folder. This file contains the demonstrations. Clicking on the file will provide metadata about the demonstration in the inspector.