Efficient Wasserstein Natural Gradients for Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
A novel optimization approach is proposed for application to policy gradient methods and evolution strategies for reinforcement learning (RL). The procedure uses a computationally efficient Wasserstein natural gradient (WNG) descent that takes advantage of the geometry induced by a Wasserstein penalty to speed optimization. This method follows the recent theme in RL of including a divergence penalty in the objective to establish a trust region. Experiments on challenging tasks demonstrate improvements in both computational cost and performance over advanced baselines.
Key facts. Cited by 23 papers in the map (data through 14 Oct 2026). First posted on arXiv 12 Oct 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 4; 2022: 5; 2023: 8; 2024: 0; 2025: 2; 2026: 4.
23 papers.
Sort:
By area:
- 2023Confronting Reward Model Overoptimization with Constrained RLHFcited by 96
- 2021Tactical Optimism and Pessimism for Deep Reinforcement Learningcited by 55
- 2021MICo: Improved representations via sampling-based state similarity for Markov decision processescited by 43
- 2023ReLOAD: Reinforcement Learning with Optimistic Ascent-Descent for Last-Iterate Convergence in Constrained MDPscited by 20
- 2023The Geometry of Neural Nets' Parameter Spaces Under Reparametrizationcited by 16
- 2021A First-Occupancy Representation for Reinforcement Learningcited by 14
- 2025Wasserstein Policy Optimizationcited by 13
- 2022Learning General World Models in a Handful of Reward-Free Deploymentscited by 12
- 2022Geometry and convergence of natural policy gradient methodscited by 12
- 2021Towards an Understanding of Default Policies in Multitask Policy Optimizationcited by 8
- 2022Trust Region Policy Optimization with Optimal Transport Discrepancies: Duality and Algorithm for Continuous Actionscited by 8
- 2023Wasserstein Gradient Flows for Optimizing Gaussian Mixture Policiescited by 8
11 more
- 2023Provably Convergent Policy Optimization via Metric-aware Trust Region Methodscited by 6
- 2026Wasserstein Proximal Policy Gradientcited by 4
- 2023A State Representation for Diminishing Rewardscited by 3
- 2023A Conservative Approach for Few-Shot Transfer in Off-Dynamics Reinforcement Learningcited by 3
- 2025Constrained Sliced Wasserstein Embeddingcited by 2
- 2026Semantic-aware Wasserstein Policy Regularization for Large Language Model Alignmentcited by 2
- 2022Minimum Description Length Controlcited by 0
- 2022Transfer RL via the Undo Maps Formalismcited by 0
- 2023Wasserstein Diversity-Enriched Regularizer for Hierarchical Reinforcement Learningcited by 0
- 2026Wasserstein Policy Gradient for Entropy-Regularized Linear-Quadratic Controlcited by 0
- 2026Multi-step Proximal Policy Improvement in Offline Reinforcement Learningcited by 0
- 2026Multi-step Proximal Policy Improvement in Offline Reinforcement Learning3 Sep 2026
- 2026Wasserstein Policy Gradient for Entropy-Regularized Linear-Quadratic Control7 Aug 2026
- 2026Wasserstein Proximal Policy Gradient3 Mar 2026
- 2026Semantic-aware Wasserstein Policy Regularization for Large Language Model Alignment2 Feb 2026
- 2025Constrained Sliced Wasserstein Embedding2 Jun 2025
- 2025Wasserstein Policy Optimization1 May 2025
- 2023A Conservative Approach for Few-Shot Transfer in Off-Dynamics Reinforcement Learning24 Dec 2023
- 2023Confronting Reward Model Overoptimization with Constrained RLHF6 Oct 2023
- 2023A State Representation for Diminishing Rewards7 Sep 2023
- 2023Wasserstein Diversity-Enriched Regularizer for Hierarchical Reinforcement Learning2 Aug 2023
- 2023Provably Convergent Policy Optimization via Metric-aware Trust Region Methods25 Jun 2023
- 2023Wasserstein Gradient Flows for Optimizing Gaussian Mixture Policies17 May 2023
11 more
- 2023The Geometry of Neural Nets' Parameter Spaces Under Reparametrization14 Feb 2023
- 2023ReLOAD: Reinforcement Learning with Optimistic Ascent-Descent for Last-Iterate Convergence in Constrained MDPs2 Feb 2023
- 2022Transfer RL via the Undo Maps Formalism26 Nov 2022
- 2022Geometry and convergence of natural policy gradient methods3 Nov 2022
- 2022Learning General World Models in a Handful of Reward-Free Deployments23 Oct 2022
- 2022Trust Region Policy Optimization with Optimal Transport Discrepancies: Duality and Algorithm for Continuous Actions20 Oct 2022
- 2022Minimum Description Length Control17 Jul 2022
- 2021Towards an Understanding of Default Policies in Multitask Policy Optimization4 Nov 2021
- 2021A First-Occupancy Representation for Reinforcement Learning28 Sep 2021
- 2021MICo: Improved representations via sampling-based state similarity for Markov decision processes3 Jun 2021
- 2021Tactical Optimism and Pessimism for Deep Reinforcement Learning7 Feb 2021
- 2025Wasserstein Policy Optimizationcited by 13
- 2026Wasserstein Proximal Policy Gradientcited by 4
- 2026Wasserstein Proximal Policy Gradient3 Mar 2026
- 2025Wasserstein Policy Optimization1 May 2025
- 2021Tactical Optimism and Pessimism for Deep Reinforcement Learningcited by 55
- 2021MICo: Improved representations via sampling-based state similarity for Markov decision processescited by 43
- 2023ReLOAD: Reinforcement Learning with Optimistic Ascent-Descent for Last-Iterate Convergence in Constrained MDPscited by 20
- 2023The Geometry of Neural Nets' Parameter Spaces Under Reparametrizationcited by 16
- 2021A First-Occupancy Representation for Reinforcement Learningcited by 14
- 2022Geometry and convergence of natural policy gradient methodscited by 12
- 2021Towards an Understanding of Default Policies in Multitask Policy Optimizationcited by 8
- 2022Trust Region Policy Optimization with Optimal Transport Discrepancies: Duality and Algorithm for Continuous Actionscited by 8
- 2023Wasserstein Gradient Flows for Optimizing Gaussian Mixture Policiescited by 8
- 2023Provably Convergent Policy Optimization via Metric-aware Trust Region Methodscited by 6
- 2023A State Representation for Diminishing Rewardscited by 3
- 2023A Conservative Approach for Few-Shot Transfer in Off-Dynamics Reinforcement Learningcited by 3
12 most cited of 17 All 17 (JSON)
- 2026Wasserstein Policy Gradient for Entropy-Regularized Linear-Quadratic Control7 Aug 2026
- 2025Constrained Sliced Wasserstein Embedding2 Jun 2025
- 2023A Conservative Approach for Few-Shot Transfer in Off-Dynamics Reinforcement Learning24 Dec 2023
- 2023A State Representation for Diminishing Rewards7 Sep 2023
- 2023Wasserstein Diversity-Enriched Regularizer for Hierarchical Reinforcement Learning2 Aug 2023
- 2023Provably Convergent Policy Optimization via Metric-aware Trust Region Methods25 Jun 2023
- 2023Wasserstein Gradient Flows for Optimizing Gaussian Mixture Policies17 May 2023
- 2023The Geometry of Neural Nets' Parameter Spaces Under Reparametrization14 Feb 2023
- 2023ReLOAD: Reinforcement Learning with Optimistic Ascent-Descent for Last-Iterate Convergence in Constrained MDPs2 Feb 2023
- 2022Transfer RL via the Undo Maps Formalism26 Nov 2022
- 2022Geometry and convergence of natural policy gradient methods3 Nov 2022
- 2022Trust Region Policy Optimization with Optimal Transport Discrepancies: Duality and Algorithm for Continuous Actions20 Oct 2022
12 newest of 17 All 17 (JSON)