Learning to Score Behaviors for Guided Policy Optimization
Not on the map. Why?
Abstract (arXiv)
We introduce a new approach for comparing reinforcement learning policies, using Wasserstein distances (WDs) in a newly defined latent behavioral space. We show that by utilizing the dual formulation of the WD, we can learn score functions over policy behaviors that can in turn be used to lead policy optimization towards (or away from) (un)desired behaviors. Combined with smoothed WDs, the dual formulation allows us to devise efficient algorithms that take stochastic gradient descent steps through WD regularizers. We incorporate these regularizers into two novel on-policy algorithms, Behavior-Guided Policy Gradient and Behavior-Guided Evolution Strategies, which we demonstrate can outperform existing methods in a variety of challenging environments. We also provide an open source demo.
Key facts. Cited by 34 papers in the map (data through 14 Oct 2026). First posted on arXiv 11 Jun 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 8; 2022: 7; 2023: 8; 2024: 5; 2025: 2; 2026: 3.
34 papers.
Sort:
By area:
- 2020Primal Wasserstein Imitation Learningcited by 121
- 2021Dueling RL: Reinforcement Learning with Trajectory Preferencescited by 106
- 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
- 2021Differentiable Trust Region Layers for Deep Reinforcement Learningcited by 25
- 2023Provable Reward-Agnostic Preference-Based Reinforcement Learningcited by 25
- 2021Wasserstein Unsupervised Reinforcement Learningcited by 22
- 2023ReLOAD: Reinforcement Learning with Optimistic Ascent-Descent for Last-Iterate Convergence in Constrained MDPscited by 20
- 2022On Pathologies in KL-Regularized Reinforcement Learning from Expert Demonstrationscited by 19
- 2024Group-Aware Coordination Graph for Multi-Agent Reinforcement Learningcited by 18
- 2025Wasserstein Policy Optimizationcited by 13
22 more
- 2022Learning General World Models in a Handful of Reward-Free Deploymentscited 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
- 2024Inferring Latent Temporal Sparse Coordination Graph for Multi-Agent Reinforcement Learningcited by 7
- 2023Provably Convergent Policy Optimization via Metric-aware Trust Region Methodscited by 6
- 2024A Behavior-Aware Approach for Deep Reinforcement Learning in Non-stationary Environments without Known Change Pointscited by 5
- 2024Optimal Transport-Assisted Risk-Sensitive Q-Learningcited by 4
- 2026Wasserstein Proximal Policy Gradientcited by 4
- 2022Truly Deterministic Policy Optimizationcited by 3
- 2022Towards A Unified Policy Abstraction Theory and Representation Learning Approach in Markov Decision Processescited by 3
- 2023Iteratively Learn Diverse Strategies with State Distance Informationcited by 3
- 2023A Conservative Approach for Few-Shot Transfer in Off-Dynamics Reinforcement Learningcited by 3
- 2025AutoQD: Automatic Discovery of Diverse Behaviors with Quality-Diversity Optimizationcited by 3
- 2021Distributionally-Constrained Policy Optimization via Unbalanced Optimal Transportcited by 1
- 2021Agent Spacescited by 1
- 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
- 2024Phasic Diversity Optimization for Population-Based Reinforcement Learningcited by 0
- 2026Wasserstein Formulation of Reinforcement Learning. An Optimal Transport Perspective on Policy Optimizationcited by 0
- 2026Implicit Neural Representations of Individual Behaviorcited by 0
- 2026Implicit Neural Representations of Individual Behavior10 Jun 2026
- 2026Wasserstein Formulation of Reinforcement Learning. An Optimal Transport Perspective on Policy Optimization16 Apr 2026
- 2026Wasserstein Proximal Policy Gradient3 Mar 2026
- 2025AutoQD: Automatic Discovery of Diverse Behaviors with Quality-Diversity Optimization5 Jun 2025
- 2025Wasserstein Policy Optimization1 May 2025
- 2024Optimal Transport-Assisted Risk-Sensitive Q-Learning17 Jun 2024
- 2024A Behavior-Aware Approach for Deep Reinforcement Learning in Non-stationary Environments without Known Change Points23 May 2024
- 2024Group-Aware Coordination Graph for Multi-Agent Reinforcement Learning17 Apr 2024
- 2024Inferring Latent Temporal Sparse Coordination Graph for Multi-Agent Reinforcement Learning28 Mar 2024
- 2024Phasic Diversity Optimization for Population-Based Reinforcement Learning17 Mar 2024
- 2023A Conservative Approach for Few-Shot Transfer in Off-Dynamics Reinforcement Learning24 Dec 2023
- 2023Iteratively Learn Diverse Strategies with State Distance Information23 Oct 2023
22 more
- 2023Confronting Reward Model Overoptimization with Constrained RLHF6 Oct 2023
- 2023Wasserstein Diversity-Enriched Regularizer for Hierarchical Reinforcement Learning2 Aug 2023
- 2023Provably Convergent Policy Optimization via Metric-aware Trust Region Methods25 Jun 2023
- 2023Provable Reward-Agnostic Preference-Based Reinforcement Learning29 May 2023
- 2023Wasserstein Gradient Flows for Optimizing Gaussian Mixture Policies17 May 2023
- 2023ReLOAD: Reinforcement Learning with Optimistic Ascent-Descent for Last-Iterate Convergence in Constrained MDPs2 Feb 2023
- 2022On Pathologies in KL-Regularized Reinforcement Learning from Expert Demonstrations28 Dec 2022
- 2022Transfer RL via the Undo Maps Formalism26 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
- 2022Towards A Unified Policy Abstraction Theory and Representation Learning Approach in Markov Decision Processes16 Sep 2022
- 2022Minimum Description Length Control17 Jul 2022
- 2022Truly Deterministic Policy Optimization30 May 2022
- 2021Agent Spaces11 Nov 2021
- 2021Dueling RL: Reinforcement Learning with Trajectory Preferences8 Nov 2021
- 2021Towards an Understanding of Default Policies in Multitask Policy Optimization4 Nov 2021
- 2021Wasserstein Unsupervised Reinforcement Learning15 Oct 2021
- 2021MICo: Improved representations via sampling-based state similarity for Markov decision processes3 Jun 2021
- 2021Distributionally-Constrained Policy Optimization via Unbalanced Optimal Transport15 Feb 2021
- 2021Tactical Optimism and Pessimism for Deep Reinforcement Learning7 Feb 2021
- 2021Differentiable Trust Region Layers for Deep Reinforcement Learning22 Jan 2021
- 2020Primal Wasserstein Imitation Learning8 Jun 2020
- 2025Wasserstein Policy Optimizationcited by 13
- 2024A Behavior-Aware Approach for Deep Reinforcement Learning in Non-stationary Environments without Known Change Pointscited by 5
- 2024Optimal Transport-Assisted Risk-Sensitive Q-Learningcited by 4
- 2026Wasserstein Proximal Policy Gradientcited by 4
- 2024Phasic Diversity Optimization for Population-Based Reinforcement Learningcited by 0
- 2026Wasserstein Formulation of Reinforcement Learning. An Optimal Transport Perspective on Policy Optimizationcited by 0
- 2026Wasserstein Formulation of Reinforcement Learning. An Optimal Transport Perspective on Policy Optimization16 Apr 2026
- 2026Wasserstein Proximal Policy Gradient3 Mar 2026
- 2025Wasserstein Policy Optimization1 May 2025
- 2024Optimal Transport-Assisted Risk-Sensitive Q-Learning17 Jun 2024
- 2024A Behavior-Aware Approach for Deep Reinforcement Learning in Non-stationary Environments without Known Change Points23 May 2024
- 2024Phasic Diversity Optimization for Population-Based Reinforcement Learning17 Mar 2024
- 2020Primal Wasserstein Imitation Learningcited by 121
- 2020Primal Wasserstein Imitation Learning8 Jun 2020
- 2021Dueling RL: Reinforcement Learning with Trajectory Preferencescited by 106
- 2021Tactical Optimism and Pessimism for Deep Reinforcement Learningcited by 55
- 2021MICo: Improved representations via sampling-based state similarity for Markov decision processescited by 43
- 2021Differentiable Trust Region Layers for Deep Reinforcement Learningcited by 25
- 2021Wasserstein Unsupervised Reinforcement Learningcited by 22
- 2023ReLOAD: Reinforcement Learning with Optimistic Ascent-Descent for Last-Iterate Convergence in Constrained MDPscited by 20
- 2022On Pathologies in KL-Regularized Reinforcement Learning from Expert Demonstrationscited by 19
- 2024Group-Aware Coordination Graph for Multi-Agent Reinforcement Learningcited by 18
- 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
- 2024Inferring Latent Temporal Sparse Coordination Graph for Multi-Agent Reinforcement Learningcited by 7
12 most cited of 24 All 24 (JSON)
- 2026Implicit Neural Representations of Individual Behavior10 Jun 2026
- 2025AutoQD: Automatic Discovery of Diverse Behaviors with Quality-Diversity Optimization5 Jun 2025
- 2024Group-Aware Coordination Graph for Multi-Agent Reinforcement Learning17 Apr 2024
- 2024Inferring Latent Temporal Sparse Coordination Graph for Multi-Agent Reinforcement Learning28 Mar 2024
- 2023A Conservative Approach for Few-Shot Transfer in Off-Dynamics Reinforcement Learning24 Dec 2023
- 2023Iteratively Learn Diverse Strategies with State Distance Information23 Oct 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
- 2023ReLOAD: Reinforcement Learning with Optimistic Ascent-Descent for Last-Iterate Convergence in Constrained MDPs2 Feb 2023
- 2022On Pathologies in KL-Regularized Reinforcement Learning from Expert Demonstrations28 Dec 2022
- 2022Transfer RL via the Undo Maps Formalism26 Nov 2022
12 newest of 24 All 24 (JSON)