MALib: A Parallel Framework for Population-based Multi-agent Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Population-based multi-agent reinforcement learning (PB-MARL) refers to the series of methods nested with reinforcement learning (RL) algorithms, which produces a self-generated sequence of tasks arising from the coupled population dynamics. By leveraging auto-curricula to induce a population of distinct emergent strategies, PB-MARL has achieved impressive success in tackling multi-agent tasks. Despite remarkable prior arts of distributed RL frameworks, PB-MARL poses new challenges for parallelizing the training frameworks due to the additional complexity of multiple nested workloads between sampling, training and evaluation involved with heterogeneous policy interactions. To solve these problems, we present MALib, a scalable and efficient computing framework for PB-MARL. Our framework is comprised of three key components: (1) a centralized task dispatching model, which supports the self-generated tasks and scalable training with heterogeneous policy combinations; (2) a programming architecture named Actor-Evaluator-Learner, which achieves high parallelism for both training and sampling, and meets the evaluation requirement of auto-curriculum learning; (3) a higher-level abstraction of MARL training paradigms, which enables efficient code reuse and flexible deployments on different distributed computing paradigms. Experiments on a series of complex tasks such as multi-agent Atari Games show that MALib achieves throughput higher than 40K FPS on a single machine with 32 CPU cores; 5x speedup than RLlib and at least 3x speedup than OpenSpiel in multi-agent training tasks. MALib is publicly available at https://github.com/sjtu-marl/malib.
Key facts. Cited by 23 papers in the map (data through 7 Oct 2026). Builds on 39 of its 68 references that are in the map. First posted on arXiv 5 Jun 2021.
Builds on
39 of 68 references.
References · 39
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
- 2015Continuous control with deep reinforcement learningcited by 3,279
- 2013Playing Atari with Deep Reinforcement Learningcited by 3,211
- 2016Asynchronous Methods for Deep Reinforcement Learningcited by 2,218
- 2017Multi-Agent Actor-Critic for Mixed Cooperative-Competitive Environmentscited by 1,519
- 2019Dota 2 with Large Scale Deep Reinforcement Learningcited by 1,084
- 2018IMPALA: Scalable Distributed Deep-RL with Importance Weighted Actor-Learner Architecturescited by 866
- 2018QMIX: Monotonic Value Function Factorisation for Deep Multi-Agent Reinforcement Learningcited by 697
- 2019The StarCraft Multi-Agent Challengecited by 589
- 2017Evolution Strategies as a Scalable Alternative to Reinforcement Learningcited by 555
27 more
- 2017Ray: A Distributed Framework for Emerging AI Applicationscited by 516
- 2017Population Based Training of Neural Networkscited by 351
- 2019On the Utility of Learning about Humans for Human-AI Coordinationcited by 345
- 2018Horovod: fast and easy distributed deep learning in TensorFlowcited by 275
- 2017StarCraft II: A New Challenge for Reinforcement Learningcited by 268
- 2017A Unified Game-Theoretic Approach to Multiagent Reinforcement Learningcited by 265
- 2018Human-level performance in first-person multiplayer games with population-based deep reinforcement learningcited by 260
- 2018Actor-Attention-Critic for Multi-Agent Reinforcement Learningcited by 259
- 2019Google Research Football: A Novel Reinforcement Learning Environmentcited by 257
- 2020PettingZoo: Gym for Multi-Agent Reinforcement Learningcited by 209
- 2019OpenSpiel: A Framework for Reinforcement Learning in Gamescited by 180
- 2016Deep Reinforcement Learning from Self-Play in Imperfect-Information Gamescited by 153
- 2020Acme: A Research Framework for Distributed Reinforcement Learningcited by 146
- 2020Sample Factory: Egocentric 3D Control from Pixels at 100000 FPS with Asynchronous Reinforcement Learningcited by 84
- 2015Massively Parallel Methods for Deep Reinforcement Learningcited by 80
- 2019A Generalized Training Approach for Multiagent Learningcited by 53
- 2019α-Rank: Multi-Agent Evaluation by Evolutioncited by 50
- 2020Pipeline PSRO: A Scalable Approach for Finding Approximate Nash Equilibria in Large Gamescited by 47
- 2016Reinforcement Learning through Asynchronous Advantage Actor-Critic on a GPUcited by 46
- 2018Accelerated Methods for Deep Reinforcement Learningcited by 40
- 2021Modelling Behavioural Diversity for Learning in Open-Ended Gamescited by 39
- 2019Multi-Agent Reinforcement Learning for Order-dispatching via Order-Vehicle Distribution Matchingcited by 26
- 2019Decentralized Multi-Agent Reinforcement Learning with Networked Agents: Recent Advancescited by 23
- 2017ELF: An Extensive, Lightweight and Flexible Research Platform for Real-time Strategy Gamescited by 16
- 2020TLeague: A Framework for Competitive Self-Play based Distributed Multi-Agent Reinforcement Learningcited by 11
- 2019αα-Rank: Practically Scaling α-Rank through Stochastic Optimisationcited by 10
- 2020Fiber: A Platform for Efficient Development and Distributed Training for Reinforcement Learning and Population-Based Methodscited by 4
Built on by
Citing papers per year: 2021: 3; 2022: 6; 2023: 6; 2024: 7; 2025: 1.
23 papers.
Sort:
By area:
- 2021Trust Region Policy Optimisation in Multi-Agent Reinforcement Learningcited by 180
- 2023Heterogeneous-Agent Reinforcement Learningcited by 67
- 2023JaxMARL: Multi-Agent RL Environments and Algorithms in JAXcited by 59
- 2025MARFT: Multi-Agent Reinforcement Fine-Tuningcited by 51
- 2024Learning Multi-Agent Communication from Graph Modeling Perspectivecited by 47
- 2021Offline Pre-trained Multi-Agent Decision Transformer: One Big Sequence Model Tackles All SMAC Taskscited by 32
- 2024Multi-Agent Reinforcement Learning for Autonomous Driving: A Surveycited by 30
- 2023Order Matters: Agent-by-agent Policy Optimizationcited by 24
- 2022MARLlib: A Scalable and Efficient Multi-agent Reinforcement Learning Librarycited by 21
- 2022An Introduction to Multi-Agent Reinforcement Learning and Review of its Application to Autonomous Mobilitycited by 16
- 2024FightLadder: A Benchmark for Competitive Multi-Agent Reinforcement Learningcited by 13
- 2021Measuring the Non-Transitivity in Chesscited by 10
11 more
- 2024Policy Space Response Oracles: A Surveycited by 10
- 2023Pearl: A Production-ready Reinforcement Learning Agentcited by 9
- 2022Understanding Value Decomposition Algorithms in Deep Cooperative Multi-Agent Reinforcement Learningcited by 6
- 2023An Empirical Study on Google Research Football Multi-agent Scenarioscited by 5
- 2024Safe Multi-Agent Reinforcement Learning with Convergence to Generalized Nash Equilibriumcited by 5
- 2022MSRL: Distributed Reinforcement Learning with Dataflow Fragmentscited by 4
- 2022Distributed Deep Reinforcement Learning: A Survey and A Multi-Player Multi-Agent Learning Toolboxcited by 4
- 2022Lamarckian Platform: Pushing the Boundaries of Evolutionary Reinforcement Learning towards Asynchronous Commercial Gamescited by 3
- 2023marl-jax: Multi-Agent Reinforcement Leaning Frameworkcited by 1
- 2024PyTAG: Tabletop Games for Multi-Agent Reinforcement Learningcited by 1
- 2024Communication Learning in Multi-Agent Systems from Graph Modeling Perspectivecited by 0
- 2025MARFT: Multi-Agent Reinforcement Fine-Tuning21 Apr 2025
- 2024Safe Multi-Agent Reinforcement Learning with Convergence to Generalized Nash Equilibrium22 Nov 2024
- 2024Communication Learning in Multi-Agent Systems from Graph Modeling Perspective1 Nov 2024
- 2024Multi-Agent Reinforcement Learning for Autonomous Driving: A Survey19 Aug 2024
- 2024FightLadder: A Benchmark for Competitive Multi-Agent Reinforcement Learning4 Jun 2024
- 2024PyTAG: Tabletop Games for Multi-Agent Reinforcement Learning28 May 2024
- 2024Learning Multi-Agent Communication from Graph Modeling Perspective14 May 2024
- 2024Policy Space Response Oracles: A Survey4 Mar 2024
- 2023Pearl: A Production-ready Reinforcement Learning Agent6 Dec 2023
- 2023JaxMARL: Multi-Agent RL Environments and Algorithms in JAX16 Nov 2023
- 2023An Empirical Study on Google Research Football Multi-agent Scenarios16 May 2023
- 2023Heterogeneous-Agent Reinforcement Learning19 Apr 2023
11 more
- 2023marl-jax: Multi-Agent Reinforcement Leaning Framework24 Mar 2023
- 2023Order Matters: Agent-by-agent Policy Optimization13 Feb 2023
- 2022Distributed Deep Reinforcement Learning: A Survey and A Multi-Player Multi-Agent Learning Toolbox1 Dec 2022
- 2022MARLlib: A Scalable and Efficient Multi-agent Reinforcement Learning Library11 Oct 2022
- 2022MSRL: Distributed Reinforcement Learning with Dataflow Fragments3 Oct 2022
- 2022Lamarckian Platform: Pushing the Boundaries of Evolutionary Reinforcement Learning towards Asynchronous Commercial Games21 Sep 2022
- 2022An Introduction to Multi-Agent Reinforcement Learning and Review of its Application to Autonomous Mobility15 Mar 2022
- 2022Understanding Value Decomposition Algorithms in Deep Cooperative Multi-Agent Reinforcement Learning10 Feb 2022
- 2021Offline Pre-trained Multi-Agent Decision Transformer: One Big Sequence Model Tackles All SMAC Tasks6 Dec 2021
- 2021Measuring the Non-Transitivity in Chess22 Oct 2021
- 2021Trust Region Policy Optimisation in Multi-Agent Reinforcement Learning23 Sep 2021
- 2025MARFT: Multi-Agent Reinforcement Fine-Tuningcited by 51
- 2025MARFT: Multi-Agent Reinforcement Fine-Tuning21 Apr 2025
- 2021Trust Region Policy Optimisation in Multi-Agent Reinforcement Learningcited by 180
- 2023Heterogeneous-Agent Reinforcement Learningcited by 67
- 2023JaxMARL: Multi-Agent RL Environments and Algorithms in JAXcited by 59
- 2024Learning Multi-Agent Communication from Graph Modeling Perspectivecited by 47
- 2023Order Matters: Agent-by-agent Policy Optimizationcited by 24
- 2022MARLlib: A Scalable and Efficient Multi-agent Reinforcement Learning Librarycited by 21
- 2022An Introduction to Multi-Agent Reinforcement Learning and Review of its Application to Autonomous Mobilitycited by 16
- 2024FightLadder: A Benchmark for Competitive Multi-Agent Reinforcement Learningcited by 13
- 2021Measuring the Non-Transitivity in Chesscited by 10
- 2024Policy Space Response Oracles: A Surveycited by 10
- 2023Pearl: A Production-ready Reinforcement Learning Agentcited by 9
- 2022Understanding Value Decomposition Algorithms in Deep Cooperative Multi-Agent Reinforcement Learningcited by 6
12 most cited of 20 All 20 (JSON)
- 2024Safe Multi-Agent Reinforcement Learning with Convergence to Generalized Nash Equilibrium22 Nov 2024
- 2024Communication Learning in Multi-Agent Systems from Graph Modeling Perspective1 Nov 2024
- 2024FightLadder: A Benchmark for Competitive Multi-Agent Reinforcement Learning4 Jun 2024
- 2024PyTAG: Tabletop Games for Multi-Agent Reinforcement Learning28 May 2024
- 2024Learning Multi-Agent Communication from Graph Modeling Perspective14 May 2024
- 2024Policy Space Response Oracles: A Survey4 Mar 2024
- 2023Pearl: A Production-ready Reinforcement Learning Agent6 Dec 2023
- 2023JaxMARL: Multi-Agent RL Environments and Algorithms in JAX16 Nov 2023
- 2023An Empirical Study on Google Research Football Multi-agent Scenarios16 May 2023
- 2023Heterogeneous-Agent Reinforcement Learning19 Apr 2023
- 2023marl-jax: Multi-Agent Reinforcement Leaning Framework24 Mar 2023
- 2023Order Matters: Agent-by-agent Policy Optimization13 Feb 2023
12 newest of 20 All 20 (JSON)