Greedification Operators for Policy Optimization: Investigating Forward and Reverse KL Divergences
Not on the map. Why?
Abstract (arXiv)
Approximate Policy Iteration (API) algorithms alternate between (approximate) policy evaluation and (approximate) greedification. Many different approaches have been explored for approximate policy evaluation, but less is understood about approximate greedification and what choices guarantee policy improvement. In this work, we investigate approximate greedification when reducing the KL divergence between the parameterized policy and the Boltzmann distribution over action values. In particular, we investigate the difference between the forward and reverse KL divergences, with varying degrees of entropy regularization. We show that the reverse KL has stronger policy improvement guarantees, but that reducing the forward KL can result in a worse policy. We also demonstrate, however, that a large enough reduction of the forward KL can induce improvement under additional assumptions. Empirically, we show on simple continuous-action environments that the forward KL can induce more exploration, but at the cost of a more suboptimal policy. No significant differences were observed in the discrete-action setting or on a suite of benchmark problems. Throughout, we highlight that many policy gradient methods can be seen as an instance of API, with either the forward or reverse KL for the policy update, and discuss next steps for understanding and improving our policy optimization algorithms.
Key facts. Cited by 40 papers in the map (data through 7 Oct 2026). Builds on 44 of its 87 references that are in the map. First posted on arXiv 17 Jul 2021.
Builds on
44 of 87 references.
References · 44
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
- 2015Continuous control with deep reinforcement learningcited by 3,279
- 2015Trust Region Policy Optimizationcited by 2,503
- 2016Asynchronous Methods for Deep Reinforcement Learningcited by 2,218
- 2016Generative Adversarial Imitation Learningcited by 1,576
- 2018QT-Opt: Scalable Deep Reinforcement Learning for Vision-Based Robotic Manipulationcited by 730
- 2017Reinforcement Learning with Deep Energy-Based Policiescited by 659
- 2017Learning Robust Rewards with Adversarial Inverse Reinforcement Learningcited by 485
- 2018Reinforcement Learning and Control as Probabilistic Inference: Tutorial and Reviewcited by 466
- 2016Input Convex Neural Networkscited by 407
32 more
- 2019Online Learning: A Modern Introduction Using Convex Optimizationcited by 407
- 2018Maximum a Posteriori Policy Optimisationcited by 318
- 2019A Theory of Regularized Markov Decision Processescited by 279
- 2020On the Global Convergence Rates of Softmax Policy Gradient Methodscited by 258
- 2019A Divergence Minimization Perspective on Imitation Learning Methodscited by 188
- 2019AlgaeDICE: Policy Gradient from Arbitrary Experiencecited by 187
- 2017Bridging the Gap Between Value and Policy Based Reinforcement Learningcited by 178
- 2017A unified view of entropy-regularized Markov decision processescited by 174
- 2017Equivalence Between Policy Gradients and Soft Q-Learningcited by 172
- 2018Understanding the impact of entropy on policy optimizationcited by 171
- 2016Continuous Deep Q-Learning with Model-based Accelerationcited by 149
- 2021Policy Mirror Descent for Reinforcement Learning: Linear Convergence, New Sampling Complexity, and Generalized Problem Classescited by 149
- 2019Global Optimality Guarantees For Policy Gradient Methodscited by 146
- 2019Adaptive Trust Region Policy Optimization: Global Convergence and Faster Rates for Regularized MDPscited by 113
- 2016Reward Augmented Maximum Likelihood for Neural Structured Predictioncited by 80
- 2017Time Limits in Reinforcement Learningcited by 72
- 2019Is the Policy Gradient a Gradient?cited by 40
- 2014Approximate Policy Iteration Schemes: A Comparisoncited by 39
- 2019Learning to Generalize from Sparse and Underspecified Rewardscited by 39
- 2019Improving Exploration in Soft-Actor-Critic with Normalizing Flows Policiescited by 37
- 2018An Off-policy Policy Gradient Theorem Using Emphatic Weightingscited by 34
- 2019The Value Function Polytope in Reinforcement Learningcited by 29
- 2021Variational Refinement for Importance Sampling Using the Forward Kullback-Leibler Divergencecited by 29
- 2016Combining policy gradient and Q-learningcited by 23
- 2019CAQL: Continuous Action Q-Learningcited by 22
- 2019MinAtar: An Atari-Inspired Testbed for Thorough and Reproducible Reinforcement Learning Experimentscited by 17
- 2020An operator view of policy gradient methodscited by 17
- 2019Entropic Regularization of Markov Decision Processescited by 15
- 2016Improving Policy Gradient by Exploring Under-appreciated Rewardscited by 14
- 2019Deep Conservative Policy Iterationcited by 14
- 2014Classification-based Approximate Policy Iteration: Experiments and Extended Discussionscited by 3
- 2020Ensuring Monotonic Policy Improvement in Entropy-regularized Value-based Reinforcement Learningcited by 3
Built on by
Citing papers per year: 2022: 4; 2023: 4; 2024: 11; 2025: 11; 2026: 10.
40 papers.
Sort:
By area:
- 2025Test-time Alignment of Diffusion Models without Reward Over-optimizationcited by 89
- 2025Retaining by Doing: The Role of On-Policy Data in Mitigating Forgettingcited by 82
- 2024Rethinking Kullback-Leibler Divergence in Knowledge Distillation for Large Language Modelscited by 60
- 2024Towards Efficient Exact Optimization of Language Model Alignmentcited by 37
- 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixescited by 37
- 2023Seizing Serendipity: Exploiting the Value of Past Success in Off-Policy Actor-Criticcited by 20
- 2022Critic Sequential Monte Carlocited by 16
- 2024BiLD: Bi-directional Logits Difference Loss for Large Language Model Distillationcited by 16
- 2025Reevaluating Policy Gradient Methods for Imperfect-Information Gamescited by 13
- 2023Iteratively Refined Behavior Regularization for Offline Reinforcement Learningcited by 10
- 2024Offline-Boosted Actor-Critic: Adaptively Blending Optimal Historical Behaviors in Deep Off-Policy RLcited by 10
- 2024Variational Schr"odinger Diffusion Modelscited by 9
28 more
- 2022HesScale: Scalable Computation of Hessian Diagonalscited by 8
- 2023Language Model Decoding as Direct Metrics Optimizationcited by 8
- 2025Reinforcement Learning with Discrete Diffusion Policies for Combinatorial Action Spacescited by 7
- 2022The Geometry of Robust Value Functionscited by 6
- 2024AdvNF: Reducing Mode Collapse in Conditional Normalising Flows using Adversarial Learningcited by 6
- 2024Value Improved Actor Critic Algorithmscited by 6
- 2024Revisiting Scalable Hessian Diagonal Approximations for Applications in Reinforcement Learningcited by 5
- 2025Trust-Region Twisted Policy Improvementcited by 5
- 2026Diversity-Aware Reverse Kullback-Leibler Divergence for Large Language Model Distillationcited by 5
- 2022TD3 with Reverse KL Regularizer for Offline Reinforcement Learning from Mixed Datasetscited by 4
- 2024OMPO: A Unified Framework for RL under Policy and Dynamics Shiftscited by 4
- 2025Twice Sequential Monte Carlo for Tree Searchcited by 3
- 2025Diffusion Alignment as Variational Expectation-Maximizationcited by 2
- 2023Generalized Munchausen Reinforcement Learning using Tsallis KL Divergencecited by 1
- 2024Policy Bifurcation in Safe Reinforcement Learningcited by 1
- 2025Bidirectional Soft Actor-Critic: Leveraging Forward and Reverse KL Divergence for Efficient Reinforcement Learningcited by 1
- 2025Lifelong Learning with Behavior Consolidation for Vehicle Routingcited by 1
- 2025Coupled Variational Reinforcement Learning for Language Model General Reasoningcited by 1
- 2026Value Bonuses using Ensemble Errors for Exploration in Reinforcement Learningcited by 1
- 2026Consistently Informative Soft-Label Temperature for Knowledge Distillationcited by 1
- 2024Shared learning of powertrain control policies for vehicle fleetscited by 0
- 2025Symmetric Behavior Regularized Policy Optimizationcited by 0
- 2026Proximal Policy Optimization for Amortized Discrete Samplingcited by 0
- 2026Flout at Your Own Risk: LLMs Struggle with Pragmatic Cooperativity Under Epistemic Asymmetrycited by 0
- 2026Deconstructing Actor-Critic: A Large-scale Empirical Study of Design Components for Practitionerscited by 0
- 2026Policy Optimality Measurement for Multi-Vehicle Decision-Making: From Extrinsic Indicators to Intrinsic Qualitycited by 0
- 2026Foundations of Reinforcement Learning and Control:Connections and New Perspectivescited by 0
- 2026Towards Optimal Policy Improvementcited by 0
- 2026Towards Optimal Policy Improvement1 Oct 2026
- 2026Foundations of Reinforcement Learning and Control:Connections and New Perspectives3 Aug 2026
- 2026Policy Optimality Measurement for Multi-Vehicle Decision-Making: From Extrinsic Indicators to Intrinsic Quality2 Aug 2026
- 2026Deconstructing Actor-Critic: A Large-scale Empirical Study of Design Components for Practitioners14 Jul 2026
- 2026Flout at Your Own Risk: LLMs Struggle with Pragmatic Cooperativity Under Epistemic Asymmetry13 Jul 2026
- 2026Proximal Policy Optimization for Amortized Discrete Sampling14 Jun 2026
- 2026Consistently Informative Soft-Label Temperature for Knowledge Distillation19 May 2026
- 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes11 May 2026
- 2026Diversity-Aware Reverse Kullback-Leibler Divergence for Large Language Model Distillation31 Mar 2026
- 2026Value Bonuses using Ensemble Errors for Exploration in Reinforcement Learning12 Feb 2026
- 2025Coupled Variational Reinforcement Learning for Language Model General Reasoning14 Dec 2025
- 2025Twice Sequential Monte Carlo for Tree Search18 Nov 2025
28 more
- 2025Retaining by Doing: The Role of On-Policy Data in Mitigating Forgetting21 Oct 2025
- 2025Diffusion Alignment as Variational Expectation-Maximization1 Oct 2025
- 2025Reinforcement Learning with Discrete Diffusion Policies for Combinatorial Action Spaces26 Sep 2025
- 2025Lifelong Learning with Behavior Consolidation for Vehicle Routing26 Sep 2025
- 2025Symmetric Behavior Regularized Policy Optimization6 Aug 2025
- 2025Bidirectional Soft Actor-Critic: Leveraging Forward and Reverse KL Divergence for Efficient Reinforcement Learning2 Jun 2025
- 2025Trust-Region Twisted Policy Improvement8 Apr 2025
- 2025Reevaluating Policy Gradient Methods for Imperfect-Information Games13 Feb 2025
- 2025Test-time Alignment of Diffusion Models without Reward Over-optimization10 Jan 2025
- 2024BiLD: Bi-directional Logits Difference Loss for Large Language Model Distillation19 Jun 2024
- 2024Revisiting Scalable Hessian Diagonal Approximations for Applications in Reinforcement Learning5 Jun 2024
- 2024Value Improved Actor Critic Algorithms3 Jun 2024
- 2024OMPO: A Unified Framework for RL under Policy and Dynamics Shifts29 May 2024
- 2024Offline-Boosted Actor-Critic: Adaptively Blending Optimal Historical Behaviors in Deep Off-Policy RL28 May 2024
- 2024Variational Schr"odinger Diffusion Models8 May 2024
- 2024Shared learning of powertrain control policies for vehicle fleets27 Apr 2024
- 2024Rethinking Kullback-Leibler Divergence in Knowledge Distillation for Large Language Models3 Apr 2024
- 2024Policy Bifurcation in Safe Reinforcement Learning19 Mar 2024
- 2024Towards Efficient Exact Optimization of Language Model Alignment1 Feb 2024
- 2024AdvNF: Reducing Mode Collapse in Conditional Normalising Flows using Adversarial Learning29 Jan 2024
- 2023Language Model Decoding as Direct Metrics Optimization2 Oct 2023
- 2023Iteratively Refined Behavior Regularization for Offline Reinforcement Learning9 Jun 2023
- 2023Seizing Serendipity: Exploiting the Value of Past Success in Off-Policy Actor-Critic5 Jun 2023
- 2023Generalized Munchausen Reinforcement Learning using Tsallis KL Divergence27 Jan 2023
- 2022TD3 with Reverse KL Regularizer for Offline Reinforcement Learning from Mixed Datasets5 Dec 2022
- 2022HesScale: Scalable Computation of Hessian Diagonals20 Oct 2022
- 2022Critic Sequential Monte Carlo30 May 2022
- 2022The Geometry of Robust Value Functions30 Jan 2022
- 2025Test-time Alignment of Diffusion Models without Reward Over-optimizationcited by 89
- 2024Variational Schr"odinger Diffusion Modelscited by 9
- 2025Reinforcement Learning with Discrete Diffusion Policies for Combinatorial Action Spacescited by 7
- 2025Diffusion Alignment as Variational Expectation-Maximizationcited by 2
- 2025Diffusion Alignment as Variational Expectation-Maximization1 Oct 2025
- 2025Reinforcement Learning with Discrete Diffusion Policies for Combinatorial Action Spaces26 Sep 2025
- 2025Test-time Alignment of Diffusion Models without Reward Over-optimization10 Jan 2025
- 2024Variational Schr"odinger Diffusion Models8 May 2024
- 2025Retaining by Doing: The Role of On-Policy Data in Mitigating Forgettingcited by 82
- 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixescited by 37
- 2023Seizing Serendipity: Exploiting the Value of Past Success in Off-Policy Actor-Criticcited by 20
- 2022Critic Sequential Monte Carlocited by 16
- 2024BiLD: Bi-directional Logits Difference Loss for Large Language Model Distillationcited by 16
- 2025Reevaluating Policy Gradient Methods for Imperfect-Information Gamescited by 13
- 2024Offline-Boosted Actor-Critic: Adaptively Blending Optimal Historical Behaviors in Deep Off-Policy RLcited by 10
- 2022HesScale: Scalable Computation of Hessian Diagonalscited by 8
- 2023Language Model Decoding as Direct Metrics Optimizationcited by 8
- 2022The Geometry of Robust Value Functionscited by 6
- 2024AdvNF: Reducing Mode Collapse in Conditional Normalising Flows using Adversarial Learningcited by 6
- 2024Value Improved Actor Critic Algorithmscited by 6
12 most cited of 30 All 30 (JSON)
- 2026Towards Optimal Policy Improvement1 Oct 2026
- 2026Foundations of Reinforcement Learning and Control:Connections and New Perspectives3 Aug 2026
- 2026Policy Optimality Measurement for Multi-Vehicle Decision-Making: From Extrinsic Indicators to Intrinsic Quality2 Aug 2026
- 2026Deconstructing Actor-Critic: A Large-scale Empirical Study of Design Components for Practitioners14 Jul 2026
- 2026Flout at Your Own Risk: LLMs Struggle with Pragmatic Cooperativity Under Epistemic Asymmetry13 Jul 2026
- 2026Proximal Policy Optimization for Amortized Discrete Sampling14 Jun 2026
- 2026Consistently Informative Soft-Label Temperature for Knowledge Distillation19 May 2026
- 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes11 May 2026
- 2026Diversity-Aware Reverse Kullback-Leibler Divergence for Large Language Model Distillation31 Mar 2026
- 2026Value Bonuses using Ensemble Errors for Exploration in Reinforcement Learning12 Feb 2026
- 2025Twice Sequential Monte Carlo for Tree Search18 Nov 2025
- 2025Retaining by Doing: The Role of On-Policy Data in Mitigating Forgetting21 Oct 2025
12 newest of 30 All 30 (JSON)