163 citations · 242 across the 8 of their papers we have counts for
3 papers · 1 filter
Agent-Centric Representations for Multi-Agent Reinforcement Learning
Wenling Shang, Lasse Espeholt, Anton Raichuk +1
Object-centric representations have recently enabled significant progress in tackling relational reasoning tasks. By building a strong object-centric inductive bias into neural arc…
GEM: Group Enhanced Model for Learning Dynamical Control Systems
Philippe Hansen-Estruch, Wenling Shang, Lerrel Pinto +2
Learning the dynamics of a physical system wherein an autonomous agent operates is an important task. Often these systems present apparent geometric structures. For instance, the t…
Reinforcement Learning with Latent Flow
Wenling Shang, Xiaofei Wang, Aravind Srinivas +4
Temporal information is essential to learning effective policies with Reinforcement Learning (RL). However, current state-of-the-art RL algorithms either assume that such informati…