Imitation learning

Imitation learning is a paradigm in reinforcement learning, where an agent learns to perform a task by supervised learning from expert demonstrations. It is also called learning from demonstration and apprenticeship learning.

It has been applied to underactuated robotics, self-driving cars, quadcopter navigation, helicopter aerobatics, and locomotion.

Approaches

Expert demonstrations are recordings of an expert performing the desired task, often collected as state-action pairs $(o_{t}^{*},a_{t}^{*})$.

Behavior Cloning

Behavior Cloning (BC) is the most basic form of imitation learning. Essentially, it uses supervised learning to train a policy $\pi _{\theta }$ such that, given an observation $o_{t}$, it would output an action distribution $\pi _{\theta }(\cdot |o_{t})$ that is approximately the same as the action distribution of the experts.

BC is susceptible to distribution shift. Specifically, if the trained policy differs from the expert policy, it might find itself straying from expert trajectory into observations that would have never occurred in expert trajectories.

This was already noted by ALVINN, where they trained a neural network to drive a van using human demonstrations. They noticed that because a human driver never strays far from the path, the network would never be trained on what action to take if it ever finds itself straying far from the path.

DAgger

Dagger (Dataset Aggregation) improves on behavior cloning by iteratively training on a dataset of expert demonstrations. In each iteration, the algorithm first collects data by rolling out the learned policy $\pi _{\theta }$. Then, it queries the expert for the optimal action $a_{t}^{*}$ on each observation $o_{t}$ encountered during the rollout. Finally, it aggregates the new data into the dataset$$D\leftarrow D\cup \{(o_{1},a_{1}^{*}),(o_{2},a_{2}^{*}),...,(o_{T},a_{T}^{*})\}$$and trains a new policy on the aggregated dataset.

Decision transformer

Architecture diagram of the decision transformer.

The Decision Transformer approach models reinforcement learning as a sequence modelling problem. Similar to Behavior Cloning, it trains a sequence model, such as a Transformer, that models rollout sequences $$(R_{1},o_{1},a_{1}),(R_{2},o_{2},a_{2}),\dots ,(R_{t},o_{t},a_{t}),$$where $R_{t}=r_{t}+r_{t+1}+\dots +r_{T}$ is the sum of future reward in the rollout. During training time, the sequence model is trained to predict each action $a_{t}$, given the previous rollout as context:$$(R_{1},o_{1},a_{1}),(R_{2},o_{2},a_{2}),\dots ,(R_{t},o_{t})$$During inference time, to use the sequence model as an effective controller, it is simply given a very high reward prediction $R$, and it would generalize by predicting an action that would result in the high reward. This was shown to scale predictably to a Transformer with 1 billion parameters that is superhuman on 41 Atari games.

Other approaches

See for more examples.

Related approaches

Inverse Reinforcement Learning (IRL) learns a reward function that explains the expert's behavior and then uses reinforcement learning to find a policy that maximizes this reward.

Generative Adversarial Imitation Learning (GAIL) uses generative adversarial networks (GANs) to match the distribution of agent behavior to the distribution of expert demonstrations. It extends a previous approach using game theory.

See also

• Reinforcement learning
• Supervised learning
• Inverse reinforcement learning

Further reading

• Hussein, Ahmed; Gaber, Mohamed Medhat; Elyan, Eyad; Jayne, Chrisina (2018-03-31). "Imitation Learning: A Survey of Learning Methods". ACM Computing Surveys. 50 (2): 1–35. doi:10.1145/3054912. hdl:10059/2298. ISSN 0360-0300.