MBRL-Lib: A Modular Library for Model-based Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Model-based reinforcement learning is a compelling framework for data-efficient learning of agents that interact with the world. This family of algorithms has many subcomponents that need to be carefully selected and tuned. As a result the entry-bar for researchers to approach the field and to deploy it in real-world tasks can be daunting. In this paper, we present MBRL-Lib -- a machine learning library for model-based reinforcement learning in continuous state-action spaces based on PyTorch. MBRL-Lib is designed as a platform for both researchers, to easily develop, debug and compare new algorithms, and non-expert user, to lower the entry-bar of deploying state-of-the-art algorithms. MBRL-Lib is open-source at https://github.com/facebookresearch/mbrl-lib.
Key facts. Cited by 42 papers in the map (data through 14 Oct 2026). Builds on 29 of its 39 references that are in the map. First posted on arXiv 20 Apr 2021.
Builds on
29 of 39 references.
References · 29
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
- 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,353
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,264
- 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Modelscited by 774
- 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 773
- 2017Deep Reinforcement Learning that Matterscited by 754
- 2019Model-Based Reinforcement Learning for Ataricited by 502
- 2020dm_control: Software and Tasks for Continuous Controlcited by 375
- 2018Model-Ensemble Trust-Region Policy Optimizationcited by 219
- 2019Deep Dynamics Models for Learning Dexterous Manipulationcited by 206
- 2015Learning Continuous Control Policies by Stochastic Value Gradientscited by 162
17 more
- 2019Benchmarking Model-Based Reinforcement Learningcited by 162
- 2016Continuous Deep Q-Learning with Model-based Accelerationcited by 157
- 2017Imagination-Augmented Agents for Deep Reinforcement Learningcited by 140
- 2018Plan Online, Learn Offline: Efficient Learning and Exploration via Model-Based Controlcited by 113
- 2015Gaussian Processes for Data-Efficient Learning in Robotics and Controlcited by 98
- 2020Objective Mismatch in Model-based Reinforcement Learningcited by 96
- 2019Exploring Model-based Planning with Policy Networkscited by 90
- 2021On the Importance of Hyperparameter Optimization for Model-based Reinforcement Learningcited by 59
- 2020On the role of planning in model-based deep reinforcement learningcited by 57
- 2019Model-Predictive Policy Learning with Uncertainty Regularization for Driving in Dense Trafficcited by 50
- 2019Low Level Control of a Quadrotor with Deep Model-Based Reinforcement Learningcited by 44
- 2018The Gap Between Model-Based and Model-Free Methods on the Linear Quadratic Regulator: An Asymptotic Viewpointcited by 40
- 2019The Differentiable Cross-Entropy Methodcited by 27
- 2019Imagined Value Gradients: Model-Based Policy Optimization with Transferable Latent Dynamics Modelscited by 26
- 2017Learning model-based planning from scratchcited by 22
- 2021Bellman: A Toolbox for Model-Based Reinforcement Learning in TensorFlowcited by 2
- 2019Baconian: A Unified Open-source Framework for Model-Based Reinforcement Learningcited by 1
Built on by
Citing papers per year: 2021: 4; 2022: 9; 2023: 16; 2024: 6; 2025: 5; 2026: 2.
42 papers.
Sort:
By area:
- 2022Saute RL: Almost Surely Safe Reinforcement Learning Using State Augmentationcited by 69
- 2021Revisiting Design Choices in Offline Model-Based Reinforcement Learningcited by 50
- 2022Value Gradient weighted Model-Based Reinforcement Learningcited by 32
- 2024Hybrid Inverse Reinforcement Learningcited by 32
- 2022Training Efficient Controllers via Analytic Policy Gradientcited by 22
- 2023Seizing Serendipity: Exploiting the Value of Past Success in Off-Policy Actor-Criticcited by 20
- 2026stable-worldmodel: A Platform for Reproducible World Modeling Research and Evaluationcited by 20
- 2024Trust the Model Where It Trusts Itself -- Model-Based Actor-Critic with Uncertainty-Aware Rollout Adaptioncited by 18
- 2021An Experimental Design Perspective on Model-Based Reinforcement Learningcited by 16
- 2023Model-Based Uncertainty in Value Functionscited by 15
- 2022When to Update Your Model: Constrained Model-based Reinforcement Learningcited by 14
- 2023The Virtues of Laziness in Model-based RL: A Unified Objective and Algorithmscited by 12
30 more
- 2022Inferring Smooth Control: Monte Carlo Posterior Policy Iteration with Gaussian Processescited by 10
- 2023How to Learn and Generalize From Three Minutes of Data: Physics-Constrained and Uncertainty-Aware Neural Stochastic Differential Equationscited by 10
- 2022Reinforcement Learning in Presence of Discrete Markovian Context Evolutioncited by 9
- 2023A Unified View on Solving Objective Mismatch in Model-Based Reinforcement Learningcited by 9
- 2023Pearl: A Production-ready Reinforcement Learning Agentcited by 9
- 2022Exploration via Planning for Information about the Optimal Trajectorycited by 8
- 2025Continual Reinforcement Learning by Planning with Online World Modelscited by 8
- 2023Model-Based Reparameterization Policy Gradient Methods: Theory and Practical Algorithmscited by 7
- 2023Neural Constraint Satisfaction: Hierarchical Abstraction for Combinatorial Generalization in Object Rearrangementcited by 6
- 2023Value-Distributional Model-Based Reinforcement Learningcited by 5
- 2022Continuous Monte Carlo Graph Searchcited by 4
- 2023Dynamic Update-to-Data Ratio: Minimizing World Model Overfittingcited by 4
- 2023Data-efficient Deep Reinforcement Learning for Vehicle Trajectory Controlcited by 4
- 2021Risk Sensitive Model-Based Reinforcement Learning using Uncertainty Guided Planningcited by 3
- 2023Certifiably Robust Reinforcement Learning through Model-Based Abstract Interpretationcited by 3
- 2023Goal-conditioned Offline Planning from Curious Explorationcited by 3
- 2023Model-Based Epistemic Variance of Values for Risk-Aware Policy Optimizationcited by 3
- 2024Efficient Model-Based Reinforcement Learning Through Optimistic Thompson Samplingcited by 3
- 2024Zero-Shot Offline Imitation Learning via Optimal Transportcited by 3
- 2021Model-Advantage and Value-Aware Models for Model-Based Reinforcement Learning: Bridging the Gap in Theory and Practicecited by 1
- 2022Model Based Residual Policy Learning with Applications to Antenna Controlcited by 1
- 2023Hybrid Systems Neural Control with Region-of-Attraction Plannercited by 1
- 2023Improving Deep Dynamics Models for Autonomous Vehicles with Multimodal Latent Mapping of Surfacescited by 1
- 2024Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learningcited by 1
- 2024Three Pathways to Neurosymbolic Reinforcement Learning with Interpretable Model and Policy Networkscited by 1
- 2025Situationally-Aware Dynamics Learningcited by 1
- 2025Learning Causal Structure Distributions for Robust Planningcited by 1
- 2025Double Horizon Model-Based Policy Optimizationcited by 1
- 2025Learning Contact Dynamics through Touching: Action-conditional Graph Neural Networks for Robotic Peg Insertioncited by 0
- 2026A Control Barrier Function-Constrained Model Predictive Control Framework for Safe Reinforcement Learningcited by 0
- 2026stable-worldmodel: A Platform for Reproducible World Modeling Research and Evaluation20 May 2026
- 2026A Control Barrier Function-Constrained Model Predictive Control Framework for Safe Reinforcement Learning7 Apr 2026
- 2025Double Horizon Model-Based Policy Optimization17 Dec 2025
- 2025Learning Contact Dynamics through Touching: Action-conditional Graph Neural Networks for Robotic Peg Insertion15 Sep 2025
- 2025Learning Causal Structure Distributions for Robust Planning8 Aug 2025
- 2025Continual Reinforcement Learning by Planning with Online World Models12 Jul 2025
- 2025Situationally-Aware Dynamics Learning26 May 2025
- 2024Zero-Shot Offline Imitation Learning via Optimal Transport11 Oct 2024
- 2024Efficient Model-Based Reinforcement Learning Through Optimistic Thompson Sampling7 Oct 2024
- 2024Trust the Model Where It Trusts Itself -- Model-Based Actor-Critic with Uncertainty-Aware Rollout Adaption29 May 2024
- 2024Hybrid Inverse Reinforcement Learning13 Feb 2024
- 2024Three Pathways to Neurosymbolic Reinforcement Learning with Interpretable Model and Policy Networks7 Feb 2024
30 more
- 2024Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learning5 Feb 2024
- 2023Model-Based Epistemic Variance of Values for Risk-Aware Policy Optimization7 Dec 2023
- 2023Pearl: A Production-ready Reinforcement Learning Agent6 Dec 2023
- 2023Data-efficient Deep Reinforcement Learning for Vehicle Trajectory Control30 Nov 2023
- 2023Goal-conditioned Offline Planning from Curious Exploration28 Nov 2023
- 2023Model-Based Reparameterization Policy Gradient Methods: Theory and Practical Algorithms30 Oct 2023
- 2023A Unified View on Solving Objective Mismatch in Model-Based Reinforcement Learning10 Oct 2023
- 2023Value-Distributional Model-Based Reinforcement Learning12 Aug 2023
- 2023How to Learn and Generalize From Three Minutes of Data: Physics-Constrained and Uncertainty-Aware Neural Stochastic Differential Equations10 Jun 2023
- 2023Seizing Serendipity: Exploiting the Value of Past Success in Off-Policy Actor-Critic5 Jun 2023
- 2023Improving Deep Dynamics Models for Autonomous Vehicles with Multimodal Latent Mapping of Surfaces21 Mar 2023
- 2023Neural Constraint Satisfaction: Hierarchical Abstraction for Combinatorial Generalization in Object Rearrangement20 Mar 2023
- 2023Hybrid Systems Neural Control with Region-of-Attraction Planner18 Mar 2023
- 2023Dynamic Update-to-Data Ratio: Minimizing World Model Overfitting17 Mar 2023
- 2023The Virtues of Laziness in Model-based RL: A Unified Objective and Algorithms1 Mar 2023
- 2023Model-Based Uncertainty in Value Functions24 Feb 2023
- 2023Certifiably Robust Reinforcement Learning through Model-Based Abstract Interpretation26 Jan 2023
- 2022Model Based Residual Policy Learning with Applications to Antenna Control16 Nov 2022
- 2022When to Update Your Model: Constrained Model-based Reinforcement Learning15 Oct 2022
- 2022Inferring Smooth Control: Monte Carlo Posterior Policy Iteration with Gaussian Processes7 Oct 2022
- 2022Exploration via Planning for Information about the Optimal Trajectory6 Oct 2022
- 2022Continuous Monte Carlo Graph Search4 Oct 2022
- 2022Training Efficient Controllers via Analytic Policy Gradient26 Sep 2022
- 2022Value Gradient weighted Model-Based Reinforcement Learning4 Apr 2022
- 2022Saute RL: Almost Surely Safe Reinforcement Learning Using State Augmentation14 Feb 2022
- 2022Reinforcement Learning in Presence of Discrete Markovian Context Evolution14 Feb 2022
- 2021An Experimental Design Perspective on Model-Based Reinforcement Learning9 Dec 2021
- 2021Risk Sensitive Model-Based Reinforcement Learning using Uncertainty Guided Planning9 Nov 2021
- 2021Revisiting Design Choices in Offline Model-Based Reinforcement Learning8 Oct 2021
- 2021Model-Advantage and Value-Aware Models for Model-Based Reinforcement Learning: Bridging the Gap in Theory and Practice26 Jun 2021
- 2024Trust the Model Where It Trusts Itself -- Model-Based Actor-Critic with Uncertainty-Aware Rollout Adaptioncited by 18
- 2024Efficient Model-Based Reinforcement Learning Through Optimistic Thompson Samplingcited by 3
- 2024Three Pathways to Neurosymbolic Reinforcement Learning with Interpretable Model and Policy Networkscited by 1
- 2025Double Horizon Model-Based Policy Optimizationcited by 1
- 2026A Control Barrier Function-Constrained Model Predictive Control Framework for Safe Reinforcement Learningcited by 0
- 2026A Control Barrier Function-Constrained Model Predictive Control Framework for Safe Reinforcement Learning7 Apr 2026
- 2025Double Horizon Model-Based Policy Optimization17 Dec 2025
- 2024Efficient Model-Based Reinforcement Learning Through Optimistic Thompson Sampling7 Oct 2024
- 2024Trust the Model Where It Trusts Itself -- Model-Based Actor-Critic with Uncertainty-Aware Rollout Adaption29 May 2024
- 2024Three Pathways to Neurosymbolic Reinforcement Learning with Interpretable Model and Policy Networks7 Feb 2024
- 2021Revisiting Design Choices in Offline Model-Based Reinforcement Learningcited by 50
- 2024Hybrid Inverse Reinforcement Learningcited by 32
- 2024Zero-Shot Offline Imitation Learning via Optimal Transportcited by 3
- 2024Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learningcited by 1
- 2024Zero-Shot Offline Imitation Learning via Optimal Transport11 Oct 2024
- 2024Hybrid Inverse Reinforcement Learning13 Feb 2024
- 2024Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learning5 Feb 2024
- 2021Revisiting Design Choices in Offline Model-Based Reinforcement Learning8 Oct 2021
- 2022Saute RL: Almost Surely Safe Reinforcement Learning Using State Augmentationcited by 69
- 2022Value Gradient weighted Model-Based Reinforcement Learningcited by 32
- 2022Training Efficient Controllers via Analytic Policy Gradientcited by 22
- 2023Seizing Serendipity: Exploiting the Value of Past Success in Off-Policy Actor-Criticcited by 20
- 2021An Experimental Design Perspective on Model-Based Reinforcement Learningcited by 16
- 2023Model-Based Uncertainty in Value Functionscited by 15
- 2022When to Update Your Model: Constrained Model-based Reinforcement Learningcited by 14
- 2023The Virtues of Laziness in Model-based RL: A Unified Objective and Algorithmscited by 12
- 2022Inferring Smooth Control: Monte Carlo Posterior Policy Iteration with Gaussian Processescited by 10
- 2023How to Learn and Generalize From Three Minutes of Data: Physics-Constrained and Uncertainty-Aware Neural Stochastic Differential Equationscited by 10
- 2022Reinforcement Learning in Presence of Discrete Markovian Context Evolutioncited by 9
- 2023A Unified View on Solving Objective Mismatch in Model-Based Reinforcement Learningcited by 9
12 most cited of 29 All 29 (JSON)
- 2025Learning Contact Dynamics through Touching: Action-conditional Graph Neural Networks for Robotic Peg Insertion15 Sep 2025
- 2025Learning Causal Structure Distributions for Robust Planning8 Aug 2025
- 2025Situationally-Aware Dynamics Learning26 May 2025
- 2023Model-Based Epistemic Variance of Values for Risk-Aware Policy Optimization7 Dec 2023
- 2023Pearl: A Production-ready Reinforcement Learning Agent6 Dec 2023
- 2023Data-efficient Deep Reinforcement Learning for Vehicle Trajectory Control30 Nov 2023
- 2023Goal-conditioned Offline Planning from Curious Exploration28 Nov 2023
- 2023Model-Based Reparameterization Policy Gradient Methods: Theory and Practical Algorithms30 Oct 2023
- 2023A Unified View on Solving Objective Mismatch in Model-Based Reinforcement Learning10 Oct 2023
- 2023Value-Distributional Model-Based Reinforcement Learning12 Aug 2023
- 2023How to Learn and Generalize From Three Minutes of Data: Physics-Constrained and Uncertainty-Aware Neural Stochastic Differential Equations10 Jun 2023
- 2023Seizing Serendipity: Exploiting the Value of Past Success in Off-Policy Actor-Critic5 Jun 2023
12 newest of 29 All 29 (JSON)