PIPPS: Flexible Model-Based Policy Search Robust to the Curse of Chaos
Not on the map. Why?
Abstract (arXiv)
Previously, the exploding gradient problem has been explained to be central in deep learning and model-based reinforcement learning, because it causes numerical issues and instability in optimization. Our experiments in model-based reinforcement learning imply that the problem is not just a numerical issue, but it may be caused by a fundamental chaos-like nature of long chains of nonlinear computations. Not only do the magnitudes of the gradients become large, the direction of the gradients becomes essentially random. We show that reparameterization gradients suffer from the problem, while likelihood ratio gradients are robust. Using our insights, we develop a model-based policy search framework, Probabilistic Inference for Particle-Based Policy Search (PIPPS), which is easily extensible, and allows for almost arbitrary models and policies, while simultaneously matching the performance of previous data-efficient learning algorithms. Finally, we invent the total propagation algorithm, which efficiently computes a union over all pathwise derivative depths during a single backwards pass, automatically giving greater weight to estimators with lower variance, sometimes improving over reparameterization gradients by 106 times.
Key facts. Cited by 56 papers in the map (data through 7 Oct 2026). First posted on arXiv 4 Feb 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 2; 2021: 10; 2022: 6; 2023: 13; 2024: 8; 2025: 8; 2026: 9.
56 papers.
Sort:
By area:
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,873
- 2020OPAL: Offline Primitive Discovery for Accelerating Offline Reinforcement Learningcited by 164
- 2023Denoising Diffusion Samplerscited by 144
- 2022Accelerated Policy Learning with Parallel Differentiable Simulationcited by 92
- 2022Tutorial on amortized optimizationcited by 83
- 2021Gradients are Not All You Needcited by 82
- 2021Unbiased Gradient Estimation in Unrolled Computation Graphs with Persistent Evolution Strategiescited by 52
- 2025What Makes a Good Diffusion Planner for Decision Making?cited by 37
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objectivecited by 28
- 2024Adaptive Horizon Actor-Critic for Policy Learning in Contact-Rich Differentiable Simulationcited by 18
- 2021Model-based Meta Reinforcement Learning using Graph Structured Surrogate Modelscited by 16
- 2022Symbolic Learning to Optimize: Towards Interpretability and Scalabilitycited by 16
38 more
- 2023Butterfly Effects of SGD Noise: Error Amplification in Behavior Cloning and Autoregressioncited by 14
- 2024Stabilizing Reinforcement Learning in Differentiable Multiphysics Simulationcited by 14
- 2021A unified view of likelihood ratio and reparameterization gradientscited by 12
- 2023Policy Optimization in a Noisy Neighborhood: On Return Landscapes in Continuous Controlcited by 11
- 2021PerSim: Data-Efficient Offline Reinforcement Learning with Heterogeneous Agents via Personalized Simulatorscited by 10
- 2024Do Transformer World Models Give Better Policy Gradients?cited by 10
- 2021Model-free Policy Learning with Reward Gradientscited by 9
- 2021Mixed Policy Gradient: off-policy reinforcement learning driven jointly by data and modelcited by 8
- 2023Fractal Landscapes in Policy Optimizationcited by 8
- 2023Gradient Informed Proximal Policy Optimizationcited by 8
- 2021Model-Based Policy Search Using Monte Carlo Gradient Estimation with Real Systems Applicationcited by 7
- 2023Low-Variance Gradient Estimation in Unrolled Computation Graphs with ES-Singlecited by 7
- 2023Model-Based Reparameterization Policy Gradient Methods: Theory and Practical Algorithmscited by 7
- 2024μLO: Compute-Efficient Meta-Generalization of Learned Optimizerscited by 7
- 2023Diminishing Return of Value Expansion Methods in Model-Based Reinforcement Learningcited by 5
- 2023Variance-Reduced Gradient Estimation via Noise-Reuse in Online Evolution Strategiescited by 5
- 2024Enhancing Robustness in Deep Reinforcement Learning: A Lyapunov Exponent Approachcited by 4
- 2025Time-Aware World Model for Adaptive Prediction and Controlcited by 4
- 2023Toward Open-ended Embodied Tasks Solvingcited by 3
- 2024Stabilizing Backpropagation Through Time to Learn Complex Physicscited by 3
- 2023DiSProD: Differentiable Symbolic Propagation of Distributions for Planningcited by 2
- 2024Mollification Effects of Policy Gradient Methodscited by 2
- 2025Improving Controller Generalization with Dimensionless Markov Decision Processescited by 2
- 2026On Advantage Estimates for Max@K Policy Gradientscited by 2
- 2021Learning Robust Controllers Via Probabilistic Model-Based Policy Searchcited by 1
- 2022Reinforcement Learning for Personalized Drug Discovery and Design for Complex Diseases: A Systems Pharmacology Perspectivecited by 1
- 2025Accelerated Learning with Linear Temporal Logic using Differentiable Simulationcited by 1
- 2025Reparameterization Proximal Policy Optimizationcited by 1
- 2025Double Horizon Model-Based Policy Optimizationcited by 1
- 2026Dream-MPC: Gradient-Based Model Predictive Control with Latent Imaginationcited by 1
- 2026Retry Policy Gradients in Continuous Action Spacescited by 1
- 2026OrderGrad: Optimizing Beyond the Mean with Order-Statistic Policy Gradient Estimationcited by 1
- 2021Model-based Policy Search for Partially Measurable Systemscited by 0
- 2022Integrated Decision and Control for High-Level Automated Vehicles by Mixed Policy Gradient and Its Experiment Verificationcited by 0
- 2023Enabling Efficient, Reliable Real-World Reinforcement Learning with Approximate Physics-Based Modelscited by 0
- 2023Physics-informed reinforcement learning via probabilistic co-adjustment functionscited by 0
- 2024Diminishing Return of Value Expansion Methodscited by 0
- 2025Accelerating Model-Based Reinforcement Learning using Non-Linear Trajectory Optimizationcited by 0
50 most cited of 56 All 56 (JSON)
- 2026Gradient-estimator design overcomes trainability barriers in neural-network-based variational optimization16 Sep 2026
- 2026Large Distant Gradients Need Not Be Reliable: reliability-weighted credit assignment for long-horizon autoregressive forecasting11 Sep 2026
- 2026Accelerating Reinforcement Learning via MPC Solver-Gradient Guidance for Weights-varying MPC1 Sep 2026
- 2026Deconstructing Actor-Critic: A Large-scale Empirical Study of Design Components for Practitioners14 Jul 2026
- 2026OrderGrad: Optimizing Beyond the Mean with Order-Statistic Policy Gradient Estimation4 Jun 2026
- 2026On Advantage Estimates for Max@K Policy Gradients4 Jun 2026
- 2026Retry Policy Gradients in Continuous Action Spaces4 Jun 2026
- 2026Dream-MPC: Gradient-Based Model Predictive Control with Latent Imagination6 May 2026
- 2026Does "Do Differentiable Simulators Give Better Policy Gradients?'' Give Better Policy Gradients?20 Apr 2026
- 2025Double Horizon Model-Based Policy Optimization17 Dec 2025
- 2025Reparameterization Proximal Policy Optimization8 Aug 2025
- 2025Fast and Stable Diffusion Planning through Variational Adaptive Weighting20 Jun 2025
38 more
- 2025Time-Aware World Model for Adaptive Prediction and Control10 Jun 2025
- 2025Accelerating Model-Based Reinforcement Learning using Non-Linear Trajectory Optimization3 Jun 2025
- 2025Accelerated Learning with Linear Temporal Logic using Differentiable Simulation1 Jun 2025
- 2025Improving Controller Generalization with Dimensionless Markov Decision Processes14 Apr 2025
- 2025What Makes a Good Diffusion Planner for Decision Making?1 Mar 2025
- 2024Diminishing Return of Value Expansion Methods29 Dec 2024
- 2024Stabilizing Reinforcement Learning in Differentiable Multiphysics Simulation16 Dec 2024
- 2024Enhancing Robustness in Deep Reinforcement Learning: A Lyapunov Exponent Approach14 Oct 2024
- 2024μLO: Compute-Efficient Meta-Generalization of Learned Optimizers31 May 2024
- 2024Mollification Effects of Policy Gradient Methods28 May 2024
- 2024Adaptive Horizon Actor-Critic for Policy Learning in Contact-Rich Differentiable Simulation28 May 2024
- 2024Stabilizing Backpropagation Through Time to Learn Complex Physics3 May 2024
- 2024Do Transformer World Models Give Better Policy Gradients?7 Feb 2024
- 2023Gradient Informed Proximal Policy Optimization14 Dec 2023
- 2023Toward Open-ended Embodied Tasks Solving10 Dec 2023
- 2023Model-Based Reparameterization Policy Gradient Methods: Theory and Practical Algorithms30 Oct 2023
- 2023Fractal Landscapes in Policy Optimization24 Oct 2023
- 2023Butterfly Effects of SGD Noise: Error Amplification in Behavior Cloning and Autoregression17 Oct 2023
- 2023Policy Optimization in a Noisy Neighborhood: On Return Landscapes in Continuous Control26 Sep 2023
- 2023Physics-informed reinforcement learning via probabilistic co-adjustment functions11 Sep 2023
- 2023Enabling Efficient, Reliable Real-World Reinforcement Learning with Approximate Physics-Based Models16 Jul 2023
- 2023Variance-Reduced Gradient Estimation via Noise-Reuse in Online Evolution Strategies21 Apr 2023
- 2023Low-Variance Gradient Estimation in Unrolled Computation Graphs with ES-Single21 Apr 2023
- 2023Diminishing Return of Value Expansion Methods in Model-Based Reinforcement Learning7 Mar 2023
- 2023Denoising Diffusion Samplers27 Feb 2023
- 2023DiSProD: Differentiable Symbolic Propagation of Distributions for Planning3 Feb 2023
- 2022Integrated Decision and Control for High-Level Automated Vehicles by Mixed Policy Gradient and Its Experiment Verification19 Oct 2022
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objective18 Sep 2022
- 2022Accelerated Policy Learning with Parallel Differentiable Simulation14 Apr 2022
- 2022Symbolic Learning to Optimize: Towards Interpretability and Scalability13 Mar 2022
- 2022Tutorial on amortized optimization1 Feb 2022
- 2022Reinforcement Learning for Personalized Drug Discovery and Design for Complex Diseases: A Systems Pharmacology Perspective21 Jan 2022
- 2021Unbiased Gradient Estimation in Unrolled Computation Graphs with Persistent Evolution Strategies27 Dec 2021
- 2021Gradients are Not All You Need10 Nov 2021
- 2021Learning Robust Controllers Via Probabilistic Model-Based Policy Search26 Oct 2021
- 2021A unified view of likelihood ratio and reparameterization gradients31 May 2021
- 2021Model-free Policy Learning with Reward Gradients9 Mar 2021
- 2021Mixed Policy Gradient: off-policy reinforcement learning driven jointly by data and model23 Feb 2021
50 newest of 56 All 56 (JSON)
- 2023Denoising Diffusion Samplerscited by 144
- 2025What Makes a Good Diffusion Planner for Decision Making?cited by 37
- 2025Fast and Stable Diffusion Planning through Variational Adaptive Weightingcited by 0
- 2025Fast and Stable Diffusion Planning through Variational Adaptive Weighting20 Jun 2025
- 2025What Makes a Good Diffusion Planner for Decision Making?1 Mar 2025
- 2023Denoising Diffusion Samplers27 Feb 2023
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,873
- 2020OPAL: Offline Primitive Discovery for Accelerating Offline Reinforcement Learningcited by 164
- 2021PerSim: Data-Efficient Offline Reinforcement Learning with Heterogeneous Agents via Personalized Simulatorscited by 10
- 2021PerSim: Data-Efficient Offline Reinforcement Learning with Heterogeneous Agents via Personalized Simulators13 Feb 2021
- 2020OPAL: Offline Primitive Discovery for Accelerating Offline Reinforcement Learning26 Oct 2020
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problems4 May 2020
- 2026On Advantage Estimates for Max@K Policy Gradientscited by 2
- 2026On Advantage Estimates for Max@K Policy Gradients4 Jun 2026
- 2022Accelerated Policy Learning with Parallel Differentiable Simulationcited by 92
- 2022Tutorial on amortized optimizationcited by 83
- 2021Gradients are Not All You Needcited by 82
- 2021Unbiased Gradient Estimation in Unrolled Computation Graphs with Persistent Evolution Strategiescited by 52
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objectivecited by 28
- 2024Adaptive Horizon Actor-Critic for Policy Learning in Contact-Rich Differentiable Simulationcited by 18
- 2021Model-based Meta Reinforcement Learning using Graph Structured Surrogate Modelscited by 16
- 2022Symbolic Learning to Optimize: Towards Interpretability and Scalabilitycited by 16
- 2023Butterfly Effects of SGD Noise: Error Amplification in Behavior Cloning and Autoregressioncited by 14
- 2024Stabilizing Reinforcement Learning in Differentiable Multiphysics Simulationcited by 14
- 2021A unified view of likelihood ratio and reparameterization gradientscited by 12
- 2023Policy Optimization in a Noisy Neighborhood: On Return Landscapes in Continuous Controlcited by 11
12 most cited of 46 All 46 (JSON)
- 2026Gradient-estimator design overcomes trainability barriers in neural-network-based variational optimization16 Sep 2026
- 2026Large Distant Gradients Need Not Be Reliable: reliability-weighted credit assignment for long-horizon autoregressive forecasting11 Sep 2026
- 2026Accelerating Reinforcement Learning via MPC Solver-Gradient Guidance for Weights-varying MPC1 Sep 2026
- 2026Deconstructing Actor-Critic: A Large-scale Empirical Study of Design Components for Practitioners14 Jul 2026
- 2026OrderGrad: Optimizing Beyond the Mean with Order-Statistic Policy Gradient Estimation4 Jun 2026
- 2026Retry Policy Gradients in Continuous Action Spaces4 Jun 2026
- 2026Dream-MPC: Gradient-Based Model Predictive Control with Latent Imagination6 May 2026
- 2026Does "Do Differentiable Simulators Give Better Policy Gradients?'' Give Better Policy Gradients?20 Apr 2026
- 2025Double Horizon Model-Based Policy Optimization17 Dec 2025
- 2025Reparameterization Proximal Policy Optimization8 Aug 2025
- 2025Accelerating Model-Based Reinforcement Learning using Non-Linear Trajectory Optimization3 Jun 2025
- 2025Accelerated Learning with Linear Temporal Logic using Differentiable Simulation1 Jun 2025
12 newest of 46 All 46 (JSON)