Scalable Constrained Policy Optimization for Safe Multi-agent Reinforcement Learning

Lijun Zhang, Lin Li, Weipeng Wei, Huizhong Song, Yaodong Yang +1 more
2/3/2026

Abstract

. Abstract A challenging problem in seeking to bring multi-agent reinforcement learning (MARL) techniques into real-world applications, such as autonomous driving and drone swarms, is how to control multiple agents safely and cooperatively to accomplish tasks. Most existing safe MARL methods learn the centralized value function by introducing a global state to guide safety cooperation. However, the global coupling arising from safety constraints and the exponential growth of the state-action space size limit their applicability in instant communication or computing resource-constrained systems and larger multi-agent systems. In this paper, we develop a novel scalable and theoretically-justified multi-agent constrained policy optimization method. This method integrates the rigorous bounds of the trust region method and the bounds of the truncated advantage function to provide a new local policy optimization objective for each agent. Also, we prove that the safety constraints and the joint policy improvement can be met when each agent adopts a sequential update scheme to optimize a κ -hop policy. Furthermore, we propose a practical algorithm called Scalable MAPPO-Lagrangian (Scal-MAPPO-L). The proposed method’s effectiveness is verified on a collection of benchmark tasks, and the results support our theory that decentralized training with local interactions can still improve reward performance and satisfy safe constraints.

DOISemantic Scholar

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 4 weaker matches are hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Cite this paper

@article{zhang2026scalable,
  title  = {Scalable Constrained Policy Optimization for Safe Multi-agent Reinforcement Learning},
  author = {Lijun Zhang and Lin Li and Weipeng Wei and Huizhong Song and Yaodong Yang and Jiye Liang},
  year   = {2026},
  doi    = {10.52202/079017-4400},
  url    = {https://doi.org/10.52202/079017-4400},
  journal = {NEURIPS 2024 2024}
}

Discussion