Multi-Agent Pathfinding

PRIMAL3: Pathfinding via Reinforcement and Imitation Multi-Agent Learning Leveraging LaCAM3

Chengyang He1Tanishq Duhan1Gadiel Sznaier Camps2Fangyuan Wang1Yuhong Cao1Jiankai Sun2Ge Sun1Mac Schwager2Guillaume Sartoretti1,†

1 Multi-Agent Robotic Motion Lab (MARMot), National University of Singapore

2 Multi-robot Systems Lab (MSL), Stanford University

Corresponding author

Overview

We present PRIMAL3, an ultra-large-scale learning-based framework for multi-agent pathfinding (MAPF) that integrates reinforcement learning, topology-aware communication, LaCAM3-guided training, and PIBT-based action refinement. PRIMAL3 targets failures at topologically critical states, where agents must coordinate decisively around bottlenecks, dead ends, and persistent conflicts. Each agent is represented using features derived from cut vertices, dead-end regions, shortest-path distances, and blocking estimates. Two complementary graphs capture agent interactions: a same-direction following graph propagates multihop context along compatible paths, while a different-direction conflict graph differentiates agents competing for shared space through masked attention and relative features. During training, we propose to let policy entropy identify uncertain agents, for which LaCAM3 provides confidence-triggered action interventions and label-smoothed imitation targets. During execution, a priority-aware PIBT module refines the proposed joint actions using persistent, learned, and distance-aware priorities together with policy-aware fallback preferences while maintaining collision-free execution. The resulting framework combines learned exploration with structured expert guidance without requiring LaCAM3 at inference. Experiments demonstrate that PRIMAL3 substantially outperforms state-of-the-art learning-based baselines and scales to ultra-large instances with up to city-level 100,000 agents. Real-world experiments further demonstrate the feasibility of deploying PRIMAL3 on physical robotic systems and ablation studies validate the individual contributions the components we proposed.

Experimental Results

Comparison of PRIMAL3 with search-based and learning-based MAPF solvers across random and maze maps
Benchmark comparison

PRIMAL3 achieves success rates comparable to those of state-of-the-art search-based solvers, including LNS2 and LaCAM3, substantially narrowing the long-standing performance gap between learning-based and classical MAPF methods.

Success-rate comparison between PRIMAL3 and HMAGAT as the number of agents increases
Large-scale performance

In large-scale experiments, PRIMAL3 maintains a high success rate as the number of agents increases, while HMAGAT degrades rapidly at larger scales.

Simulation at Scale

From 1,000 to 100,000 agents

1,000 agents
5,000 agents
10,000 agents
50,000 agents

Real-World Validation

Physical & mixed reality

Physical deployment
Experiment 01
8 agents

8 physical robots performing MAPF on a 10 × 10 random map.

Four stages of the eight-robot physical MAPF experiment
Experiment timeline
Mixed reality
Experiment 02
100 agents

Mixed-reality experiment with 8 physical robots and 92 virtual agents on a 32 × 32 random map.

Four stages of the 100-agent mixed-reality MAPF experiment
Experiment timeline

Resources

Code, paper, team, and citation materials are collected here.

Citation

BibTeX

Select and copy the citation below.

@article{he2026primal3,
  title={PRIMAL3: Pathfinding via Reinforcement and Imitation \\ Multi-Agent Learning - Leveraging LaCAM3},
  author={He, Chengyang and Duhan, Tanishq and Camps, Gadiel Sznaier and Wang, Fangyuan and Cao, Yuhong and Sun, Jiankai and Sun, Ge and Schwager, Mac and Sartoretti, Guillaume},
  journal={arXiv preprint arXiv:XXXX.XXXXX},
  year={2026}
}

Contact

For questions about PRIMAL3, please feel free to contact Chengyang He at chengyanghe@u.nus.edu or hecy@stanford.edu.