Lifelong Policy Gradient Learning of Factored Policies for Faster Training Without Forgetting

Advances in Neural Information Processing Systems (NeurIPS), 2020
PDF   Code   Video  

Abstract: Policy gradient methods have shown success in learning control policies for high-dimensional dynamical systems. Their biggest downside is the amount of exploration they require before yielding high-performing policies. In a lifelong learning setting, in which an agent is faced with multiple consecutive tasks over its lifetime, reusing information from previously seen tasks can substantially accelerate the learning of new tasks. We provide a novel method for lifelong policy gradient learning that trains lifelong function approximators directly via policy gradients, allowing the agent to benefit from accumulated knowledge throughout the entire training process. We show empirically that our algorithm learns faster and converges to better policies than single-task and lifelong learning baselines, and completely avoids catastrophic forgetting on a variety of challenging domains.

Recommended citation:

@inproceedings{mendez2020lifelong, 
author = {Mendez-Mendez, Jorge and Wang, Boyu and Eaton, Eric},
booktitle = {Advances in Neural Information Processing Systems 33 (NeurIPS-20)},
pages = {14398--14409},
title = {Lifelong Policy Gradient Learning of Factored Policies for Faster Training Without Forgetting},
year = {2020}
}