English

Scalable and Reliable Multi-agent Reinforcement Learning for Traffic Assignment

Machine Learning 2025-12-16 v1

Abstract

The evolution of metropolitan cities and the increase in travel demands impose stringent requirements on traffic assignment methods. Multi-agent reinforcement learning (MARL) approaches outperform traditional methods in modeling adaptive routing behavior without requiring explicit system dynamics, which is beneficial for real-world deployment. However, MARL frameworks face challenges in scalability and reliability when managing extensive networks with substantial travel demand, which limiting their practical applicability in solving large-scale traffic assignment problems. To address these challenges, this study introduces MARL-OD-DA, a new MARL framework for the traffic assignment problem, which redefines agents as origin-destination (OD) pair routers rather than individual travelers, significantly enhancing scalability. Additionally, a Dirichlet-based action space with action pruning and a reward function based on the local relative gap are designed to enhance solution reliability and improve convergence efficiency. Experiments demonstrate that the proposed MARL framework effectively handles medium-sized networks with extensive and varied city-level OD demand, surpassing existing MARL methods. When implemented in the SiouxFalls network, MARL-OD-DA achieves better assignment solutions in 10 steps, with a relative gap that is 94.99% lower than that of conventional methods.

Keywords

Cite

@article{arxiv.2506.17029,
  title  = {Scalable and Reliable Multi-agent Reinforcement Learning for Traffic Assignment},
  author = {Leizhen Wang and Peibo Duan and Cheng Lyu and Zewen Wang and Zhiqiang He and Nan Zheng and Zhenliang Ma},
  journal= {arXiv preprint arXiv:2506.17029},
  year   = {2025}
}
R2 v1 2026-07-01T03:26:41.040Z