Bayesian Multi-type Mean Field Multi-agent Imitation Learning
Fan Yang, Alina Vereshchaka, Changyou Chen, Wen Dong
Spotlight presentation: Orals & Spotlights Track 04: Reinforcement Learning
on 2020-12-07T20:00:00-08:00 - 2020-12-07T20:10:00-08:00
on 2020-12-07T20:00:00-08:00 - 2020-12-07T20:10:00-08:00
Toggle Abstract Paper (in Proceedings / .pdf)
Abstract: Multi-agent Imitation learning (MAIL) refers to the problem that agents learn to perform a task interactively in a multi-agent system through observing and mimicking expert demonstrations, without any knowledge of a reward function from the environment. MAIL has received a lot of attention due to promising results achieved on synthesized tasks, with the potential to be applied to complex real-world multi-agent tasks. Key challenges for MAIL include sample efficiency and scalability. In this paper, we proposed Bayesian multi-type mean field multi-agent imitation learning (BM3IL). Our method improves sample efficiency through establishing a Bayesian formulation for MAIL, and enhances scalability through introducing a new multi-type mean field approximation. We demonstrate the performance of our algorithm through benchmarking with three state-of-the-art multi-agent imitation learning algorithms on several tasks, including solving a multi-agent traffic optimization problem in a real-world transportation network. Experimental results indicate that our algorithm significantly outperforms all other algorithms in all scenarios.