Scalable Imitation Learning for Lifelong Multi-Agent Path Finding - Robotics Institute Carnegie Mellon University
Loading Events

MSR Thesis Defense

July

28
Mon
He Jiang MSR Student / Research Assistant Robotics Institute,
Carnegie Mellon University
Monday, July 28
1:15 pm to 2:45 pm
NSH 4305
Scalable Imitation Learning for Lifelong Multi-Agent Path Finding

Abstract:
With the advancement of AI and robotics, autonomous agents are becoming an essential part of our daily lives. Future large-scale multi-agent systems—such as those envisioned in smart cities—may involve tens of thousands of ground vehicles, aerial drones, and diverse robots, all requiring highly efficient coordination.

This thesis focuses on a fundamental coordination problem underlying many such systems: Lifelong Multi-Agent Path Finding (LMAPF). LMAPF is the task of constantly planning efficient, collision-free paths for multiple agents, which get new goals every time they reach their current ones. Despite decades of research in related areas, the rapid increase in the number of agents in modern systems demands increasingly scalable algorithms—capable, for example, of coordinating thousands of mobile robots in automated warehouses operated by companies such as Amazon and Ocado.

Therefore, we introduce an algorithm named Scalable Imitation Learning for LMAPF (SILLM), which can effectively learn a neural policy shared by up to ten thousand agents to generate high-quality single-step decisions based on their local observations. One of the key ingredients in our learning recipe is the scalable expert, a windowed anytime search-based solver named Windowed PIBT-LNS (WPL), originally proposed as a part of our winning solution in the 2023 League of Robot Runners (LoRR) competition. To emulate WPL’s decision-making logic, we carefully design a neural policy representation that integrates a novel Spatially Sensitive Communication (SSC) module that allows for precise spatial reasoning, with other heuristic search progresses on single-step collision resolution and global heuristic guidance.

Our algorithm substantially outperforms state-of-the-art learning-based solvers in both inference speed and solution quality. When planning time is limited, it also surpasses existing search-based methods, including our competition-winning solution.

Committee:
Prof. Jiaoyang Li (advisor)
Prof. Maxim Likhachev
Rishi Veerapaneni