MAMBPO: Sample-efficient multi-robot reinforcement learning using learned world models
On the map: World models → Latent dynamics models and model-based RL
Abstract (arXiv)
Multi-robot systems can benefit from reinforcement learning (RL) algorithms that learn behaviours in a small number of trials, a property known as sample efficiency. This research thus investigates the use of learned world models to improve sample efficiency. We present a novel multi-agent model-based RL algorithm: Multi-Agent Model-Based Policy Optimization (MAMBPO), utilizing the Centralized Learning for Decentralized Execution (CLDE) framework. CLDE algorithms allow a group of agents to act in a fully decentralized manner after training. This is a desirable property for many systems comprising of multiple robots. MAMBPO uses a learned world model to improve sample efficiency compared to model-free Multi-Agent Soft Actor-Critic (MASAC). We demonstrate this on two simulated multi-robot tasks, where MAMBPO achieves a similar performance to MASAC, but requires far fewer samples to do so. Through this, we take an important step towards making real-life learning for multi-robot systems possible.
Key facts. Cited by 16 papers in the map (data through 7 Oct 2026). Placed in Latent dynamics models and model-based RL, one of 11 directions in World models. Builds on 11 of its 17 references that are in the map. First posted on arXiv 5 Mar 2021.
Builds on
11 of 17 references.
Closest ancestors · 2
Same direction, newest first
- 2019When to Trust Your Model: Model-Based Policy Optimization19 Jun
- 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Models30 May
Other references · 9
- 2017Multi-Agent Actor-Critic for Mixed Cooperative-Competitive Environmentscited by 1,519
- 2017Domain Randomization for Transferring Deep Neural Networks from Simulation to the Real Worldcited by 1,401
- 2018QMIX: Monotonic Value Function Factorisation for Deep Multi-Agent Reinforcement Learningcited by 697
- 2018Actor-Attention-Critic for Multi-Agent Reinforcement Learningcited by 259
- 2017Emergence of Grounded Compositional Language in Multi-Agent Populationscited by 257
- 2019Benchmarking Model-Based Reinforcement Learningcited by 158
- 2017Deep Decentralized Multi-task Multi-Agent Reinforcement Learning under Partial Observabilitycited by 143
- 2017Regularization for Deep Learning: A Taxonomycited by 91
- 2019Multi-Agent Reinforcement Learning with Multi-Step Generative Modelscited by 18
Built on by
Citing papers per year: 2021: 3; 2022: 3; 2023: 5; 2024: 3; 2025: 2.
16 papers.
Sort:
By area:
- 2021Model-Based Opponent Modelingcited by 21
- 2023A Model-Based Solution to the Offline Multi-Agent Reinforcement Learning Coordination Problemcited by 18
- 2022Model-based Multi-agent Reinforcement Learning: Recent Progress and Prospectscited by 17
- 2021Centralized Model and Exploration Policy for Multi-Agent RLcited by 11
- 2024Decentralized Transformers with Centralized Aggregation are Sample-Efficient Multi-Agent World Modelscited by 10
- 2024Efficient Multi-agent Reinforcement Learning by Planningcited by 9
- 2022CLAS: Coordinating Multi-Robot Manipulation with Central Latent Action Spacescited by 6
- 2024CoDreamer: Communication-Based Decentralised World Modelscited by 5
- 2023Co-Learning Empirical Games and World Modelscited by 4
- 2021Nearest-Neighbor-based Collision Avoidance for Quadrotors via Reinforcement Learningcited by 3
- 2022Centralized Training with Hybrid Execution in Multi-Agent Reinforcement Learningcited by 2
- 2023Model-Based Decentralized Policy Optimizationcited by 1
4 more
- 2025Combining Bayesian Inference and Reinforcement Learning for Agent Decision Making: A Reviewcited by 1
- 2023An Algorithm For Adversary Aware Decentralized Networked MARLcited by 0
- 2023Formal Modelling for Multi-Robot Systems Under Uncertaintycited by 0
- 2025Toward Adaptable Multi-Agent Reinforcement Learning: An Assumption-Aware Reviewcited by 0
- 2025Toward Adaptable Multi-Agent Reinforcement Learning: An Assumption-Aware Review14 Jul 2025
- 2025Combining Bayesian Inference and Reinforcement Learning for Agent Decision Making: A Review12 May 2025
- 2024Decentralized Transformers with Centralized Aggregation are Sample-Efficient Multi-Agent World Models22 Jun 2024
- 2024CoDreamer: Communication-Based Decentralised World Models19 Jun 2024
- 2024Efficient Multi-agent Reinforcement Learning by Planning20 May 2024
- 2023A Model-Based Solution to the Offline Multi-Agent Reinforcement Learning Coordination Problem26 May 2023
- 2023Formal Modelling for Multi-Robot Systems Under Uncertainty26 May 2023
- 2023Co-Learning Empirical Games and World Models23 May 2023
- 2023An Algorithm For Adversary Aware Decentralized Networked MARL9 May 2023
- 2023Model-Based Decentralized Policy Optimization16 Feb 2023
- 2022CLAS: Coordinating Multi-Robot Manipulation with Central Latent Action Spaces28 Nov 2022
- 2022Centralized Training with Hybrid Execution in Multi-Agent Reinforcement Learning12 Oct 2022
4 more
- 2022Model-based Multi-agent Reinforcement Learning: Recent Progress and Prospects20 Mar 2022
- 2021Model-Based Opponent Modeling4 Aug 2021
- 2021Centralized Model and Exploration Policy for Multi-Agent RL14 Jul 2021
- 2021Nearest-Neighbor-based Collision Avoidance for Quadrotors via Reinforcement Learning30 Apr 2021
- 2021Model-Based Opponent Modelingcited by 21
- 2022Model-based Multi-agent Reinforcement Learning: Recent Progress and Prospectscited by 17
- 2021Centralized Model and Exploration Policy for Multi-Agent RLcited by 11
- 2024Efficient Multi-agent Reinforcement Learning by Planningcited by 9
- 2022CLAS: Coordinating Multi-Robot Manipulation with Central Latent Action Spacescited by 6
- 2021Nearest-Neighbor-based Collision Avoidance for Quadrotors via Reinforcement Learningcited by 3
- 2022Centralized Training with Hybrid Execution in Multi-Agent Reinforcement Learningcited by 2
- 2023Model-Based Decentralized Policy Optimizationcited by 1
- 2025Combining Bayesian Inference and Reinforcement Learning for Agent Decision Making: A Reviewcited by 1
- 2023An Algorithm For Adversary Aware Decentralized Networked MARLcited by 0
- 2023Formal Modelling for Multi-Robot Systems Under Uncertaintycited by 0
- 2025Toward Adaptable Multi-Agent Reinforcement Learning: An Assumption-Aware Reviewcited by 0
- 2025Toward Adaptable Multi-Agent Reinforcement Learning: An Assumption-Aware Review14 Jul 2025
- 2025Combining Bayesian Inference and Reinforcement Learning for Agent Decision Making: A Review12 May 2025
- 2024Efficient Multi-agent Reinforcement Learning by Planning20 May 2024
- 2023Formal Modelling for Multi-Robot Systems Under Uncertainty26 May 2023
- 2023An Algorithm For Adversary Aware Decentralized Networked MARL9 May 2023
- 2023Model-Based Decentralized Policy Optimization16 Feb 2023
- 2022CLAS: Coordinating Multi-Robot Manipulation with Central Latent Action Spaces28 Nov 2022
- 2022Centralized Training with Hybrid Execution in Multi-Agent Reinforcement Learning12 Oct 2022
- 2022Model-based Multi-agent Reinforcement Learning: Recent Progress and Prospects20 Mar 2022
- 2021Model-Based Opponent Modeling4 Aug 2021
- 2021Centralized Model and Exploration Policy for Multi-Agent RL14 Jul 2021
- 2021Nearest-Neighbor-based Collision Avoidance for Quadrotors via Reinforcement Learning30 Apr 2021
Similar, not cited
Close in wording, no citation either way
- 2022Scalable Multi-Agent Model-Based Reinforcement Learning
- 2025Empowering Multi-Robot Cooperation via Sequential World Models
- 2024Bi-CL: A Reinforcement Learning Framework for Robots Coordination Through Bi-level Optimization
- 2025Accelerating Model-Based Reinforcement Learning with State-Space World Models
- 2026MA-WAM: Multi-Agent World-Action Model for Test-Time Planning
- 2024MAMBA: an Effective World Model Approach for Meta-Reinforcement Learning
- 2025Robotic World Model: A Neural Network Simulator for Robust Policy Optimization in Robotics
- 2026MA-VLCM: A Vision Language Critic Model for Value Estimation of Policies in Multi-Agent Team Settings