Benchmarking Model-Based Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Model-based reinforcement learning (MBRL) is widely seen as having the potential to be significantly more sample efficient than model-free RL. However, research in model-based RL has not been very standardized. It is fairly common for authors to experiment with self-designed environments, and there are several separate lines of research, which are sometimes closed-sourced or not reproducible. Accordingly, it is an open question how these various existing MBRL algorithms perform relative to each other. To facilitate research in MBRL, in this paper we gather a wide collection of MBRL algorithms and propose over 18 benchmarking environments specially designed for MBRL. We benchmark these algorithms with unified problem settings, including noisy environments. Beyond cataloguing performance, we explore and unify the underlying algorithmic differences across MBRL algorithms. We characterize three key research challenges for future MBRL research: the dynamics bottleneck, the planning horizon dilemma, and the early-termination dilemma. Finally, to maximally facilitate future research on MBRL, we open-source our benchmark in http://www.cs.toronto.edu/~tingwuwang/mbrl.html.
Key facts. Cited by 158 papers in the map (data through 7 Oct 2026). First posted on arXiv 3 Jul 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2019: 1; 2020: 4; 2021: 41; 2022: 35; 2023: 27; 2024: 23; 2025: 21; 2026: 6.
158 papers.
Sort:
By area:
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,023
- 2021What Matters in Learning from Offline Human Demonstrations for Robot Manipulationcited by 699
- 2020MOPO: Model-based Offline Policy Optimizationcited by 630
- 2020MOReL : Model-Based Offline Reinforcement Learningcited by 582
- 2021Randomized Ensembled Double Q-Learning: Learning Fast Without a Modelcited by 278
- 2021What Matters for Adversarial Imitation Learning?cited by 70
- 2022A Survey on Model-based Reinforcement Learningcited by 65
- 2021On the Importance of Hyperparameter Optimization for Model-based Reinforcement Learningcited by 59
- 2024Bridging State and History Representations: Understanding Self-Predictive RLcited by 52
- 2022VRL3: A Data-Driven Framework for Visual Deep Reinforcement Learningcited by 51
- 2022Making Linear MDPs Practical via Contrastive Representation Learningcited by 49
- 2022Investigating Compounding Prediction Errors in Learned Dynamics Modelscited by 47
38 more
- 2021MobILE: Model-Based Imitation Learning From Observation Alonecited by 44
- 2021safe-control-gym: a Unified Benchmark Suite for Safe Learning-based Control and Reinforcement Learning in Roboticscited by 42
- 2021MBRL-Lib: A Modular Library for Model-based Reinforcement Learningcited by 41
- 2022Physical Design using Differentiable Learned Simulatorscited by 37
- 2022Spectral Decomposition Representation for Reinforcement Learningcited by 35
- 2023World Models and Predictive Coding for Cognitive and Developmental Robotics: Frontiers and Challengescited by 35
- 2021Deep reinforcement learning in medical imaging: A literature reviewcited by 34
- 2022Generalizing Goal-Conditioned Reinforcement Learning with Variational Causal Reasoningcited by 33
- 2021Autoregressive Dynamics Models for Offline Policy Evaluation and Optimizationcited by 32
- 2024Diffusion World Model: Future Modeling Beyond Step-by-Step Rollout for Offline Reinforcement Learningcited by 31
- 2021Procedural Generalization by Planning with Self-Supervised World Modelscited by 30
- 2023Rethinking Closed-loop Training for Autonomous Drivingcited by 29
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objectivecited by 28
- 2020Differentiable Physics Models for Real-world Offline Model-based Reinforcement Learningcited by 27
- 2021A Consciousness-Inspired Planning Agent for Model-Based Reinforcement Learningcited by 27
- 2022Model-based Safe Deep Reinforcement Learning via a Constrained Proximal Policy Optimization Algorithmcited by 27
- 2023RL-ViGen: A Reinforcement Learning Benchmark for Visual Generalizationcited by 27
- 2022Sample Efficient Deep Reinforcement Learning via Uncertainty Estimationcited by 26
- 2023CCIL: Continuity-based Data Augmentation for Corrective Imitation Learningcited by 26
- 2022Deep Koopman Operator with Control for Nonlinear Systemscited by 25
- 2021Model-based Multi-agent Policy Optimization with Adaptive Opponent-wise Rolloutscited by 24
- 2021A Free Lunch from the Noise: Provable and Practical Exploration for Representation Learningcited by 24
- 2024SINDy-RL: Interpretable and Efficient Model-Based Reinforcement Learningcited by 24
- 2021Sample-Efficient Reinforcement Learning via Conservative Model-Based Actor-Criticcited by 23
- 2025AdaWM: Adaptive World Model based Planning for Autonomous Drivingcited by 23
- 2021Physics-informed Dyna-Style Model-Based Deep Reinforcement Learning for Dynamic Controlcited by 22
- 2022SAM-RL: Sensing-Aware Model-Based Reinforcement Learning via Differentiable Physics-Based Simulation and Renderingcited by 22
- 2023Causal Reinforcement Learning: A Surveycited by 22
- 2021Dynamics-Aware Quality-Diversity for Efficient Learning of Skill Repertoirescited by 20
- 2023Seizing Serendipity: Exploiting the Value of Past Success in Off-Policy Actor-Criticcited by 20
- 2025Learning Transformer-based World Models with Contrastive Predictive Codingcited by 20
- 2025Synthesis of Model Predictive Control and Reinforcement Learning: Survey and Classificationcited by 19
- 2021Learning Modular Robot Control Policiescited by 18
- 2022Model-based Multi-agent Reinforcement Learning: Recent Progress and Prospectscited by 17
- 2021MAMBPO: Sample-efficient multi-robot reinforcement learning using learned world modelscited by 16
- 2021SAGCI-System: Towards Sample-Efficient, Generalizable, Compositional, and Incremental Robot Learningcited by 16
- 2021Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learningcited by 15
- 2024Diffusion Spectral Representation for Reinforcement Learningcited by 15
50 most cited of 158 All 158 (JSON)
- 2026Agentic High-Dimensional Bayesian Optimization with Hypothesis- and Evidence-Guided Search28 Sep 2026
- 2026Breaking the Computational Barrier: Provably Efficient Actor-Critic for Low-Rank MDPs2 May 2026
- 2026Advantage-Guided Diffusion for Model-Based Reinforcement Learning10 Apr 2026
- 2026Soft MPCritic: Amortized Model Predictive Value Iteration1 Apr 2026
- 2026Multi-Agent Model-Based Reinforcement Learning with Joint State-Action Learned Embeddings13 Feb 2026
- 2026Laplacian Representations for Decision-Time Planning4 Feb 2026
- 2025Spectral Representation-based Reinforcement Learning17 Dec 2025
- 2025D2 Actor Critic: Diffusion Actor Meets Distributional Critic3 Oct 2025
- 2025Emergent World Representations in OpenVLA29 Sep 2025
- 2025Adversarial Diffusion for Robust Reinforcement Learning28 Sep 2025
- 2025Tree-Guided Diffusion Planner29 Aug 2025
- 2025Model-free Reinforcement Learning for Model-based Control: Towards Safe, Interpretable and Sample-efficient Agents17 Jul 2025
38 more
- 2025GenAI-based Multi-Agent Reinforcement Learning towards Distributed Agent Intelligence: A Generative-RL Agent Perspective13 Jul 2025
- 2025Ego-centric Learning of Communicative World Models for Autonomous Driving9 Jun 2025
- 2025Bregman Centroid Guided Cross-Entropy Method2 Jun 2025
- 2025Is Single-View Mesh Reconstruction Ready for Robotics?23 May 2025
- 2025Reinforcement Twinning for Hybrid Control of Flapping-Wing Drones21 May 2025
- 2025Improving planning and MBRL with temporally-extended actions21 May 2025
- 2025Look Before Leap: Look-Ahead Planning with Uncertainty in Reinforcement Learning26 Mar 2025
- 2025Sample-Efficient Reinforcement Learning of Koopman eNMPC24 Mar 2025
- 2025Deep Belief Markov Models for POMDP Inference17 Mar 2025
- 2025Learning Transformer-based World Models with Contrastive Predictive Coding6 Mar 2025
- 2025Implicit Search via Discrete Diffusion: A Study on Chess27 Feb 2025
- 2025A New Paradigm in Tuning Learned Indexes: A Reinforcement Learning Enhanced Approach7 Feb 2025
- 2025Synthesis of Model Predictive Control and Reinforcement Learning: Survey and Classification4 Feb 2025
- 2025Benchmarking Quantum Reinforcement Learning27 Jan 2025
- 2025AdaWM: Adaptive World Model based Planning for Autonomous Driving22 Jan 2025
- 2024Learning to Navigate in Mazes with Novel Layouts using Abstract Top-down Maps16 Dec 2024
- 2024MGDA: Model-based Goal Data Augmentation for Offline Goal-conditioned Weighted Supervised Learning16 Dec 2024
- 2024Dynamical-VAE-based Hindsight to Learn the Causal Dynamics of Factored-POMDPs12 Nov 2024
- 2024Interpretable and Efficient Data-driven Discovery and Control of Distributed Systems6 Nov 2024
- 2024Stepping Out of the Shadows: Reinforcement Learning in Shadow Mode30 Oct 2024
- 2024When to Trust Your Data: Enhancing Dyna-Style Model-Based Reinforcement Learning With Data Filter16 Oct 2024
- 2024D5RL: Diverse Datasets for Data-Driven Deep Reinforcement Learning15 Aug 2024
- 2024How to Choose a Reinforcement-Learning Algorithm30 Jul 2024
- 2024Learning Long-Horizon Predictions for Quadrotor Dynamics17 Jul 2024
- 2024Satisficing Exploration for Deep Reinforcement Learning16 Jul 2024
- 2024Diffusion Spectral Representation for Reinforcement Learning23 Jun 2024
- 2024MuDreamer: Learning Predictive World Models without Reconstruction23 May 2024
- 2024Efficient Multi-agent Reinforcement Learning by Planning20 May 2024
- 2024UDUC: An Uncertainty-driven Approach for Learning-based Robust Control4 May 2024
- 2024A Differential and Pointwise Control Approach to Reinforcement Learning24 Apr 2024
- 2024Imitation Game: A Model-based and Imitation Learning Deep Reinforcement Learning Hybrid2 Apr 2024
- 2024Robust Model Based Reinforcement Learning Using L1 Adaptive Control21 Mar 2024
- 2024SINDy-RL: Interpretable and Efficient Model-Based Reinforcement Learning14 Mar 2024
- 2024Diffusion World Model: Future Modeling Beyond Step-by-Step Rollout for Offline Reinforcement Learning5 Feb 2024
- 2024Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learning5 Feb 2024
- 2024Machine learning for industrial sensing and control: A survey and practical perspective24 Jan 2024
- 2024Bridging State and History Representations: Understanding Self-Predictive RL17 Jan 2024
- 2024Unsupervised Salient Patch Selection for Data-Efficient Reinforcement Learning10 Jan 2024
50 newest of 158 All 158 (JSON)
- 2022A Survey on Model-based Reinforcement Learningcited by 65
- 2023World Models and Predictive Coding for Cognitive and Developmental Robotics: Frontiers and Challengescited by 35
- 2024Diffusion World Model: Future Modeling Beyond Step-by-Step Rollout for Offline Reinforcement Learningcited by 31
- 2021Procedural Generalization by Planning with Self-Supervised World Modelscited by 30
- 2025AdaWM: Adaptive World Model based Planning for Autonomous Drivingcited by 23
- 2025Learning Transformer-based World Models with Contrastive Predictive Codingcited by 20
- 2021MAMBPO: Sample-efficient multi-robot reinforcement learning using learned world modelscited by 16
- 2024MuDreamer: Learning Predictive World Models without Reconstructioncited by 13
- 2023Novelty Detection in Reinforcement Learning with World Modelscited by 8
- 2025Emergent World Representations in OpenVLAcited by 6
- 2023MABL: Bi-Level Latent-Variable World Model for Sample-Efficient Multi-Agent Reinforcement Learningcited by 4
- 2023Go Beyond Imagination: Maximizing Episodic Reachability with World Modelscited by 3
12 most cited of 16 All 16 (JSON)
- 2026Advantage-Guided Diffusion for Model-Based Reinforcement Learning10 Apr 2026
- 2025Emergent World Representations in OpenVLA29 Sep 2025
- 2025Ego-centric Learning of Communicative World Models for Autonomous Driving9 Jun 2025
- 2025Learning Transformer-based World Models with Contrastive Predictive Coding6 Mar 2025
- 2025AdaWM: Adaptive World Model based Planning for Autonomous Driving22 Jan 2025
- 2024MuDreamer: Learning Predictive World Models without Reconstruction23 May 2024
- 2024Diffusion World Model: Future Modeling Beyond Step-by-Step Rollout for Offline Reinforcement Learning5 Feb 2024
- 2023Novelty Detection in Reinforcement Learning with World Models12 Oct 2023
- 2023Go Beyond Imagination: Maximizing Episodic Reachability with World Models25 Aug 2023
- 2023MABL: Bi-Level Latent-Variable World Model for Sample-Efficient Multi-Agent Reinforcement Learning12 Apr 2023
- 2023World Models and Predictive Coding for Cognitive and Developmental Robotics: Frontiers and Challenges14 Jan 2023
- 2022Model-based Reinforcement Learning with a Hamiltonian Canonical ODE Network2 Nov 2022
12 newest of 16 All 16 (JSON)
- 2021What Matters in Learning from Offline Human Demonstrations for Robot Manipulationcited by 699
- 2020MOPO: Model-based Offline Policy Optimizationcited by 630
- 2020MOReL : Model-Based Offline Reinforcement Learningcited by 582
- 2021What Matters for Adversarial Imitation Learning?cited by 70
- 2021MobILE: Model-Based Imitation Learning From Observation Alonecited by 44
- 2020Differentiable Physics Models for Real-world Offline Model-based Reinforcement Learningcited by 27
- 2023CCIL: Continuity-based Data Augmentation for Corrective Imitation Learningcited by 26
- 2024D5RL: Diverse Datasets for Data-Driven Deep Reinforcement Learningcited by 11
- 2021PerSim: Data-Efficient Offline Reinforcement Learning with Heterogeneous Agents via Personalized Simulatorscited by 10
- 2023Neural Laplace Control for Continuous-time Delayed Systemscited by 10
- 2023Improving and Benchmarking Offline Reinforcement Learning Algorithmscited by 6
- 2019Multi-task Batch Reinforcement Learning with Metric Learningcited by 1
12 most cited of 15 All 15 (JSON)
- 2024MGDA: Model-based Goal Data Augmentation for Offline Goal-conditioned Weighted Supervised Learning16 Dec 2024
- 2024D5RL: Diverse Datasets for Data-Driven Deep Reinforcement Learning15 Aug 2024
- 2024Imitation Game: A Model-based and Imitation Learning Deep Reinforcement Learning Hybrid2 Apr 2024
- 2024Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learning5 Feb 2024
- 2023CCIL: Continuity-based Data Augmentation for Corrective Imitation Learning19 Oct 2023
- 2023Improving and Benchmarking Offline Reinforcement Learning Algorithms1 Jun 2023
- 2023Neural Laplace Control for Continuous-time Delayed Systems24 Feb 2023
- 2021What Matters in Learning from Offline Human Demonstrations for Robot Manipulation6 Aug 2021
- 2021What Matters for Adversarial Imitation Learning?1 Jun 2021
- 2021MobILE: Model-Based Imitation Learning From Observation Alone22 Feb 2021
- 2021PerSim: Data-Efficient Offline Reinforcement Learning with Heterogeneous Agents via Personalized Simulators13 Feb 2021
- 2020Differentiable Physics Models for Real-world Offline Model-based Reinforcement Learning3 Nov 2020
12 newest of 15 All 15 (JSON)
- 2024Diffusion Spectral Representation for Reinforcement Learningcited by 15
- 2025Implicit Search via Discrete Diffusion: A Study on Chesscited by 11
- 2025Tree-Guided Diffusion Plannercited by 4
- 2025Adversarial Diffusion for Robust Reinforcement Learningcited by 4
- 2025D2 Actor Critic: Diffusion Actor Meets Distributional Criticcited by 0
- 2025D2 Actor Critic: Diffusion Actor Meets Distributional Critic3 Oct 2025
- 2025Adversarial Diffusion for Robust Reinforcement Learning28 Sep 2025
- 2025Tree-Guided Diffusion Planner29 Aug 2025
- 2025Implicit Search via Discrete Diffusion: A Study on Chess27 Feb 2025
- 2024Diffusion Spectral Representation for Reinforcement Learning23 Jun 2024
- 2023Rethinking Closed-loop Training for Autonomous Drivingcited by 29
- 2021Safe Model-based Off-policy Reinforcement Learning for Eco-Driving in Connected and Automated Hybrid Electric Vehiclescited by 6
- 2022Driver Dojo: A Benchmark for Generalizable Reinforcement Learning for Autonomous Drivingcited by 2
- 2023Rethinking Closed-loop Training for Autonomous Driving27 Jun 2023
- 2022Driver Dojo: A Benchmark for Generalizable Reinforcement Learning for Autonomous Driving23 Jul 2022
- 2021Safe Model-based Off-policy Reinforcement Learning for Eco-Driving in Connected and Automated Hybrid Electric Vehicles25 May 2021
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,023
- 2021Randomized Ensembled Double Q-Learning: Learning Fast Without a Modelcited by 278
- 2021On the Importance of Hyperparameter Optimization for Model-based Reinforcement Learningcited by 59
- 2024Bridging State and History Representations: Understanding Self-Predictive RLcited by 52
- 2022VRL3: A Data-Driven Framework for Visual Deep Reinforcement Learningcited by 51
- 2022Making Linear MDPs Practical via Contrastive Representation Learningcited by 49
- 2022Investigating Compounding Prediction Errors in Learned Dynamics Modelscited by 47
- 2021safe-control-gym: a Unified Benchmark Suite for Safe Learning-based Control and Reinforcement Learning in Roboticscited by 42
- 2021MBRL-Lib: A Modular Library for Model-based Reinforcement Learningcited by 41
- 2022Physical Design using Differentiable Learned Simulatorscited by 37
- 2022Spectral Decomposition Representation for Reinforcement Learningcited by 35
- 2021Deep reinforcement learning in medical imaging: A literature reviewcited by 34
12 most cited of 117 All 117 (JSON)
- 2026Breaking the Computational Barrier: Provably Efficient Actor-Critic for Low-Rank MDPs2 May 2026
- 2026Soft MPCritic: Amortized Model Predictive Value Iteration1 Apr 2026
- 2026Multi-Agent Model-Based Reinforcement Learning with Joint State-Action Learned Embeddings13 Feb 2026
- 2026Laplacian Representations for Decision-Time Planning4 Feb 2026
- 2025Spectral Representation-based Reinforcement Learning17 Dec 2025
- 2025Model-free Reinforcement Learning for Model-based Control: Towards Safe, Interpretable and Sample-efficient Agents17 Jul 2025
- 2025GenAI-based Multi-Agent Reinforcement Learning towards Distributed Agent Intelligence: A Generative-RL Agent Perspective13 Jul 2025
- 2025Bregman Centroid Guided Cross-Entropy Method2 Jun 2025
- 2025Is Single-View Mesh Reconstruction Ready for Robotics?23 May 2025
- 2025Reinforcement Twinning for Hybrid Control of Flapping-Wing Drones21 May 2025
- 2025Improving planning and MBRL with temporally-extended actions21 May 2025
- 2025Look Before Leap: Look-Ahead Planning with Uncertainty in Reinforcement Learning26 Mar 2025
12 newest of 117 All 117 (JSON)