Learning Control Admissibility Models with Graph Neural Networks for Multi-Agent NavigationDownload PDF

16 Jun 2022, 10:45 (modified: 28 Sept 2022, 15:23)CoRL 2022 PosterReaders: Everyone
Student First Author: yes
Keywords: Graph Neural Networks, Multi-Agent Navigation
TL;DR: Control Admissibility Model decouples the properties of goal-reaching and collision avoidance, and is compositional which enables generalization to challengingly high density of agents.
Abstract: Deep reinforcement learning in continuous domains focuses on learning control policies that map states to distributions over actions that ideally concentrate on the optimal choices in each step. In multi-agent navigation problems, the optimal actions depend heavily on the agents' density. Their interaction patterns grow exponentially with respect to such density, making it hard for learning-based methods to generalize. We propose to switch the learning objectives from predicting the optimal actions to predicting sets of admissible actions, which we call control admissibility models (CAMs), such that they can be easily composed and used for online inference for an arbitrary number of agents. We design CAMs using graph neural networks and develop training methods that optimize the CAMs in the standard model-free setting, with the additional benefit of eliminating the need for reward engineering typically required to balance collision avoidance and goal-reaching requirements. We evaluate the proposed approach in multi-agent navigation environments. We show that the CAM models can be trained in environments with only a few agents and be easily composed for deployment in dense environments with hundreds of agents, achieving better performance than state-of-the-art methods.
Supplementary Material: zip
Website: https://rainorangelemon.github.io/CoRL2022/
Code: https://github.com/rainorangelemon/pytorch_geometric_multiagent
11 Replies