Reinforcement Learning with Prototypical Representations

  • 2021-02-22 18:56:34
  • Denis Yarats, Rob Fergus, Alessandro Lazaric, Lerrel Pinto
  • 28

Abstract

Learning effective representations in image-based environments is crucial forsample efficient Reinforcement Learning (RL). Unfortunately, in RL,representation learning is confounded with the exploratory experience of theagent -- learning a useful representation requires diverse data, whileeffective exploration is only possible with coherent representations.Furthermore, we would like to learn representations that not only generalizeacross tasks but also accelerate downstream exploration for efficienttask-specific training. To address these challenges we propose Proto-RL, aself-supervised framework that ties representation learning with explorationthrough prototypical representations. These prototypes simultaneously serve asa summarization of the exploratory experience of an agent as well as a basisfor representing observations. We pre-train these task-agnostic representationsand prototypes on environments without downstream task information. Thisenables state-of-the-art downstream policy learning on a set of difficultcontinuous control tasks.

 

Quick Read (beta)

loading the full paper ...