Exploring Model-based Planning with Policy Networks
On the map: World models → Latent dynamics models and model-based RL
Abstract (arXiv)
Model-based reinforcement learning (MBRL) with model-predictive control or online planning has shown great potential for locomotion control tasks in terms of both sample efficiency and asymptotic performance. Despite their initial successes, the existing planning methods search from candidate sequences randomly generated in the action space, which is inefficient in complex high-dimensional environments. In this paper, we propose a novel MBRL algorithm, model-based policy planning (POPLIN), that combines policy networks with online planning. More specifically, we formulate action planning at each time-step as an optimization problem using neural networks. We experiment with both optimization w.r.t. the action sequences initialized from the policy network, and also online optimization directly w.r.t. the parameters of the policy network. We show that POPLIN obtains state-of-the-art performance in the MuJoCo benchmarking environments, being about 3x more sample efficient than the state-of-the-art algorithms, such as PETS, TD3 and SAC. To explain the effectiveness of our algorithm, we show that the optimization surface in parameter space is smoother than in action space. Further more, we found the distilled policy network can be effectively applied without the expansive model predictive control during test time for some environments such as Cheetah. Code is released in https://github.com/WilsonWangTHU/POPLIN.
Key facts. Cited by 90 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 9 of its 44 references that are in the map. First posted on arXiv 20 Jun 2019.
Builds on
9 of 44 references.
Closest ancestors · 7
Same direction, newest first
- 2019When to Trust Your Model: Model-Based Policy Optimization19 Jun
- 2018Recurrent World Models Facilitate Policy Evolution4 Sep
- 2018Algorithmic Framework for Model-based Deep Reinforcement Learning with Theoretical Guarantees10 Jul
- 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
Other references · 2
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
Built on by
Citing papers per year: 2019: 1; 2020: 4; 2021: 23; 2022: 23; 2023: 11; 2024: 12; 2025: 10; 2026: 6.
90 papers.
Sort:
By area:
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2020Mastering Atari with Discrete World Modelscited by 990
- 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
- 2022Temporal Difference Learning for Model Predictive Controlcited by 375
- 2024Understanding World or Predicting Future? A Comprehensive Survey of World Modelscited by 161
- 2021Actionable Models: Unsupervised Offline Reinforcement Learning of Robotic Skillscited by 125
- 2022Tutorial on amortized optimizationcited by 83
- 2023DREAMWALKER: Mental Planning for Continuous Vision-Language Navigationcited by 82
- 2021Residual Pathway Priors for Soft Equivariance Constraintscited by 74
- 2024Simple Hierarchical Planning with Diffusioncited by 71
38 more
- 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
- 2022Efficient Planning in a Compact Latent Action Spacecited by 57
- 2022Skill-based Model-based Reinforcement Learningcited by 50
- 2021MBRL-Lib: A Modular Library for Model-based Reinforcement Learningcited by 41
- 2022Deconstructing the Inductive Biases of Hamiltonian Neural Networkscited by 31
- 2021Model-Based Offline Planning with Trajectory Pruningcited by 30
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objectivecited by 28
- 2021Model Predictive Actor-Critic: Accelerating Robot Skill Acquisition with Deep Reinforcement Learningcited by 25
- 2021Sample-Efficient Reinforcement Learning via Conservative Model-Based Actor-Criticcited by 23
- 2024Bridging Evolutionary Algorithms and Reinforcement Learning: A Comprehensive Survey on Hybrid Algorithmscited by 23
- 2022Double Check Your State Before Trusting It: Confidence-Aware Bidirectional Offline Model-Based Imaginationcited by 21
- 2021Where to go next: Learning a Subgoal Recommendation Policy for Navigation Among Pedestrianscited by 20
- 2022Combining Evolution and Deep Reinforcement Learning for Policy Search: a Surveycited by 20
- 2025Learning Transformer-based World Models with Contrastive Predictive Codingcited by 20
- 2023Stabilizing Contrastive RL: Techniques for Robotic Goal Reaching from Offline Datacited by 19
- 2025Synthesis of Model Predictive Control and Reinforcement Learning: Survey and Classificationcited by 19
- 2021Model-Invariant State Abstractions for Model-Based Reinforcement Learningcited by 18
- 2021Scalable Online Planning via Reinforcement Learning Fine-Tuningcited by 18
- 2021Learning Pessimism for Robust and Efficient Off-Policy Reinforcement Learningcited by 18
- 2022Live in the Moment: Learning Dynamics Model Adapted to Evolving Policycited by 17
- 2021Model-based Meta Reinforcement Learning using Graph Structured Surrogate Modelscited by 16
- 2021An Experimental Design Perspective on Model-Based Reinforcement Learningcited by 16
- 2021Evaluating model-based planning and planner amortization for continuous controlcited by 15
- 2025Rapidly Adapting Policies to the Real World via Simulation-Guided Fine-Tuningcited by 15
- 2023COPlanner: Plan to Roll Out Conservatively but to Explore Optimistically for Model-Based RLcited by 14
- 2021MapGo: Model-Assisted Policy Optimization for Goal-Oriented Taskscited by 13
- 2024MuDreamer: Learning Predictive World Models without Reconstructioncited by 13
- 2021Model-based micro-data reinforcement learning: what are the crucial model properties and which model to choose?cited by 11
- 2024Dynamic Model Predictive Shielding for Provably Safe Reinforcement Learningcited by 11
- 2021PerSim: Data-Efficient Offline Reinforcement Learning with Heterogeneous Agents via Personalized Simulatorscited by 10
- 2022Generative Planning for Temporally Coordinated Exploration in Reinforcement Learningcited by 10
- 2022Learning Sampling Distributions for Model Predictive Controlcited by 10
- 2022Learning to Optimize in Model Predictive Controlcited by 10
- 2024Efficient Multi-agent Reinforcement Learning by Planningcited by 9
- 2021Measuring Progress in Deep Reinforcement Learning Sample Efficiencycited by 8
- 2022EUCLID: Towards Efficient Unsupervised Reinforcement Learning with Multi-choice Dynamics Modelcited by 8
50 most cited of 90 All 90 (JSON)
- 2026FlexiWorld: Learning and Planning via Flexible Action Chunks Across Multiple Time Scales28 Sep 2026
- 2026Amortising Trajectory Optimisation for Residual MPC via Implicit Contact Differentiation27 Jul 2026
- 2026HDFlow: Hierarchical Diffusion-Flow Planning for Long-horizon Tasks6 May 2026
- 2026WOMBET: World Model-Based Experience Transfer for Robust and Sample-efficient Reinforcement Learning10 Apr 2026
- 2026Laplacian Representations for Decision-Time Planning4 Feb 2026
- 2026Neural Nonmyopic Bayesian Optimization in Dynamic Cost Settings10 Jan 2026
- 2025SCOPE: Language Models as One-Time Teacher for Hierarchical Planning in Text Environments10 Dec 2025
- 2025Bootstrap Off-policy with World Model1 Nov 2025
- 2025RT-HCP: Dealing with Inference Delays and Sample Efficiency to Learn Directly on Robotic Platforms8 Sep 2025
- 2025First Order Model-Based RL through Decoupled Backpropagation29 Aug 2025
- 2025Bounding Distributional Shifts in World Modeling through Novelty Detection8 Aug 2025
- 2025Continual Reinforcement Learning by Planning with Online World Models12 Jul 2025
38 more
- 2025Automating Experimental Optics with Sample Efficient Machine Learning Methods18 Mar 2025
- 2025Learning Transformer-based World Models with Contrastive Predictive Coding6 Mar 2025
- 2025Rapidly Adapting Policies to the Real World via Simulation-Guided Fine-Tuning4 Feb 2025
- 2025Synthesis of Model Predictive Control and Reinforcement Learning: Survey and Classification4 Feb 2025
- 2024Understanding World or Predicting Future? A Comprehensive Survey of World Models21 Nov 2024
- 2024Learning Multiple Initial Solutions to Optimization Problems4 Nov 2024
- 2024Bisimulation metric for Model Predictive Control6 Oct 2024
- 2024Navigating the Human Maze: Real-Time Robot Pathfinding with Generative Imitation Learning7 Aug 2024
- 2024Physics-Informed Model and Hybrid Planning for Efficient Dyna-Style Reinforcement Learning2 Jul 2024
- 2024MuDreamer: Learning Predictive World Models without Reconstruction23 May 2024
- 2024Dynamic Model Predictive Shielding for Provably Safe Reinforcement Learning22 May 2024
- 2024Efficient Multi-agent Reinforcement Learning by Planning20 May 2024
- 2024Robust Model Based Reinforcement Learning Using L1 Adaptive Control21 Mar 2024
- 2024Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learning5 Feb 2024
- 2024Bridging Evolutionary Algorithms and Reinforcement Learning: A Comprehensive Survey on Hybrid Algorithms22 Jan 2024
- 2024Simple Hierarchical Planning with Diffusion5 Jan 2024
- 2023PcLast: Discovering Plannable Continuous Latent States6 Nov 2023
- 2023COPlanner: Plan to Roll Out Conservatively but to Explore Optimistically for Model-Based RL11 Oct 2023
- 2023On Representation Complexity of Model-based and Model-free Reinforcement Learning3 Oct 2023
- 2023How to Fine-tune the Model: Unified Model Shift and Model Bias Policy Optimization22 Sep 2023
- 2023Machine Learning Meets Advanced Robotic Manipulation22 Sep 2023
- 2023Mind the Uncertainty: Risk-Aware and Actively Exploring Model-Based Reinforcement Learning11 Sep 2023
- 2023DREAMWALKER: Mental Planning for Continuous Vision-Language Navigation14 Aug 2023
- 2023Deep Generative Models for Decision-Making and Control15 Jun 2023
- 2023Stabilizing Contrastive RL: Techniques for Robotic Goal Reaching from Offline Data6 Jun 2023
- 2023Sample-efficient Real-time Planning with Curiosity Cross-Entropy Method and Contrastive Learning7 Mar 2023
- 2023Learning a model is paramount for sample efficiency in reinforcement learning control of PDEs14 Feb 2023
- 2022A Simple Decentralized Cross-Entropy Method16 Dec 2022
- 2022Learning to Optimize in Model Predictive Control5 Dec 2022
- 2022Learning Sampling Distributions for Model Predictive Control5 Dec 2022
- 2022Scaling up and Stabilizing Differentiable Planning with Implicit Differentiation24 Oct 2022
- 2022EUCLID: Towards Efficient Unsupervised Reinforcement Learning with Multi-choice Dynamics Model2 Oct 2022
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objective18 Sep 2022
- 2022Conservative Dual Policy Optimization for Efficient Model-Based Reinforcement Learning16 Sep 2022
- 2022Variational Inference for Model-Free and Model-Based Reinforcement Learning4 Sep 2022
- 2022Efficient Planning in a Compact Latent Action Space22 Aug 2022
- 2022Making Reinforcement Learning Work on Swimmer16 Aug 2022
- 2022Live in the Moment: Learning Dynamics Model Adapted to Evolving Policy25 Jul 2022
50 newest of 90 All 90 (JSON)
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2020Mastering Atari with Discrete World Modelscited by 990
- 2021Offline Reinforcement Learning as One Big Sequence Modeling Problemcited by 636
- 2022Temporal Difference Learning for Model Predictive Controlcited by 375
- 2024Understanding World or Predicting Future? A Comprehensive Survey of World Modelscited by 161
- 2023DREAMWALKER: Mental Planning for Continuous Vision-Language Navigationcited by 82
- 2020World Model as a Graph: Learning Latent Landmarks for Planningcited by 67
- 2022A Survey on Model-based Reinforcement Learningcited by 65
- 2025Learning Transformer-based World Models with Contrastive Predictive Codingcited by 20
- 2024MuDreamer: Learning Predictive World Models without Reconstructioncited by 13
- 2025Continual Reinforcement Learning by Planning with Online World Modelscited by 8
- 2025Bootstrap Off-policy with World Modelcited by 6
12 most cited of 15 All 15 (JSON)
- 2026FlexiWorld: Learning and Planning via Flexible Action Chunks Across Multiple Time Scales28 Sep 2026
- 2026WOMBET: World Model-Based Experience Transfer for Robust and Sample-efficient Reinforcement Learning10 Apr 2026
- 2025Bootstrap Off-policy with World Model1 Nov 2025
- 2025Bounding Distributional Shifts in World Modeling through Novelty Detection8 Aug 2025
- 2025Continual Reinforcement Learning by Planning with Online World Models12 Jul 2025
- 2025Learning Transformer-based World Models with Contrastive Predictive Coding6 Mar 2025
- 2024Understanding World or Predicting Future? A Comprehensive Survey of World Models21 Nov 2024
- 2024MuDreamer: Learning Predictive World Models without Reconstruction23 May 2024
- 2023DREAMWALKER: Mental Planning for Continuous Vision-Language Navigation14 Aug 2023
- 2022A Survey on Model-based Reinforcement Learning19 Jun 2022
- 2022Temporal Difference Learning for Model Predictive Control9 Mar 2022
- 2021Offline Reinforcement Learning as One Big Sequence Modeling Problem3 Jun 2021
12 newest of 15 All 15 (JSON)
- 2020MOPO: Model-based Offline Policy Optimizationcited by 630
- 2020MOReL : Model-Based Offline Reinforcement Learningcited by 582
- 2021Actionable Models: Unsupervised Offline Reinforcement Learning of Robotic Skillscited by 125
- 2021Model-Based Offline Planning with Trajectory Pruningcited by 30
- 2022Double Check Your State Before Trusting It: Confidence-Aware Bidirectional Offline Model-Based Imaginationcited by 21
- 2021PerSim: Data-Efficient Offline Reinforcement Learning with Heterogeneous Agents via Personalized Simulatorscited by 10
- 2024Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learningcited by 1
- 2022Reinforcement Learning from Demonstrations by Novel Interactive Expert and Application to Automatic Berthing Control Systems for Unmanned Surface Vesselcited by 0
- 2024Navigating the Human Maze: Real-Time Robot Pathfinding with Generative Imitation Learningcited by 0
- 2024Navigating the Human Maze: Real-Time Robot Pathfinding with Generative Imitation Learning7 Aug 2024
- 2024Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learning5 Feb 2024
- 2022Double Check Your State Before Trusting It: Confidence-Aware Bidirectional Offline Model-Based Imagination16 Jun 2022
- 2022Reinforcement Learning from Demonstrations by Novel Interactive Expert and Application to Automatic Berthing Control Systems for Unmanned Surface Vessel23 Feb 2022
- 2021Model-Based Offline Planning with Trajectory Pruning16 May 2021
- 2021Actionable Models: Unsupervised Offline Reinforcement Learning of Robotic Skills15 Apr 2021
- 2021PerSim: Data-Efficient Offline Reinforcement Learning with Heterogeneous Agents via Personalized Simulators13 Feb 2021
- 2020MOPO: Model-based Offline Policy Optimization27 May 2020
- 2020MOReL : Model-Based Offline Reinforcement Learning12 May 2020
- 2024Simple Hierarchical Planning with Diffusioncited by 71
- 2026HDFlow: Hierarchical Diffusion-Flow Planning for Long-horizon Taskscited by 2
- 2026HDFlow: Hierarchical Diffusion-Flow Planning for Long-horizon Tasks6 May 2026
- 2024Simple Hierarchical Planning with Diffusion5 Jan 2024
- 2022Tutorial on amortized optimizationcited by 83
- 2021Residual Pathway Priors for Soft Equivariance Constraintscited by 74
- 2022Safe Reinforcement Learning by Imagining the Near Futurecited by 67
- 2022Efficient Planning in a Compact Latent Action Spacecited by 57
- 2022Skill-based Model-based Reinforcement Learningcited by 50
- 2021MBRL-Lib: A Modular Library for Model-based Reinforcement Learningcited by 41
- 2022Deconstructing the Inductive Biases of Hamiltonian Neural Networkscited by 31
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objectivecited by 28
- 2021Model Predictive Actor-Critic: Accelerating Robot Skill Acquisition with Deep Reinforcement Learningcited by 25
- 2021Sample-Efficient Reinforcement Learning via Conservative Model-Based Actor-Criticcited by 23
- 2024Bridging Evolutionary Algorithms and Reinforcement Learning: A Comprehensive Survey on Hybrid Algorithmscited by 23
- 2021Where to go next: Learning a Subgoal Recommendation Policy for Navigation Among Pedestrianscited by 20
12 most cited of 64 All 64 (JSON)
- 2026Amortising Trajectory Optimisation for Residual MPC via Implicit Contact Differentiation27 Jul 2026
- 2026Laplacian Representations for Decision-Time Planning4 Feb 2026
- 2026Neural Nonmyopic Bayesian Optimization in Dynamic Cost Settings10 Jan 2026
- 2025SCOPE: Language Models as One-Time Teacher for Hierarchical Planning in Text Environments10 Dec 2025
- 2025RT-HCP: Dealing with Inference Delays and Sample Efficiency to Learn Directly on Robotic Platforms8 Sep 2025
- 2025First Order Model-Based RL through Decoupled Backpropagation29 Aug 2025
- 2025Automating Experimental Optics with Sample Efficient Machine Learning Methods18 Mar 2025
- 2025Rapidly Adapting Policies to the Real World via Simulation-Guided Fine-Tuning4 Feb 2025
- 2025Synthesis of Model Predictive Control and Reinforcement Learning: Survey and Classification4 Feb 2025
- 2024Learning Multiple Initial Solutions to Optimization Problems4 Nov 2024
- 2024Bisimulation metric for Model Predictive Control6 Oct 2024
- 2024Physics-Informed Model and Hybrid Planning for Efficient Dyna-Style Reinforcement Learning2 Jul 2024
12 newest of 64 All 64 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2024PWM: Policy Learning with Multi-Task World Models
- 2025Accelerating Model-Based Reinforcement Learning with State-Space World Models
- 2025Bootstrapped Model Predictive Control
- 2020Ready Policy One: World Building Through Active Learning
- 2025Flexible Locomotion Learning with Diffusion Model Predictive Control
- 2025Robotic World Model: A Neural Network Simulator for Robust Policy Optimization in Robotics
- 2026Dream-MPC: Gradient-Based Model Predictive Control with Latent Imagination
- 2024Diffusion-based learning of contact plans for agile locomotion ↗