Learning Invariances for Policy Generalization

09/07/2018 ∙ by Remi Tachet des Combes, et al. ∙ 0

While recent progress has spawned very powerful machine learning systems, those agents remain extremely specialized and fail to transfer the knowledge they gain to similar yet unseen tasks. In this paper, we study a simple reinforcement learning problem and focus on learning policies that encode the proper invariances for generalization to different settings. We evaluate three potential methods for policy generalization: data augmentation, meta-learning and adversarial training. We find our data augmentation method to be effective, and study the potential of meta-learning and adversarial learning as alternative task-agnostic approaches. Keywords: reinforcement learning, generalization, data augmentation, meta-learning, adversarial learning.



page 3

This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.