Policy Consolidation for Continual Reinforcement Learning


Christos Kaplanis, Murray Shanahan, Claudia Clopath ;
Proceedings of the 36th International Conference on Machine Learning, PMLR 97:3242-3251, 2019.


We propose a method for tackling catastrophic forgetting in deep reinforcement learning that is agnostic to the timescale of changes in the distribution of experiences, does not require knowledge of task boundaries and can adapt in continuously changing environments. In our policy consolidation model, the policy network interacts with a cascade of hidden networks that simultaneously remember the agent’s policy at a range of timescales and regularise the current policy by its own history, thereby improving its ability to learn without forgetting. We find that the model improves continual learning relative to baselines on a number of continuous control tasks in single-task, alternating two-task, and multi-agent competitive self-play settings.

Related Material