Neural Network Dynamics for Model-Based Deep Reinforcement Learning with Model-Free Fine-Tuning
On the map: World models → Latent dynamics models and model-based RL
Abstract (arXiv)
Model-free deep reinforcement learning algorithms have been shown to be capable of learning a wide range of robotic skills, but typically require a very large number of samples to achieve good performance. Model-based algorithms, in principle, can provide for much more efficient learning, but have proven difficult to extend to expressive, high-capacity models such as deep neural networks. In this work, we demonstrate that medium-sized neural network models can in fact be combined with model predictive control (MPC) to achieve excellent sample complexity in a model-based reinforcement learning algorithm, producing stable and plausible gaits to accomplish various complex locomotion tasks. We also propose using deep neural network dynamics models to initialize a model-free learner, in order to combine the sample efficiency of model-based approaches with the high task-specific performance of model-free methods. We empirically demonstrate on MuJoCo locomotion tasks that our pure model-based approach trained on just random action data can follow arbitrary trajectories with excellent sample efficiency, and that our hybrid algorithm can accelerate model-free learning on high-speed benchmark tasks, achieving sample efficiency gains of 3-5x on swimmer, cheetah, hopper, and ant agents. Videos can be found at https://sites.google.com/view/mbmf
Key facts. Cited by 288 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 0 of its 46 references that are in the map. First posted on arXiv 8 Aug 2017.
Builds on
0 of 46 references.
No references here yet.
Built on by
Citing papers per year: 2018: 6; 2019: 6; 2020: 11; 2021: 78; 2022: 44; 2023: 44; 2024: 43; 2025: 31; 2026: 25.
288 papers.
Sort:
By area:
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,873
- 2018World Modelscited by 1,444
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,023
- 2020Mastering Atari with Discrete World Modelscited by 990
- 2021E(n) Equivariant Graph Neural Networkscited by 951
- 2018Recurrent World Models Facilitate Policy Evolutioncited by 761
- 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
- 2020MOReL : Model-Based Offline Reinforcement Learningcited by 582
- 2022Temporal Difference Learning for Model Predictive Controlcited by 375
38 more
- 2018Model-Ensemble Trust-Region Policy Optimizationcited by 214
- 2024Understanding World or Predicting Future? A Comprehensive Survey of World Modelscited by 161
- 2023Foundation Models for Decision Making: Problems, Methods, and Opportunitiescited by 160
- 2023Structured World Models from Human Videoscited by 144
- 2018Algorithmic Framework for Model-based Deep Reinforcement Learning with Theoretical Guaranteescited by 138
- 2021Pathdreamer: A World Model for Indoor Navigationcited by 117
- 2019Exploring Model-based Planning with Policy Networkscited by 90
- 2020State-Only Imitation Learning for Dexterous Manipulationcited by 90
- 2023Diffusion Models for Reinforcement Learning: A Surveycited by 84
- 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
- 2022Causal Dynamics Learning for Task-Independent State Abstractioncited by 64
- 2025Robotic World Model: A Neural Network Simulator for Robust Policy Optimization in Roboticscited by 47
- 2021MobILE: Model-Based Imitation Learning From Observation Alonecited by 44
- 2021Learning Neuro-Symbolic Relational Transition Models for Bilevel Planningcited by 43
- 2023A Survey on Causal Reinforcement Learningcited by 43
- 2019Random Expert Distillation: Imitation Learning via Expert Policy Support Estimationcited by 42
- 2020Overcoming Model Bias for Robust Offline Deep Reinforcement Learningcited by 42
- 2021Continuous-Time Model-Based Reinforcement Learningcited by 39
- 2023Adaptive Online Replanning with Diffusion Modelscited by 33
- 2021Autoregressive Dynamics Models for Offline Policy Evaluation and Optimizationcited by 32
- 2021Few-shot Language Coordination by Modeling Theory of Mindcited by 32
- 2024Diffusion World Model: Future Modeling Beyond Step-by-Step Rollout for Offline Reinforcement Learningcited by 31
- 2023Rethinking Closed-loop Training for Autonomous Drivingcited by 29
- 2024ASID: Active Exploration for System Identification in Robotic Manipulationcited 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
- 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
- 2023Explainable Reinforcement Learning via a Causal World Modelcited by 22
- 2021Learning Dynamics Models for Model Predictive Agentscited by 21
- 2021Safe Control with Neural Network Dynamic Modelscited by 21
- 2021Where to go next: Learning a Subgoal Recommendation Policy for Navigation Among Pedestrianscited by 20
- 2023GOPlan: Goal-conditioned Offline Reinforcement Learning by Planning with Learned Modelscited by 20
- 2022A Survey on Reinforcement Learning in Aviation Applicationscited by 19
- 2025Synthesis of Model Predictive Control and Reinforcement Learning: Survey and Classificationcited by 19
- 2021VisuoSpatial Foresight for Physical Sequential Fabric Manipulationcited by 18
50 most cited of 288 All 288 (JSON)
- 2026Think Fast, Plan Selectively: Adaptive Deliberation for Efficient Data-Driven MPC26 Sep 2026
- 2026Improving Offline Goal-Conditioned Reinforcement Learning via Selective Reward Stimulation16 Sep 2026
- 2026Contrastive Energy Fields for Inference-Time Procedure Planning in Instructional Videos17 Aug 2026
- 2026History-informed Lagrangian Neural Networks13 Aug 2026
- 2026Analytic Planning under Uncertainty with Moment Closure3 Aug 2026
- 2026When Is a Learned Command Adapter Worth It? Closed-Loop Identification and Counterfactual Auditing of Frozen Locomotion Policies23 Jul 2026
- 2026UBP2: Uncertainty-Balanced Preference Planning for Efficient Preference-based Reinforcement Learning17 Jun 2026
- 2026CLAW: Learning Continuous Latent Action World Models via Adversarial Latent Regularization2 Jun 2026
- 2026Intercepting the Future: Latent-Space Predictive World Model for Dynamic VLA Manipulation1 Jun 2026
- 2026All Models are Wrong, Knowing Where is Useful: On Model Uncertainty in Reinforcement Learning31 May 2026
- 2026CA-AC-MPC: CUDA-Accelerated Actor-Critic Model Predictive Control27 May 2026
- 2026Ready from Day 1: Population-Aware Coordination for Large-Scale Constrained Multi-Agent Systems12 May 2026
38 more
- 2026ELVIS: Ensemble-Calibrated Latent Imagination for Long-Horizon Visual MPC6 May 2026
- 2026Constraint-Enhanced Reinforcement Learning Based on Dynamic Decoupled Spherical Radial Squashing5 May 2026
- 2026Biased Dreams: Limitations to Epistemic Uncertainty Quantification in Latent Dynamics Models28 Apr 2026
- 2026A Co-Evolutionary Theory of Human-AI Coexistence: Mutualism, Governance, and Dynamics in Complex Societies24 Apr 2026
- 2026Advantage-Guided Diffusion for Model-Based Reinforcement Learning10 Apr 2026
- 2026PriPG-RL: Privileged Planner-Guided Reinforcement Learning for Partially Observable Systems with Anytime-Feasible MPC9 Apr 2026
- 2026Neural ODE and SDE Models for Adaptation and Planning in Model-Based Reinforcement Learning24 Mar 2026
- 2026ResWM: Residual-Action World Model for Visual RL11 Mar 2026
- 2026STRIDE: Structured Lagrangian and Stochastic Residual Dynamics via Flow Matching9 Mar 2026
- 2026Integrating Lagrangian Neural Networks into the Dyna Framework for Reinforcement Learning9 Mar 2026
- 2026TRIAGE: Type-Routed Interventions via Aleatoric-Epistemic Gated Estimation in Robotic Manipulation and Adaptive Perception -- Don't Treat All Uncertainty the Same9 Mar 2026
- 2026Cerebellar-Inspired Residual Control for Fault Recovery: From Inference-Time Adaptation to Structural Consolidation6 Feb 2026
- 2026The Surprising Difficulty of Search in Model-Based Reinforcement Learning29 Jan 2026
- 2025Closing the Train-Test Gap in World Models for Gradient-Based Planning10 Dec 2025
- 2025Model-Based Reinforcement Learning Under Confounding8 Dec 2025
- 2025Embodied Co-Design for Rapidly Evolving Agents: Taxonomy, Frontiers, and Challenges4 Dec 2025
- 2025Graph Neural Networks, Deep Reinforcement Learning and Probabilistic Topic Modeling for Strategic Multiagent Settings13 Nov 2025
- 2025Bootstrap Off-policy with World Model1 Nov 2025
- 2025Latent Mixture of Symmetries for Sample-Efficient Dynamic Learning4 Oct 2025
- 2025From Pixels to Factors: Learning Independently Controllable State Variables for Reinforcement Learning2 Oct 2025
- 2025A Forensic Analysis of Synthetic Data in RL: Diagnosing and Solving Algorithmic Failures in Model-Based Policy Optimization1 Oct 2025
- 2025Adversarial Diffusion for Robust Reinforcement Learning28 Sep 2025
- 2025Simulation Priors for Data-Efficient Deep Learning6 Sep 2025
- 2025Jacobian Exploratory Dual-Phase Reinforcement Learning for Dynamic Endoluminal Navigation of Deformable Continuum Robots30 Aug 2025
- 2025RecoMind: A Reinforcement Learning Framework for Optimizing In-Session User Satisfaction in Recommendation Systems31 Jul 2025
- 2025Model-free Reinforcement Learning for Model-based Control: Towards Safe, Interpretable and Sample-efficient Agents17 Jul 2025
- 2025Bregman Centroid Guided Cross-Entropy Method2 Jun 2025
- 2025Data-assimilated model-informed reinforcement learning2 Jun 2025
- 2025Scalable In-Context Q-Learning2 Jun 2025
- 2025Of Mice and Machines: A Comparison of Learning Between Real World Mice and RL Agents18 May 2025
- 2025Sample Efficient Reinforcement Learning via Large Vision Language Model Distillation16 May 2025
- 2025MInCo: Mitigating Information Conflicts in Distracted Visual Model-based Reinforcement Learning5 Apr 2025
- 2025Multi-agent Uncertainty-Aware Pessimistic Model-Based Reinforcement Learning for Connected Autonomous Vehicles26 Mar 2025
- 2025Look Before Leap: Look-Ahead Planning with Uncertainty in Reinforcement Learning26 Mar 2025
- 2025Deep Belief Markov Models for POMDP Inference17 Mar 2025
- 2025Accelerating Model-Based Reinforcement Learning with State-Space World Models27 Feb 2025
- 2025Distilling Reinforcement Learning Algorithms for In-Context Model-Based Planning26 Feb 2025
- 2025Model Selection for Off-policy Evaluation: New Algorithms and Experimental Protocol11 Feb 2025
50 newest of 288 All 288 (JSON)
- 2018World Modelscited by 1,444
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
- 2020Mastering Atari with Discrete World Modelscited by 990
- 2018Recurrent World Models Facilitate Policy Evolutioncited by 761
- 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
- 2022Temporal Difference Learning for Model Predictive Controlcited by 375
- 2018Model-Ensemble Trust-Region Policy Optimizationcited by 214
- 2024Understanding World or Predicting Future? A Comprehensive Survey of World Modelscited by 161
- 2023Structured World Models from Human Videoscited by 144
- 2018Algorithmic Framework for Model-based Deep Reinforcement Learning with Theoretical Guaranteescited by 138
12 most cited of 36 All 36 (JSON)
- 2026Think Fast, Plan Selectively: Adaptive Deliberation for Efficient Data-Driven MPC26 Sep 2026
- 2026CLAW: Learning Continuous Latent Action World Models via Adversarial Latent Regularization2 Jun 2026
- 2026Intercepting the Future: Latent-Space Predictive World Model for Dynamic VLA Manipulation1 Jun 2026
- 2026Advantage-Guided Diffusion for Model-Based Reinforcement Learning10 Apr 2026
- 2026ResWM: Residual-Action World Model for Visual RL11 Mar 2026
- 2025Closing the Train-Test Gap in World Models for Gradient-Based Planning10 Dec 2025
- 2025Bootstrap Off-policy with World Model1 Nov 2025
- 2025Scalable In-Context Q-Learning2 Jun 2025
- 2025Accelerating Model-Based Reinforcement Learning with State-Space World Models27 Feb 2025
- 2025Robotic World Model: A Neural Network Simulator for Robust Policy Optimization in Robotics17 Jan 2025
- 2024Understanding World or Predicting Future? A Comprehensive Survey of World Models21 Nov 2024
- 2024Grounded Answers for Multi-agent Decision-making Problem through Generative World Model3 Oct 2024
12 newest of 36 All 36 (JSON)
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,873
- 2020MOReL : Model-Based Offline Reinforcement Learningcited by 582
- 2020State-Only Imitation Learning for Dexterous Manipulationcited by 90
- 2021MobILE: Model-Based Imitation Learning From Observation Alonecited by 44
- 2019Random Expert Distillation: Imitation Learning via Expert Policy Support Estimationcited by 42
- 2020Overcoming Model Bias for Robust Offline Deep Reinforcement Learningcited by 42
- 2023GOPlan: Goal-conditioned Offline Reinforcement Learning by Planning with Learned Modelscited by 20
- 2019On-Policy Robot Imitation Learning from a Converging Supervisorcited by 10
- 2022Model-based trajectory stitching for improved behavioural cloning and its applicationscited by 7
- 2019Randomized Adversarial Imitation Learning for Autonomous Drivingcited by 6
- 2023Learning non-Markovian Decision-Making from State-only Sequencescited by 6
- 2025Model Selection for Off-policy Evaluation: New Algorithms and Experimental Protocolcited by 6
12 most cited of 27 All 27 (JSON)
- 2026Improving Offline Goal-Conditioned Reinforcement Learning via Selective Reward Stimulation16 Sep 2026
- 2025Model Selection for Off-policy Evaluation: New Algorithms and Experimental Protocol11 Feb 2025
- 2024COSBO: Conservative Offline Simulation-Based Policy Optimization22 Sep 2024
- 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
- 2023GOPlan: Goal-conditioned Offline Reinforcement Learning by Planning with Learned Models30 Oct 2023
- 2023Multi-timestep models for Model-based Reinforcement Learning9 Oct 2023
- 2023Offline to Online Learning for Real-Time Bandwidth Estimation23 Sep 2023
- 2023Learning non-Markovian Decision-Making from State-only Sequences27 Jun 2023
- 2022Model-based trajectory stitching for improved behavioural cloning and its applications8 Dec 2022
- 2022Imitating, Fast and Slow: Robust learning from demonstrations via decision-time planning7 Apr 2022
- 2022Reinforcement Learning from Demonstrations by Novel Interactive Expert and Application to Automatic Berthing Control Systems for Unmanned Surface Vessel23 Feb 2022
12 newest of 27 All 27 (JSON)
- 2023Diffusion Models for Reinforcement Learning: A Surveycited by 84
- 2023Adaptive Online Replanning with Diffusion Modelscited by 33
- 2024Diffusion Spectral Representation for Reinforcement Learningcited by 15
- 2024Diffusion Models as Optimizers for Efficient Planning in Offline RLcited by 8
- 2025Adversarial Diffusion for Robust Reinforcement Learningcited by 4
- 2023PlayBest: Professional Basketball Player Behavior Synthesis via Planning with Diffusioncited by 3
- 2024A Contact Model based on Denoising Diffusion to Learn Variable Impedance Control for Contact-rich Manipulationcited by 2
- 2026STRIDE: Structured Lagrangian and Stochastic Residual Dynamics via Flow Matchingcited by 0
- 2026STRIDE: Structured Lagrangian and Stochastic Residual Dynamics via Flow Matching9 Mar 2026
- 2025Adversarial Diffusion for Robust Reinforcement Learning28 Sep 2025
- 2024Diffusion Models as Optimizers for Efficient Planning in Offline RL23 Jul 2024
- 2024Diffusion Spectral Representation for Reinforcement Learning23 Jun 2024
- 2024A Contact Model based on Denoising Diffusion to Learn Variable Impedance Control for Contact-rich Manipulation20 Mar 2024
- 2023Diffusion Models for Reinforcement Learning: A Survey2 Nov 2023
- 2023Adaptive Online Replanning with Diffusion Models14 Oct 2023
- 2023PlayBest: Professional Basketball Player Behavior Synthesis via Planning with Diffusion7 Jun 2023
- 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
- 2021Instance-Aware Predictive Navigation in Multi-Agent Environmentscited by 5
- 2021Uncertainty-Aware Model-Based Reinforcement Learning with Application to Autonomous Drivingcited by 3
- 2021Parallelized and Randomized Adversarial Imitation Learning for Safety-Critical Self-Driving Vehiclescited by 1
- 2023Multi-Agent Probabilistic Ensembles with Trajectory Sampling for Connected Autonomous Vehiclescited by 1
- 2025Multi-agent Uncertainty-Aware Pessimistic Model-Based Reinforcement Learning for Connected Autonomous Vehiclescited by 0
- 2025Multi-agent Uncertainty-Aware Pessimistic Model-Based Reinforcement Learning for Connected Autonomous Vehicles26 Mar 2025
- 2023Multi-Agent Probabilistic Ensembles with Trajectory Sampling for Connected Autonomous Vehicles21 Dec 2023
- 2023Rethinking Closed-loop Training for Autonomous Driving27 Jun 2023
- 2021Parallelized and Randomized Adversarial Imitation Learning for Safety-Critical Self-Driving Vehicles26 Dec 2021
- 2021Uncertainty-Aware Model-Based Reinforcement Learning with Application to Autonomous Driving23 Jun 2021
- 2021Safe Model-based Off-policy Reinforcement Learning for Eco-Driving in Connected and Automated Hybrid Electric Vehicles25 May 2021
- 2021Instance-Aware Predictive Navigation in Multi-Agent Environments14 Jan 2021
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,023
- 2021E(n) Equivariant Graph Neural Networkscited by 951
- 2023Foundation Models for Decision Making: Problems, Methods, and Opportunitiescited by 160
- 2022Causal Dynamics Learning for Task-Independent State Abstractioncited by 64
- 2021Learning Neuro-Symbolic Relational Transition Models for Bilevel Planningcited by 43
- 2023A Survey on Causal Reinforcement Learningcited by 43
- 2021Continuous-Time Model-Based Reinforcement Learningcited by 39
- 2021Autoregressive Dynamics Models for Offline Policy Evaluation and Optimizationcited by 32
- 2021Few-shot Language Coordination by Modeling Theory of Mindcited by 32
- 2024ASID: Active Exploration for System Identification in Robotic Manipulationcited 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
12 most cited of 205 All 205 (JSON)
- 2026Contrastive Energy Fields for Inference-Time Procedure Planning in Instructional Videos17 Aug 2026
- 2026History-informed Lagrangian Neural Networks13 Aug 2026
- 2026Analytic Planning under Uncertainty with Moment Closure3 Aug 2026
- 2026When Is a Learned Command Adapter Worth It? Closed-Loop Identification and Counterfactual Auditing of Frozen Locomotion Policies23 Jul 2026
- 2026All Models are Wrong, Knowing Where is Useful: On Model Uncertainty in Reinforcement Learning31 May 2026
- 2026CA-AC-MPC: CUDA-Accelerated Actor-Critic Model Predictive Control27 May 2026
- 2026Ready from Day 1: Population-Aware Coordination for Large-Scale Constrained Multi-Agent Systems12 May 2026
- 2026ELVIS: Ensemble-Calibrated Latent Imagination for Long-Horizon Visual MPC6 May 2026
- 2026Constraint-Enhanced Reinforcement Learning Based on Dynamic Decoupled Spherical Radial Squashing5 May 2026
- 2026Biased Dreams: Limitations to Epistemic Uncertainty Quantification in Latent Dynamics Models28 Apr 2026
- 2026A Co-Evolutionary Theory of Human-AI Coexistence: Mutualism, Governance, and Dynamics in Complex Societies24 Apr 2026
- 2026PriPG-RL: Privileged Planner-Guided Reinforcement Learning for Partially Observable Systems with Anytime-Feasible MPC9 Apr 2026
12 newest of 205 All 205 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2024An Efficient Model-Based Approach on Learning Agile Motor Skills without Reinforcement
- 2022A Walk in the Park: Learning to Walk in 20 Minutes With Model-Free Reinforcement Learning
- 2026Learning Loco-Manipulation From SMPC Demonstrations With Sparse Offline-to-Online RL
- 2022MoDem: Accelerating Visual Model-Based Reinforcement Learning with Demonstrations
- 2025Flexible Locomotion Learning with Diffusion Model Predictive Control
- 2023DecAP: Decaying Action Priors for Accelerated Imitation Learning of Torque-Based Legged Locomotion Policies
- 2023Imitating and Finetuning Model Predictive Control for Robust and Symmetric Quadrupedal Locomotion
- 2020Neural Dynamic Policies for End-to-End Sensorimotor Learning