Agent-Centric Representations for Multi-Agent Reinforcement Learning
Object-centric representations have recently enabled significant progress in tackling relational reasoning tasks. By building a strong object-centric inductive bias into neural architectures, recent efforts have improved generalization and data efficiency of machine learning algorithms for these problems. One problem class involving relational reasoning that still remains under-explored is multi-agent reinforcement learning (MARL). Here we investigate whether object-centric representations are also beneficial in the fully cooperative MARL setting. Specifically, we study two ways of incorporating an agent-centric inductive bias into our RL algorithm: 1. Introducing an agent-centric attention module with explicit connections across agents 2. Adding an agent-centric unsupervised predictive objective (i.e. not using action labels), to be used as an auxiliary loss for MARL, or as the basis of a pre-training step. We evaluate these approaches on the Google Research Football environment as well as DeepMind Lab 2D. Empirically, agent-centric representation learning leads to the emergence of more complex cooperation strategies between agents as well as enhanced sample efficiency and generalization.
Code (0)
등록된 구현이 없습니다.
Tasks
Inductive BiasMulti-agent Reinforcement LearningObjectreinforcement-learningReinforcement LearningReinforcement Learning (RL)Relational ReasoningRepresentation LearningSimilar Papers 제목 키워드 기반
StepPO: Step-Aligned Policy Optimization for Agentic Reinforcement Learning
Agentic reinforcement learning (RL) is emerging as a critical post-training paradigm for improving LLM agent capabilities. Existing RL algorithms for LLMs largely follow the token-centric paradigm as in RLHF and RLVR, wh…
Reinforcement LearningInterpretable end-to-end Neurosymbolic Reinforcement Learning agents
Deep reinforcement learning (RL) agents rely on shortcut learning, preventing them from generalizing to slightly different environments. To address this problem, symbolic method, that use object-centric states, have been…
Atari GamesDeep Reinforcement LearningObjectreinforcement-learning+2Deep Reinforcement Learning via Object-Centric Attention
Deep reinforcement learning agents, trained on raw pixel inputs, often fail to generalize beyond their training environments, relying on spurious correlations and irrelevant background details. To address this issue, obj…
Deep Reinforcement LearningInductive BiasObjectreinforcement-learning+1Feature-Attending Recurrent Modules for Generalization in Reinforcement Learning
Many important tasks are defined in terms of object. To generalize across these tasks, a reinforcement learning (RL) agent needs to exploit the structure that the objects induce. Prior work has either hard-coded object-c…
Objectreinforcement-learningReinforcement LearningReinforcement Learning (RL)Self-supervised Visual Reinforcement Learning with Object-centric Representations
Autonomous agents need large repertoires of skills to act reasonably on new tasks that they have not seen before. However, acquiring these skills using only a stream of high-dimensional, unstructured, and unlabeled obser…
Objectreinforcement-learningReinforcement LearningReinforcement Learning (RL)