Leader-based Pre-training Framework for Cooperative Multi-Agent Reinforcement LearningDownload PDF

26 May 2022, 20:09 (modified: 23 Jul 2022, 02:24)ICML 2022 Pre-training WorkshopReaders: Everyone
Keywords: multi-agent reinforcement learning, pre-training, leader-based learning
TL;DR: Our contribution is proposing a successful leader-based pre-training framework for MARL which performs and generalizes better.
Abstract: A leader in the team enables efficient learning for other novices in the social learning setting for both humans and animals. This paper constructs the leader-based pre-training framework for Multi-Agent Reinforcement Learning and investigates whether the leader enables the learning of novices as well. We compare three different approaches to distilling a leader's experiences from the pre-training model: Linear Layer Dimension Reduction, Attentive Graph Pooling, and Attention-based Graph Neural Network. We successfully show that a leader-based pre-training framework can 1) enable agents to learn faster, cooperate more effectively, and escape local optimum, and 2) promote the generalizability of agents in more challenging and unseen environments. The key to effective distillation is to maintain and aggregate important information.
0 Replies