Model-Ensemble Trust-Region Policy Optimization
On the map: World models → Latent dynamics models and model-based RL
Abstract (arXiv)
Model-free reinforcement learning (RL) methods are succeeding in a growing number of tasks, aided by recent advances in deep learning. However, they tend to suffer from high sample complexity, which hinders their use in real-world domains. Alternatively, model-based reinforcement learning promises to reduce sample complexity, but tends to require careful tuning and to date have succeeded mainly in restrictive domains where simple models are sufficient for learning. In this paper, we analyze the behavior of vanilla model-based reinforcement learning methods when deep neural networks are used to learn both the model and the policy, and show that the learned policy tends to exploit regions where insufficient data is available for the model to be learned, causing instability in training. To overcome this issue, we propose to use an ensemble of models to maintain the model uncertainty and regularize the learning process. We further show that the use of likelihood ratio derivatives yields much more stable learning than backpropagation through time. Altogether, our approach Model-Ensemble Trust-Region Policy Optimization (ME-TRPO) significantly reduces the sample complexity compared to model-free deep RL methods on challenging continuous control benchmark tasks.
Key facts. Cited by 214 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 3 of its 45 references that are in the map. First posted on arXiv 28 Feb 2018.
Builds on
3 of 45 references.
Closest ancestors · 1
Same direction, newest first
- 2017Neural Network Dynamics for Model-Based Deep Reinforcement Learning with Model-Free Fine-Tuning8 Aug
Other references · 2
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2016Learning to Poke by Poking: Experiential Learning of Intuitive Physicscited by 152
Built on by
Citing papers per year: 2018: 3; 2019: 5; 2020: 7; 2021: 66; 2022: 44; 2023: 38; 2024: 28; 2025: 13; 2026: 10.
214 papers.
Sort:
By area:
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
- 2020Mastering Atari with Discrete World Modelscited by 990
- 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 753
- 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Modelscited by 748
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2021Offline Reinforcement Learning as One Big Sequence Modeling Problemcited by 636
- 2020MOPO: Model-based Offline Policy Optimizationcited by 630
- 2020MOReL : Model-Based Offline Reinforcement Learningcited by 582
- 2021How to Train Your Robot with Deep Reinforcement Learning; Lessons We've Learnedcited by 201
- 2022Masked World Models for Visual Controlcited by 169
- 2024Understanding World or Predicting Future? A Comprehensive Survey of World Modelscited by 161
38 more
- 2018Algorithmic Framework for Model-based Deep Reinforcement Learning with Theoretical Guaranteescited by 138
- 2022RAMBO-RL: Robust Adversarial Model-Based Offline Reinforcement Learningcited by 115
- 2021Learning to drive from a world on railscited by 114
- 2021Dropout Q-Functions for Doubly Efficient Reinforcement Learningcited by 108
- 2022Accelerated Policy Learning with Parallel Differentiable Simulationcited by 92
- 2025A Comprehensive Survey of Mixture-of-Experts: Algorithms, Theory, and Applicationscited by 91
- 2019Exploring Model-based Planning with Policy Networkscited by 90
- 2021Gradients are Not All You Needcited by 82
- 2020A Game Theoretic Framework for Model Based Reinforcement Learningcited by 72
- 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
- 2022Causal Dynamics Learning for Task-Independent State Abstractioncited by 64
- 2021NeoRL: A Near Real-World Benchmark for Offline Reinforcement Learningcited by 63
- 2022Deep Ensembles Work, But Are They Necessary?cited by 63
- 2021On the Importance of Hyperparameter Optimization for Model-based Reinforcement Learningcited by 59
- 2022VRL3: A Data-Driven Framework for Visual Deep Reinforcement Learningcited by 51
- 2021Revisiting Design Choices in Offline Model-Based Reinforcement Learningcited by 50
- 2022Making Linear MDPs Practical via Contrastive Representation Learningcited by 49
- 2021Vector Quantized Models for Planningcited by 47
- 2021MobILE: Model-Based Imitation Learning From Observation Alonecited by 44
- 2023A Survey on Causal Reinforcement Learningcited by 43
- 2020Overcoming Model Bias for Robust Offline Deep Reinforcement Learningcited by 42
- 2021Augmented World Models Facilitate Zero-Shot Dynamics Generalization From a Single Offline Environmentcited by 41
- 2021MBRL-Lib: A Modular Library for Model-based Reinforcement Learningcited by 41
- 2021Continuous-Time Model-Based Reinforcement Learningcited by 39
- 2021Adaptive-Control-Oriented Meta-Learning for Nonlinear Systemscited by 38
- 2022Spectral Decomposition Representation for Reinforcement Learningcited by 35
- 2021Deep reinforcement learning in medical imaging: A literature reviewcited by 34
- 2021Autoregressive Dynamics Models for Offline Policy Evaluation and Optimizationcited by 32
- 2021Neural Auction: End-to-End Learning of Auction Mechanisms for E-Commerce Advertisingcited 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
- 2024Non-myopic Generation of Language Models for Reasoning and Planningcited by 25
50 most cited of 214 All 214 (JSON)
- 2026Lucid Dreaming for World Models: Learning to Doubt Imagination and Decide by Trust29 Sep 2026
- 2026The Intervention Gap in Latent World Models30 Aug 2026
- 2026Q-Learning With World Models17 Aug 2026
- 2026Imperfect World Models are Exploitable15 May 2026
- 2026Quantifying the Utility of User Simulators for Building Collaborative LLM Assistants10 May 2026
- 2026Detecting is Easy, Adapting is Hard: Local Expert Growth for Visual Model-Based Reinforcement Learning under Distribution Shift30 Apr 2026
- 2026Advantage-Guided Diffusion for Model-Based Reinforcement Learning10 Apr 2026
- 2026WIMLE: Uncertainty-Aware World Models with IMLE for Sample-Efficient Continuous Control15 Feb 2026
- 2026Optimistic World Models: Efficient Exploration in Model-Based Deep Reinforcement Learning10 Feb 2026
- 2026Coupled Local and Global World Models for Efficient First Order RL5 Feb 2026
- 2025Spectral Representation-based Reinforcement Learning17 Dec 2025
- 2025Controllable Flow Matching for Online Reinforcement Learning10 Nov 2025
38 more
- 2025Cavity Duplexer Tuning with 1d Resnet-like Neural Networks17 Oct 2025
- 2025First Order Model-Based RL through Decoupled Backpropagation29 Aug 2025
- 2025TD-MPC-Opt: Distilling Model-Based Multi-Task Reinforcement Learning Agents2 Jul 2025
- 2025Meta-reinforcement learning with minimum attention22 May 2025
- 2025Sample-Efficient Reinforcement Learning of Koopman eNMPC24 Mar 2025
- 2025A Comprehensive Survey of Mixture-of-Experts: Algorithms, Theory, and Applications10 Mar 2025
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRitic27 Feb 2025
- 2025Scalable Model Merging with Progressive Layer-wise Distillation18 Feb 2025
- 2025Trajectory World Models for Heterogeneous Environments3 Feb 2025
- 2025Digital Twin Calibration with Model-Based Reinforcement Learning4 Jan 2025
- 2025DFF: Decision-Focused Fine-tuning for Smarter Predict-then-Optimize with Limited Data3 Jan 2025
- 2024SimuDICE: Offline Policy Optimization Through World Model Updates and DICE Estimation9 Dec 2024
- 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
- 2024Coverage Analysis for Digital Cousin Selection -- Improving Multi-Environment Q-Learning13 Nov 2024
- 2024Acceleration for Deep Reinforcement Learning using Parallel and Distributed Computing: A Survey8 Nov 2024
- 2024Learning World Models for Unconstrained Goal Navigation3 Nov 2024
- 2024Non-myopic Generation of Language Models for Reasoning and Planning22 Oct 2024
- 2024Dual Action Policy for Robust Sim-to-Real Reinforcement Learning16 Oct 2024
- 2024When to Trust Your Data: Enhancing Dyna-Style Model-Based Reinforcement Learning With Data Filter16 Oct 2024
- 2024On Reward Transferability in Adversarial Inverse Reinforcement Learning: Insights from Random Matrix Theory10 Oct 2024
- 2024COSBO: Conservative Offline Simulation-Based Policy Optimization22 Sep 2024
- 2024Offline Model-Based Reinforcement Learning with Anti-Exploration20 Aug 2024
- 2024Mixture of Experts in a Mixture of RL settings26 Jun 2024
- 2024Diffusion Spectral Representation for Reinforcement Learning23 Jun 2024
- 2024Learning to Play Atari in a World of Tokens3 Jun 2024
- 2024Trust the Model Where It Trusts Itself -- Model-Based Actor-Critic with Uncertainty-Aware Rollout Adaption29 May 2024
- 2024Efficient Multi-agent Reinforcement Learning by Planning20 May 2024
- 2024The Curse of Diversity in Ensemble-Based Exploration7 May 2024
- 2024Learning Control Barrier Functions and their application in Reinforcement Learning: A Survey22 Apr 2024
- 2024Grid-Mapping Pseudo-Count Constraint for Offline Reinforcement Learning3 Apr 2024
- 2024Robust Model Based Reinforcement Learning Using L1 Adaptive Control21 Mar 2024
- 2024An Efficient Model-Based Approach on Learning Agile Motor Skills without Reinforcement4 Mar 2024
- 2024Model-based deep reinforcement learning for accelerated learning from flow simulations26 Feb 2024
- 2024Leveraging Digital Cousins for Ensemble Q-Learning in Large-Scale Wireless Networks12 Feb 2024
- 2024Multi-Timescale Ensemble Q-learning for Markov Decision Process Policy Optimization8 Feb 2024
- 2024Building Minimal and Reusable Causal State Abstractions for Reinforcement Learning23 Jan 2024
- 2024Episodic Reinforcement Learning with Expanded State-reward Space19 Jan 2024
50 newest of 214 All 214 (JSON)
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
- 2020Mastering Atari with Discrete World Modelscited by 990
- 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 753
- 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Modelscited by 748
- 2021Offline Reinforcement Learning as One Big Sequence Modeling Problemcited by 636
- 2022Masked World Models for Visual Controlcited by 169
- 2024Understanding World or Predicting Future? A Comprehensive Survey of World Modelscited by 161
- 2018Algorithmic Framework for Model-based Deep Reinforcement Learning with Theoretical Guaranteescited by 138
- 2019Exploring Model-based Planning with Policy Networkscited by 90
- 2020A Game Theoretic Framework for Model Based Reinforcement Learningcited by 72
- 2020World Model as a Graph: Learning Latent Landmarks for Planningcited by 67
12 most cited of 37 All 37 (JSON)
- 2026Lucid Dreaming for World Models: Learning to Doubt Imagination and Decide by Trust29 Sep 2026
- 2026The Intervention Gap in Latent World Models30 Aug 2026
- 2026Q-Learning With World Models17 Aug 2026
- 2026Imperfect World Models are Exploitable15 May 2026
- 2026Advantage-Guided Diffusion for Model-Based Reinforcement Learning10 Apr 2026
- 2026WIMLE: Uncertainty-Aware World Models with IMLE for Sample-Efficient Continuous Control15 Feb 2026
- 2026Optimistic World Models: Efficient Exploration in Model-Based Deep Reinforcement Learning10 Feb 2026
- 2026Coupled Local and Global World Models for Efficient First Order RL5 Feb 2026
- 2025TD-MPC-Opt: Distilling Model-Based Multi-Task Reinforcement Learning Agents2 Jul 2025
- 2025Trajectory World Models for Heterogeneous Environments3 Feb 2025
- 2024SimuDICE: Offline Policy Optimization Through World Model Updates and DICE Estimation9 Dec 2024
- 2024Bounded Exploration with World Model Uncertainty in Soft Actor-Critic Reinforcement Learning Algorithm9 Dec 2024
12 newest of 37 All 37 (JSON)
- 2020MOPO: Model-based Offline Policy Optimizationcited by 630
- 2020MOReL : Model-Based Offline Reinforcement Learningcited by 582
- 2022RAMBO-RL: Robust Adversarial Model-Based Offline Reinforcement Learningcited by 115
- 2021NeoRL: A Near Real-World Benchmark for Offline Reinforcement Learningcited by 63
- 2021Revisiting Design Choices in Offline Model-Based Reinforcement Learningcited by 50
- 2021MobILE: Model-Based Imitation Learning From Observation Alonecited by 44
- 2020Overcoming Model Bias for Robust Offline Deep Reinforcement Learningcited by 42
- 2021Behavioral Priors and Dynamics Models: Improving Performance and Domain Transfer in Offline RLcited by 24
- 2022One Risk to Rule Them All: A Risk-Sensitive Perspective on Model-Based Offline Reinforcement Learningcited by 21
- 2022Q-Ensemble for Offline RL: Don't Scale the Ensemble, Scale the Batch Sizecited by 20
- 2023Fighting Uncertainty with Gradients: Offline Reinforcement Learning via Diffusion Score Matchingcited by 17
- 2020Expert-Supervised Reinforcement Learning for Offline Policy Learning and Evaluationcited by 14
12 most cited of 25 All 25 (JSON)
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRitic27 Feb 2025
- 2024On Reward Transferability in Adversarial Inverse Reinforcement Learning: Insights from Random Matrix Theory10 Oct 2024
- 2024COSBO: Conservative Offline Simulation-Based Policy Optimization22 Sep 2024
- 2024Offline Model-Based Reinforcement Learning with Anti-Exploration20 Aug 2024
- 2024Grid-Mapping Pseudo-Count Constraint for Offline Reinforcement Learning3 Apr 2024
- 2024MOTO: Offline Pre-training to Online Fine-tuning for Model-based Robot Learning6 Jan 2024
- 2023Learning non-Markovian Decision-Making from State-only Sequences27 Jun 2023
- 2023Fighting Uncertainty with Gradients: Offline Reinforcement Learning via Diffusion Score Matching24 Jun 2023
- 2022One Risk to Rule Them All: A Risk-Sensitive Perspective on Model-Based Offline Reinforcement Learning30 Nov 2022
- 2022Domain Generalization for Robust Model-Based Offline Reinforcement Learning27 Nov 2022
- 2022Q-Ensemble for Offline RL: Don't Scale the Ensemble, Scale the Batch Size20 Nov 2022
- 2022A Unified Framework for Alternating Offline Model Training and Policy Learning12 Oct 2022
12 newest of 25 All 25 (JSON)
- 2023Refining Diffusion Planner for Reliable Behavior Synthesis by Automatic Detection of Infeasible Planscited by 27
- 2024Diffusion Spectral Representation for Reinforcement Learningcited by 15
- 2023Censored Sampling of Diffusion Models Using 3 Minutes of Human Feedbackcited by 12
- 2025Controllable Flow Matching for Online Reinforcement Learningcited by 0
- 2025Controllable Flow Matching for Online Reinforcement Learning10 Nov 2025
- 2024Diffusion Spectral Representation for Reinforcement Learning23 Jun 2024
- 2023Refining Diffusion Planner for Reliable Behavior Synthesis by Automatic Detection of Infeasible Plans30 Oct 2023
- 2023Censored Sampling of Diffusion Models Using 3 Minutes of Human Feedback6 Jul 2023
- 2021Learning to drive from a world on railscited by 114
- 2021Learning to drive from a world on rails3 May 2021
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2021How to Train Your Robot with Deep Reinforcement Learning; Lessons We've Learnedcited by 201
- 2021Dropout Q-Functions for Doubly Efficient Reinforcement Learningcited by 108
- 2022Accelerated Policy Learning with Parallel Differentiable Simulationcited by 92
- 2021Gradients are Not All You Needcited by 82
- 2022Safe Reinforcement Learning by Imagining the Near Futurecited by 67
- 2022Causal Dynamics Learning for Task-Independent State Abstractioncited by 64
- 2022Deep Ensembles Work, But Are They Necessary?cited by 63
- 2021On the Importance of Hyperparameter Optimization for Model-based Reinforcement Learningcited by 59
- 2022VRL3: A Data-Driven Framework for Visual Deep Reinforcement Learningcited by 51
- 2022Making Linear MDPs Practical via Contrastive Representation Learningcited by 49
- 2021Vector Quantized Models for Planningcited by 47
12 most cited of 144 All 144 (JSON)
- 2026Quantifying the Utility of User Simulators for Building Collaborative LLM Assistants10 May 2026
- 2026Detecting is Easy, Adapting is Hard: Local Expert Growth for Visual Model-Based Reinforcement Learning under Distribution Shift30 Apr 2026
- 2025Spectral Representation-based Reinforcement Learning17 Dec 2025
- 2025Cavity Duplexer Tuning with 1d Resnet-like Neural Networks17 Oct 2025
- 2025First Order Model-Based RL through Decoupled Backpropagation29 Aug 2025
- 2025Meta-reinforcement learning with minimum attention22 May 2025
- 2025Sample-Efficient Reinforcement Learning of Koopman eNMPC24 Mar 2025
- 2025Scalable Model Merging with Progressive Layer-wise Distillation18 Feb 2025
- 2025Digital Twin Calibration with Model-Based Reinforcement Learning4 Jan 2025
- 2025DFF: Decision-Focused Fine-tuning for Smarter Predict-then-Optimize with Limited Data3 Jan 2025
- 2024Coverage Analysis for Digital Cousin Selection -- Improving Multi-Environment Q-Learning13 Nov 2024
- 2024Acceleration for Deep Reinforcement Learning using Parallel and Distributed Computing: A Survey8 Nov 2024
12 newest of 144 All 144 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2024Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learning
- 2025Accelerating Model-Based Reinforcement Learning with State-Space World Models
- 2020Adversarial Imitation Learning via Random Search
- 2025Model-Based Offline Reinforcement Learning with Adversarial Data Augmentation
- 2024PWM: Policy Learning with Multi-Task World Models
- 2026Trust-Region Diffusion Policies for Massively Parallel On-Policy RL
- 2025Policy-Driven World Model Adaptation for Robust Offline Model-based Reinforcement Learning
- 2024Robust Model-Based Reinforcement Learning with an Adversarial Auxiliary Model