Algorithmic Framework for Model-based Deep Reinforcement Learning with Theoretical Guarantees
On the map: World models → Latent dynamics models and model-based RL
Abstract (arXiv)
Model-based reinforcement learning (RL) is considered to be a promising approach to reduce the sample complexity that hinders model-free RL. However, the theoretical understanding of such methods has been rather limited. This paper introduces a novel algorithmic framework for designing and analyzing model-based RL algorithms with theoretical guarantees. We design a meta-algorithm with a theoretical guarantee of monotone improvement to a local maximum of the expected reward. The meta-algorithm iteratively builds a lower bound of the expected reward based on the estimated dynamical model and sample trajectories, and then maximizes the lower bound jointly over the policy and the model. The framework extends the optimism-in-face-of-uncertainty principle to non-linear dynamical models in a way that requires textitno explicit uncertainty quantification. Instantiating our framework with simplification gives a variant of model-based RL algorithms Stochastic Lower Bounds Optimization (SLBO). Experiments demonstrate that SLBO achieves state-of-the-art performance when only one million or fewer samples are permitted on a range of continuous control benchmark tasks.
Key facts. Cited by 138 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 6 of its 75 references that are in the map. First posted on arXiv 10 Jul 2018.
Builds on
6 of 75 references.
Closest ancestors · 5
Same direction, newest first
- 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Models30 May
- 2018World Models27 Mar
- 2018Model-Ensemble Trust-Region Policy Optimization28 Feb
- 2017Neural Network Dynamics for Model-Based Deep Reinforcement Learning with Model-Free Fine-Tuning8 Aug
- 2017Value Prediction Network11 Jul
Other references · 1
- 2017Proximal Policy Optimization Algorithmscited by 14,441
Built on by
Citing papers per year: 2019: 3; 2020: 5; 2021: 35; 2022: 25; 2023: 25; 2024: 27; 2025: 10; 2026: 8.
138 papers.
Sort:
By area:
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,873
- 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 753
- 2020MOPO: Model-based Offline Policy Optimizationcited by 630
- 2020MOReL : Model-Based Offline Reinforcement Learningcited by 582
- 2024Understanding World or Predicting Future? A Comprehensive Survey of World Modelscited by 161
- 2019Exploring Model-based Planning with Policy Networkscited by 90
- 2020A Game Theoretic Framework for Model Based Reinforcement Learningcited by 72
- 2022RoboCraft: Learning to See, Simulate, and Shape Elasto-Plastic Objects with Graph Networkscited by 69
- 2020World Model as a Graph: Learning Latent Landmarks for Planningcited by 67
- 2022Safe Reinforcement Learning by Imagining the Near Futurecited by 67
- 2022A Survey on Model-based Reinforcement Learningcited by 65
- 2021MobILE: Model-Based Imitation Learning From Observation Alonecited by 44
38 more
- 2023Behavior Proximal Policy Optimizationcited by 43
- 2021On the Sample Complexity of Stability Constrained Imitation Learningcited by 38
- 2021Learning Barrier Certificates: Towards Safe Reinforcement Learning with Zero Training-time Violationscited by 36
- 2021Mismatched No More: Joint Model-Policy Optimization for Model-Based RLcited by 35
- 2023Policy Optimization in RLHF: The Impact of Out-of-preference Datacited by 35
- 2021Provable Model-based Nonlinear Bandit and Reinforcement Learning: Shelve Optimism, Embrace Virtual Curvaturecited by 32
- 2022Value Gradient weighted Model-Based Reinforcement Learningcited by 31
- 2023Cross-Domain Policy Adaptation via Value-Guided Data Filteringcited by 30
- 2023Multi-agent Reinforcement Learning: A Comprehensive Surveycited by 30
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objectivecited by 28
- 2022Model-based Safe Deep Reinforcement Learning via a Constrained Proximal Policy Optimization Algorithmcited by 27
- 2023Refining Diffusion Planner for Reliable Behavior Synthesis by Automatic Detection of Infeasible Planscited by 27
- 2021Model Predictive Actor-Critic: Accelerating Robot Skill Acquisition with Deep Reinforcement Learningcited 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
- 2023Learning Energy Decompositions for Partial Inference of GFlowNetscited by 24
- 2024Cross-Domain Policy Adaptation by Capturing Representation Mismatchcited by 24
- 2021Sample-Efficient Reinforcement Learning via Conservative Model-Based Actor-Criticcited by 23
- 2021Physics-informed Dyna-Style Model-Based Deep Reinforcement Learning for Dynamic Controlcited by 22
- 2024Theoretical Analysis of Meta Reinforcement Learning: Generalization Bounds and Convergence Guaranteescited by 22
- 2021Model-Based Opponent Modelingcited by 21
- 2023Deep Learning in Deterministic Computational Mechanicscited by 21
- 2023Efficient Exploration in Continuous-time Model-based Reinforcement Learningcited by 21
- 2023Seizing Serendipity: Exploiting the Value of Past Success in Off-Policy Actor-Criticcited by 20
- 2024Contrastive Representation for Data Filtering in Cross-Domain Offline Reinforcement Learningcited by 18
- 2024RoboPack: Learning Tactile-Informed Dynamics Models for Dense Packingcited by 18
- 2022Live in the Moment: Learning Dynamics Model Adapted to Evolving Policycited by 17
- 2024Trust the Model Where It Trusts Itself -- Model-Based Actor-Critic with Uncertainty-Aware Rollout Adaptioncited by 17
- 2024World Model-based Perception for Visual Legged Locomotioncited by 17
- 2021Minimax Model Learningcited by 16
- 2022Safe Model-Based Reinforcement Learning with an Uncertainty-Aware Reachability Certificatecited by 16
- 2023Plan To Predict: Learning an Uncertainty-Foreseeing Model for Model-Based Reinforcement Learningcited by 16
- 2024Diffusion Spectral Representation for Reinforcement Learningcited by 15
- 2021PC-MLP: Model-based Reinforcement Learning with Policy Cover Guided Explorationcited by 14
- 2022Efficient Model-based Multi-agent Reinforcement Learning via Optimistic Equilibrium Computationcited by 14
- 2022A Unified Framework for Alternating Offline Model Training and Policy Learningcited by 14
- 2022When to Update Your Model: Constrained Model-based Reinforcement Learningcited by 14
- 2022Latent Variable Representation for Reinforcement Learningcited by 14
50 most cited of 138 All 138 (JSON)
- 2026Active Inference as a Convex Markov Decision Process22 Jul 2026
- 2026DADiff: Diffusion-Driven Cross-Domain Policy Adaptation for Reinforcement Learning17 Jul 2026
- 2026All Models are Wrong, Knowing Where is Useful: On Model Uncertainty in Reinforcement Learning31 May 2026
- 2026Mind Dreamer: Untethering Imagination via Active Causal Intervention on Latent Manifolds15 May 2026
- 2026Conservative Continuous-Time Treatment Optimization17 Mar 2026
- 2026ADV-0: Closed-Loop Min-Max Adversarial Training for Long-Tail Robustness in Autonomous Driving16 Mar 2026
- 2026Localized Dynamics-Aware Domain Adaption for Off-Dynamics Offline Reinforcement Learning24 Feb 2026
- 2026Cross-Domain Offline Policy Adaptation via Selective Transition Correction5 Feb 2026
- 2025Spectral Representation-based Reinforcement Learning17 Dec 2025
- 2025Graph Neural Networks, Deep Reinforcement Learning and Probabilistic Topic Modeling for Strategic Multiagent Settings13 Nov 2025
- 2025DmC: Nearest Neighbor Guidance Diffusion Model for Offline Cross-domain Reinforcement Learning28 Jul 2025
- 2025GenAI-based Multi-Agent Reinforcement Learning towards Distributed Agent Intelligence: A Generative-RL Agent Perspective13 Jul 2025
38 more
- 2025Permutation Equivariant Model-based Offline Reinforcement Learning for Auto-bidding22 Jun 2025
- 2025Local Manifold Approximation and Projection for Manifold-Aware Diffusion Planning1 Jun 2025
- 2025Hybrid Cross-domain Robust Reinforcement Learning29 May 2025
- 2025Offline Guarded Safe Reinforcement Learning for Medical Treatment Optimization Strategies22 May 2025
- 2025Look Before Leap: Look-Ahead Planning with Uncertainty in Reinforcement Learning26 Mar 2025
- 2025Data-driven inventory management for new products: An adjusted Dyna-Q approach with transfer learning14 Jan 2025
- 2024Bounded Exploration with World Model Uncertainty in Soft Actor-Critic Reinforcement Learning Algorithm9 Dec 2024
- 2024Understanding World or Predicting Future? A Comprehensive Survey of World Models21 Nov 2024
- 2024Learning World Models for Unconstrained Goal Navigation3 Nov 2024
- 2024Exploring the Edges of Latent State Clusters for Goal-Conditioned Reinforcement Learning3 Nov 2024
- 2024When to Trust Your Data: Enhancing Dyna-Style Model-Based Reinforcement Learning With Data Filter16 Oct 2024
- 2024Bisimulation metric for Model Predictive Control6 Oct 2024
- 2024Grounded Answers for Multi-agent Decision-making Problem through Generative World Model3 Oct 2024
- 2024World Model-based Perception for Visual Legged Locomotion25 Sep 2024
- 2024Mitigating Distribution Shift in Model-based Offline RL via Shifts-aware Reward Learning23 Aug 2024
- 2024BECAUSE: Bilinear Causal Representation for Generalizable Offline Model-based Reinforcement Learning15 Jul 2024
- 2024Short-Long Policy Evaluation with Novel Actions4 Jul 2024
- 2024RoboPack: Learning Tactile-Informed Dynamics Models for Dense Packing1 Jul 2024
- 2024Diffusion Spectral Representation for Reinforcement Learning23 Jun 2024
- 2024Defending Against Sophisticated Poisoning Attacks with RL-based Aggregation in Federated Learning20 Jun 2024
- 2024SeMOPO: Learning High-quality Model and Policy from Low-quality Offline Visual Datasets13 Jun 2024
- 2024Trust the Model Where It Trusts Itself -- Model-Based Actor-Critic with Uncertainty-Aware Rollout Adaption29 May 2024
- 2024BWArea Model: Learning World Model, Inverse Dynamics, and Policy for Controllable Language Generation27 May 2024
- 2024Any-step Dynamics Model Improves Future Predictions for Online and Offline Reinforcement Learning27 May 2024
- 2024Cross-Domain Policy Adaptation by Capturing Representation Mismatch24 May 2024
- 2024Theoretical Analysis of Meta Reinforcement Learning: Generalization Bounds and Convergence Guarantees22 May 2024
- 2024Scrutinize What We Ignore: Reining In Task Representation Shift Of Context-Based Offline Meta Reinforcement Learning20 May 2024
- 2024Contrastive Representation for Data Filtering in Cross-Domain Offline Reinforcement Learning10 May 2024
- 2024An Efficient Model-Based Approach on Learning Agile Motor Skills without Reinforcement4 Mar 2024
- 2024Policy Learning for Off-Dynamics RL with Deficient Support16 Feb 2024
- 2024Episodic Reinforcement Learning with Expanded State-reward Space19 Jan 2024
- 2024Optimistic Model Rollouts for Pessimistic Offline Policy Optimization11 Jan 2024
- 2024MOTO: Offline Pre-training to Online Fine-tuning for Model-based Robot Learning6 Jan 2024
- 2023Policy Optimization in RLHF: The Impact of Out-of-preference Data17 Dec 2023
- 2023Multi-agent Reinforcement Learning: A Comprehensive Survey15 Dec 2023
- 2023Data-efficient Deep Reinforcement Learning for Vehicle Trajectory Control30 Nov 2023
- 2023Efficient Exploration in Continuous-time Model-based Reinforcement Learning30 Oct 2023
- 2023Refining Diffusion Planner for Reliable Behavior Synthesis by Automatic Detection of Infeasible Plans30 Oct 2023
50 newest of 138 All 138 (JSON)
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,873
- 2020MOPO: Model-based Offline Policy Optimizationcited by 630
- 2020MOReL : Model-Based Offline Reinforcement Learningcited by 582
- 2021MobILE: Model-Based Imitation Learning From Observation Alonecited by 44
- 2023Behavior Proximal Policy Optimizationcited by 43
- 2021On the Sample Complexity of Stability Constrained Imitation Learningcited by 38
- 2024Contrastive Representation for Data Filtering in Cross-Domain Offline Reinforcement Learningcited by 18
- 2022A Unified Framework for Alternating Offline Model Training and Policy Learningcited by 14
- 2024MOTO: Offline Pre-training to Online Fine-tuning for Model-based Robot Learningcited by 13
- 2021PerSim: Data-Efficient Offline Reinforcement Learning with Heterogeneous Agents via Personalized Simulatorscited by 10
- 2024Any-step Dynamics Model Improves Future Predictions for Online and Offline Reinforcement Learningcited by 10
- 2022MOORe: Model-based Offline-to-Online Reinforcement Learningcited by 9
12 most cited of 22 All 22 (JSON)
- 2026Localized Dynamics-Aware Domain Adaption for Off-Dynamics Offline Reinforcement Learning24 Feb 2026
- 2025Permutation Equivariant Model-based Offline Reinforcement Learning for Auto-bidding22 Jun 2025
- 2025Offline Guarded Safe Reinforcement Learning for Medical Treatment Optimization Strategies22 May 2025
- 2024Mitigating Distribution Shift in Model-based Offline RL via Shifts-aware Reward Learning23 Aug 2024
- 2024BECAUSE: Bilinear Causal Representation for Generalizable Offline Model-based Reinforcement Learning15 Jul 2024
- 2024SeMOPO: Learning High-quality Model and Policy from Low-quality Offline Visual Datasets13 Jun 2024
- 2024Any-step Dynamics Model Improves Future Predictions for Online and Offline Reinforcement Learning27 May 2024
- 2024Contrastive Representation for Data Filtering in Cross-Domain Offline Reinforcement Learning10 May 2024
- 2024Optimistic Model Rollouts for Pessimistic Offline Policy Optimization11 Jan 2024
- 2024MOTO: Offline Pre-training to Online Fine-tuning for Model-based Robot Learning6 Jan 2024
- 2023Model-based Offline Policy Optimization with Adversarial Network5 Sep 2023
- 2023Learning non-Markovian Decision-Making from State-only Sequences27 Jun 2023
12 newest of 22 All 22 (JSON)
- 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 753
- 2024Understanding World or Predicting Future? A Comprehensive Survey of World Modelscited by 161
- 2019Exploring Model-based Planning with Policy Networkscited by 90
- 2020A Game Theoretic Framework for Model Based Reinforcement Learningcited by 72
- 2022RoboCraft: Learning to See, Simulate, and Shape Elasto-Plastic Objects with Graph Networkscited by 69
- 2020World Model as a Graph: Learning Latent Landmarks for Planningcited by 67
- 2022A Survey on Model-based Reinforcement Learningcited by 65
- 2024World Model-based Perception for Visual Legged Locomotioncited by 17
- 2022Value Memory Graph: A Graph-Structured World Model for Offline Reinforcement Learningcited by 11
- 2024Grounded Answers for Multi-agent Decision-making Problem through Generative World Modelcited by 6
- 2024Learning World Models for Unconstrained Goal Navigationcited by 6
- 2024BWArea Model: Learning World Model, Inverse Dynamics, and Policy for Controllable Language Generationcited by 3
12 most cited of 16 All 16 (JSON)
- 2024Bounded Exploration with World Model Uncertainty in Soft Actor-Critic Reinforcement Learning Algorithm9 Dec 2024
- 2024Understanding World or Predicting Future? A Comprehensive Survey of World Models21 Nov 2024
- 2024Learning World Models for Unconstrained Goal Navigation3 Nov 2024
- 2024Grounded Answers for Multi-agent Decision-making Problem through Generative World Model3 Oct 2024
- 2024World Model-based Perception for Visual Legged Locomotion25 Sep 2024
- 2024BWArea Model: Learning World Model, Inverse Dynamics, and Policy for Controllable Language Generation27 May 2024
- 2024An Efficient Model-Based Approach on Learning Agile Motor Skills without Reinforcement4 Mar 2024
- 2022A Survey on Model-based Reinforcement Learning19 Jun 2022
- 2022Value Memory Graph: A Graph-Structured World Model for Offline Reinforcement Learning9 Jun 2022
- 2022RoboCraft: Learning to See, Simulate, and Shape Elasto-Plastic Objects with Graph Networks5 May 2022
- 2021ED2: Environment Dynamics Decomposition World Models for Continuous Control6 Dec 2021
- 2020World Model as a Graph: Learning Latent Landmarks for Planning25 Nov 2020
12 newest of 16 All 16 (JSON)
- 2023Refining Diffusion Planner for Reliable Behavior Synthesis by Automatic Detection of Infeasible Planscited by 27
- 2024Diffusion Spectral Representation for Reinforcement Learningcited by 15
- 2025Local Manifold Approximation and Projection for Manifold-Aware Diffusion Planningcited by 8
- 2025DmC: Nearest Neighbor Guidance Diffusion Model for Offline Cross-domain Reinforcement Learningcited by 7
- 2025DmC: Nearest Neighbor Guidance Diffusion Model for Offline Cross-domain Reinforcement Learning28 Jul 2025
- 2025Local Manifold Approximation and Projection for Manifold-Aware Diffusion Planning1 Jun 2025
- 2024Diffusion Spectral Representation for Reinforcement Learning23 Jun 2024
- 2023Refining Diffusion Planner for Reliable Behavior Synthesis by Automatic Detection of Infeasible Plans30 Oct 2023
- 2022Safe Reinforcement Learning by Imagining the Near Futurecited by 67
- 2021Learning Barrier Certificates: Towards Safe Reinforcement Learning with Zero Training-time Violationscited by 36
- 2021Mismatched No More: Joint Model-Policy Optimization for Model-Based RLcited by 35
- 2021Provable Model-based Nonlinear Bandit and Reinforcement Learning: Shelve Optimism, Embrace Virtual Curvaturecited by 32
- 2022Value Gradient weighted Model-Based Reinforcement Learningcited by 31
- 2023Cross-Domain Policy Adaptation via Value-Guided Data Filteringcited by 30
- 2023Multi-agent Reinforcement Learning: A Comprehensive Surveycited by 30
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objectivecited by 28
- 2022Model-based Safe Deep Reinforcement Learning via a Constrained Proximal Policy Optimization Algorithmcited by 27
- 2021Model Predictive Actor-Critic: Accelerating Robot Skill Acquisition with Deep Reinforcement Learningcited 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
12 most cited of 93 All 93 (JSON)
- 2026Active Inference as a Convex Markov Decision Process22 Jul 2026
- 2026DADiff: Diffusion-Driven Cross-Domain Policy Adaptation for Reinforcement Learning17 Jul 2026
- 2026All Models are Wrong, Knowing Where is Useful: On Model Uncertainty in Reinforcement Learning31 May 2026
- 2026Mind Dreamer: Untethering Imagination via Active Causal Intervention on Latent Manifolds15 May 2026
- 2026Conservative Continuous-Time Treatment Optimization17 Mar 2026
- 2026Cross-Domain Offline Policy Adaptation via Selective Transition Correction5 Feb 2026
- 2025Spectral Representation-based Reinforcement Learning17 Dec 2025
- 2025Graph Neural Networks, Deep Reinforcement Learning and Probabilistic Topic Modeling for Strategic Multiagent Settings13 Nov 2025
- 2025GenAI-based Multi-Agent Reinforcement Learning towards Distributed Agent Intelligence: A Generative-RL Agent Perspective13 Jul 2025
- 2025Hybrid Cross-domain Robust Reinforcement Learning29 May 2025
- 2025Look Before Leap: Look-Ahead Planning with Uncertainty in Reinforcement Learning26 Mar 2025
- 2025Data-driven inventory management for new products: An adjusted Dyna-Q approach with transfer learning14 Jan 2025
12 newest of 93 All 93 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2024Model-based RL as a Minimalist Approach to Horizon-Free and Second-Order Bounds
- 2024MAMBA: an Effective World Model Approach for Meta-Reinforcement Learning
- 2022On Practical Reinforcement Learning: Provable Robustness, Scalability, and Statistical Efficiency
- 2026Maximum Likelihood Reinforcement Learning
- 2025Statistical and Algorithmic Foundations of Reinforcement Learning
- 2022Oracle Inequalities for Model Selection in Offline Reinforcement Learning
- 2024Deterministic Uncertainty Propagation for Improved Model-Based Offline Reinforcement Learning
- 2026A Pontryagin Method of Model-based Reinforcement Learning via Hamiltonian Actor-Critic