ELIGN: Expectation Alignment as a Multi-Agent Intrinsic Reward
This work addresses the problem of enabling effective coordination in multi-agent systems for real-world applications where centralized training or dense rewards are unavailable, representing a novel approach but with incremental improvements over existing intrinsic reward methods.
The paper tackles the challenge of multi-agent reinforcement learning in decentralized training or sparse reward settings by proposing ELIGN, an intrinsic reward based on expectation alignment, which enables agents to learn collaborative behaviors without external rewards or centralized training, achieving improved coordination and scaling across 6 tasks in multi-agent environments.
Modern multi-agent reinforcement learning frameworks rely on centralized training and reward shaping to perform well. However, centralized training and dense rewards are not readily available in the real world. Current multi-agent algorithms struggle to learn in the alternative setup of decentralized training or sparse rewards. To address these issues, we propose a self-supervised intrinsic reward ELIGN - expectation alignment - inspired by the self-organization principle in Zoology. Similar to how animals collaborate in a decentralized manner with those in their vicinity, agents trained with expectation alignment learn behaviors that match their neighbors' expectations. This allows the agents to learn collaborative behaviors without any external reward or centralized training. We demonstrate the efficacy of our approach across 6 tasks in the multi-agent particle and the complex Google Research football environments, comparing ELIGN to sparse and curiosity-based intrinsic rewards. When the number of agents increases, ELIGN scales well in all multi-agent tasks except for one where agents have different capabilities. We show that agent coordination improves through expectation alignment because agents learn to divide tasks amongst themselves, break coordination symmetries, and confuse adversaries. These results identify tasks where expectation alignment is a more useful strategy than curiosity-driven exploration for multi-agent coordination, enabling agents to do zero-shot coordination.