Model-Augmented Actor-Critic: Backpropagating through Paths
Not on the map. Why?
Abstract (arXiv)
Current model-based reinforcement learning approaches use the model simply as a learned black-box simulator to augment the data for policy optimization or value function learning. In this paper, we show how to make more effective use of the model by exploiting its differentiability. We construct a policy optimization algorithm that uses the pathwise derivative of the learned model and policy across future timesteps. Instabilities of learning across many timesteps are prevented by using a terminal value function, learning the policy in an actor-critic fashion. Furthermore, we present a derivation on the monotonic improvement of our objective in terms of the gradient error in the model and value function. We show that our approach (i) is consistently more sample efficient than existing state-of-the-art model-based algorithms, (ii) matches the asymptotic performance of model-free algorithms, and (iii) scales to long horizons, a regime where typically past model-based approaches have struggled.
Key facts. Cited by 61 papers in the map (data through 7 Oct 2026). First posted on arXiv 16 May 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 14; 2022: 19; 2023: 10; 2024: 10; 2025: 3; 2026: 5.
61 papers.
Sort:
By area:
- 2022Temporal Difference Learning for Model Predictive Controlcited by 375
- 2021COMBO: Conservative Offline Model-Based Policy Optimizationcited by 365
- 2021Randomized Ensembled Double Q-Learning: Learning Fast Without a Modelcited by 278
- 2022RAMBO-RL: Robust Adversarial Model-Based Offline Reinforcement Learningcited by 115
- 2022Accelerated Policy Learning with Parallel Differentiable Simulationcited by 92
- 2021Gradients are Not All You Needcited by 82
- 2022A Survey on Model-based Reinforcement Learningcited by 65
- 2022VRL3: A Data-Driven Framework for Visual Deep Reinforcement Learningcited by 51
- 2021Provable Model-based Nonlinear Bandit and Reinforcement Learning: Shelve Optimism, Embrace Virtual Curvaturecited by 32
- 2025CarPlanner: Consistent Auto-regressive Trajectory Planning for Large-scale Reinforcement Learning in Autonomous Drivingcited by 31
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objectivecited by 28
- 2022Deep Interactive Motion Prediction and Planning: Playing Games with Motion Prediction Modelscited by 27
38 more
- 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
- 2022The Free Energy Principle for Perception and Action: A Deep Learning Perspectivecited by 24
- 2021Sample-Efficient Reinforcement Learning via Conservative Model-Based Actor-Criticcited by 23
- 2021Contrastive Active Inferencecited by 21
- 2023Deep Learning in Deterministic Computational Mechanicscited by 21
- 2023Is Model Ensemble Necessary? Model-based RL via a Single Model with Lipschitz Regularized Value Functioncited by 19
- 2022Live in the Moment: Learning Dynamics Model Adapted to Evolving Policycited by 17
- 2022Physics-Informed Model-Based Reinforcement Learningcited by 17
- 2023Simplified Temporal Consistency Reinforcement Learningcited by 17
- 2023IQL-TD-MPC: Implicit Q-Learning for Hierarchical Model Predictive Controlcited by 14
- 2024Stabilizing Reinforcement Learning in Differentiable Multiphysics Simulationcited by 14
- 2021On Effective Scheduling of Model-based Reinforcement Learningcited by 13
- 2024MOTO: Offline Pre-training to Online Fine-tuning for Model-based Robot Learningcited by 13
- 2024Do Transformer World Models Give Better Policy Gradients?cited by 10
- 2024Any-step Dynamics Model Improves Future Predictions for Online and Offline Reinforcement Learningcited by 10
- 2023Model-Based Reparameterization Policy Gradient Methods: Theory and Practical Algorithmscited by 7
- 2021Optimistic Reinforcement Learning by Forward Kullback-Leibler Divergence Optimizationcited by 6
- 2021On-Policy Model Errors in Reinforcement Learningcited by 6
- 2022Integrating Symmetry into Differentiable Planning with Steerable Convolutionscited by 6
- 2022Model-based Reinforcement Learning with Multi-step Plan Value Estimationcited by 6
- 2022Scaling up and Stabilizing Differentiable Planning with Implicit Differentiationcited by 6
- 2024Optimistic Model Rollouts for Pessimistic Offline Policy Optimizationcited by 6
- 2024Grounded Answers for Multi-agent Decision-making Problem through Generative World Modelcited by 6
- 2021OffCon3: What is state of the art anyway?cited by 5
- 2022Proximal Policy Optimization with Adaptive Threshold for Symmetric Relative Density Ratiocited by 5
- 2022Conservative Dual Policy Optimization for Efficient Model-Based Reinforcement Learningcited by 5
- 2023Diminishing Return of Value Expansion Methods in Model-Based Reinforcement Learningcited by 5
- 2023Learning an Inventory Control Policy with General Inventory Arrival Dynamicscited by 5
- 2024Implicit Bias of Policy Gradient in Linear Quadratic Control: Extrapolation to Unseen Initial Statescited by 4
- 2025Differentiable Information Enhanced Model-Based Reinforcement Learningcited by 3
- 2025First Order Model-Based RL through Decoupled Backpropagationcited by 3
- 2026WIMLE: Uncertainty-Aware World Models with IMLE for Sample-Efficient Continuous Controlcited by 3
- 2022Imitation Learning via Differentiable Physicscited by 2
- 2022Backward Imitation and Forward Reinforcement Learning via Bi-directional Model Rolloutscited by 2
- 2023Hint assisted reinforcement learning: an application in radio astronomycited by 2
- 2024Physics-Informed Model and Hybrid Planning for Efficient Dyna-Style Reinforcement Learningcited by 2
- 2024Safe Reinforcement Learning using Finite-Horizon Gradient-based Estimationcited by 2
50 most cited of 61 All 61 (JSON)
- 2026Accelerating Reinforcement Learning via MPC Solver-Gradient Guidance for Weights-varying MPC1 Sep 2026
- 2026GradMAP: Gradient-Based Multi-Agent Proximal Learning for Grid-Edge Flexibility27 Apr 2026
- 2026Tempered Sequential Monte Carlo for Trajectory and Policy Optimization with Differentiable Dynamics23 Apr 2026
- 2026WIMLE: Uncertainty-Aware World Models with IMLE for Sample-Efficient Continuous Control15 Feb 2026
- 2026Coupled Local and Global World Models for Efficient First Order RL5 Feb 2026
- 2025First Order Model-Based RL through Decoupled Backpropagation29 Aug 2025
- 2025Differentiable Information Enhanced Model-Based Reinforcement Learning3 Mar 2025
- 2025CarPlanner: Consistent Auto-regressive Trajectory Planning for Large-scale Reinforcement Learning in Autonomous Driving27 Feb 2025
- 2024Diminishing Return of Value Expansion Methods29 Dec 2024
- 2024Stabilizing Reinforcement Learning in Differentiable Multiphysics Simulation16 Dec 2024
- 2024Safe Reinforcement Learning using Finite-Horizon Gradient-based Estimation15 Dec 2024
- 2024Grounded Answers for Multi-agent Decision-making Problem through Generative World Model3 Oct 2024
38 more
- 2024Physics-Informed Model and Hybrid Planning for Efficient Dyna-Style Reinforcement Learning2 Jul 2024
- 2024Any-step Dynamics Model Improves Future Predictions for Online and Offline Reinforcement Learning27 May 2024
- 2024Implicit Bias of Policy Gradient in Linear Quadratic Control: Extrapolation to Unseen Initial States12 Feb 2024
- 2024Do Transformer World Models Give Better Policy Gradients?7 Feb 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
- 2023An introduction to reinforcement learning for neuroscience13 Nov 2023
- 2023Model-Based Reparameterization Policy Gradient Methods: Theory and Practical Algorithms30 Oct 2023
- 2023Learning an Inventory Control Policy with General Inventory Arrival Dynamics26 Oct 2023
- 2023Deep Learning in Deterministic Computational Mechanics27 Sep 2023
- 2023Simplified Temporal Consistency Reinforcement Learning15 Jun 2023
- 2023IQL-TD-MPC: Implicit Q-Learning for Hierarchical Model Predictive Control1 Jun 2023
- 2023Diminishing Return of Value Expansion Methods in Model-Based Reinforcement Learning7 Mar 2023
- 2023Taylor TD-learning27 Feb 2023
- 2023Is Model Ensemble Necessary? Model-based RL via a Single Model with Lipschitz Regularized Value Function2 Feb 2023
- 2023Hint assisted reinforcement learning: an application in radio astronomy10 Jan 2023
- 2022Physics-Informed Model-Based Reinforcement Learning5 Dec 2022
- 2022Scaling up and Stabilizing Differentiable Planning with Implicit Differentiation24 Oct 2022
- 2022On Many-Actions Policy Gradient24 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
- 2022Model-based Reinforcement Learning with Multi-step Plan Value Estimation12 Sep 2022
- 2022What deep reinforcement learning tells us about human motor learning and vice-versa23 Aug 2022
- 2022Backward Imitation and Forward Reinforcement Learning via Bi-directional Model Rollouts4 Aug 2022
- 2022Live in the Moment: Learning Dynamics Model Adapted to Evolving Policy25 Jul 2022
- 2022The Free Energy Principle for Perception and Action: A Deep Learning Perspective13 Jul 2022
- 2022A Survey on Model-based Reinforcement Learning19 Jun 2022
- 2022Imitation Learning via Differentiable Physics10 Jun 2022
- 2022Integrating Symmetry into Differentiable Planning with Steerable Convolutions8 Jun 2022
- 2022RAMBO-RL: Robust Adversarial Model-Based Offline Reinforcement Learning26 Apr 2022
- 2022Accelerated Policy Learning with Parallel Differentiable Simulation14 Apr 2022
- 2022Deep Interactive Motion Prediction and Planning: Playing Games with Motion Prediction Models5 Apr 2022
- 2022Proximal Policy Optimization with Adaptive Threshold for Symmetric Relative Density Ratio18 Mar 2022
- 2022Temporal Difference Learning for Model Predictive Control9 Mar 2022
- 2022VRL3: A Data-Driven Framework for Visual Deep Reinforcement Learning17 Feb 2022
- 2021Sample-Efficient Reinforcement Learning via Conservative Model-Based Actor-Critic16 Dec 2021
- 2021On Effective Scheduling of Model-based Reinforcement Learning16 Nov 2021
- 2021Gradients are Not All You Need10 Nov 2021
50 newest of 61 All 61 (JSON)
- 2021COMBO: Conservative Offline Model-Based Policy Optimizationcited by 365
- 2022RAMBO-RL: Robust Adversarial Model-Based Offline Reinforcement Learningcited by 115
- 2023IQL-TD-MPC: Implicit Q-Learning for Hierarchical Model Predictive Controlcited by 14
- 2024MOTO: Offline Pre-training to Online Fine-tuning for Model-based Robot Learningcited by 13
- 2024Any-step Dynamics Model Improves Future Predictions for Online and Offline Reinforcement Learningcited by 10
- 2024Optimistic Model Rollouts for Pessimistic Offline Policy Optimizationcited by 6
- 2022Imitation Learning via Differentiable Physicscited by 2
- 2024Any-step Dynamics Model Improves Future Predictions for Online and Offline Reinforcement Learning27 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
- 2023IQL-TD-MPC: Implicit Q-Learning for Hierarchical Model Predictive Control1 Jun 2023
- 2022Imitation Learning via Differentiable Physics10 Jun 2022
- 2022RAMBO-RL: Robust Adversarial Model-Based Offline Reinforcement Learning26 Apr 2022
- 2021COMBO: Conservative Offline Model-Based Policy Optimization16 Feb 2021
- 2022Temporal Difference Learning for Model Predictive Controlcited by 375
- 2022A Survey on Model-based Reinforcement Learningcited by 65
- 2024Do Transformer World Models Give Better Policy Gradients?cited by 10
- 2024Grounded Answers for Multi-agent Decision-making Problem through Generative World Modelcited by 6
- 2026WIMLE: Uncertainty-Aware World Models with IMLE for Sample-Efficient Continuous Controlcited by 3
- 2026Coupled Local and Global World Models for Efficient First Order RLcited by 0
- 2026WIMLE: Uncertainty-Aware World Models with IMLE for Sample-Efficient Continuous Control15 Feb 2026
- 2026Coupled Local and Global World Models for Efficient First Order RL5 Feb 2026
- 2024Grounded Answers for Multi-agent Decision-making Problem through Generative World Model3 Oct 2024
- 2024Do Transformer World Models Give Better Policy Gradients?7 Feb 2024
- 2022A Survey on Model-based Reinforcement Learning19 Jun 2022
- 2022Temporal Difference Learning for Model Predictive Control9 Mar 2022
- 2021Randomized Ensembled Double Q-Learning: Learning Fast Without a Modelcited by 278
- 2022Accelerated Policy Learning with Parallel Differentiable Simulationcited by 92
- 2021Gradients are Not All You Needcited by 82
- 2022VRL3: A Data-Driven Framework for Visual Deep Reinforcement Learningcited by 51
- 2021Provable Model-based Nonlinear Bandit and Reinforcement Learning: Shelve Optimism, Embrace Virtual Curvaturecited by 32
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objectivecited by 28
- 2022Deep Interactive Motion Prediction and Planning: Playing Games with Motion Prediction Modelscited 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
- 2022The Free Energy Principle for Perception and Action: A Deep Learning Perspectivecited by 24
- 2021Sample-Efficient Reinforcement Learning via Conservative Model-Based Actor-Criticcited by 23
- 2021Contrastive Active Inferencecited by 21
12 most cited of 47 All 47 (JSON)
- 2026Accelerating Reinforcement Learning via MPC Solver-Gradient Guidance for Weights-varying MPC1 Sep 2026
- 2026GradMAP: Gradient-Based Multi-Agent Proximal Learning for Grid-Edge Flexibility27 Apr 2026
- 2026Tempered Sequential Monte Carlo for Trajectory and Policy Optimization with Differentiable Dynamics23 Apr 2026
- 2025First Order Model-Based RL through Decoupled Backpropagation29 Aug 2025
- 2025Differentiable Information Enhanced Model-Based Reinforcement Learning3 Mar 2025
- 2024Diminishing Return of Value Expansion Methods29 Dec 2024
- 2024Stabilizing Reinforcement Learning in Differentiable Multiphysics Simulation16 Dec 2024
- 2024Safe Reinforcement Learning using Finite-Horizon Gradient-based Estimation15 Dec 2024
- 2024Physics-Informed Model and Hybrid Planning for Efficient Dyna-Style Reinforcement Learning2 Jul 2024
- 2024Implicit Bias of Policy Gradient in Linear Quadratic Control: Extrapolation to Unseen Initial States12 Feb 2024
- 2023An introduction to reinforcement learning for neuroscience13 Nov 2023
- 2023Model-Based Reparameterization Policy Gradient Methods: Theory and Practical Algorithms30 Oct 2023
12 newest of 47 All 47 (JSON)