A Fisher-Rao gradient flow for entropy-regularised Markov decision processes in Polish spaces
Not on the map. Why?
Abstract (arXiv)
We study the global convergence of a Fisher-Rao policy gradient flow for infinite-horizon entropy-regularised Markov decision processes with Polish state and action space. The flow is a continuous-time analogue of a policy mirror descent method. We establish the global well-posedness of the gradient flow and demonstrate its exponential convergence to the optimal policy. Moreover, we prove the flow is stable with respect to gradient evaluation, offering insights into the performance of a natural policy gradient flow with log-linear policy parameterisation. To overcome challenges stemming from the lack of the convexity of the objective function and the discontinuity arising from the entropy regulariser, we leverage the performance difference lemma and the duality relationship between the gradient and mirror descent flows. Our analysis provides a theoretical foundation for developing various discrete policy gradient algorithms.
Key facts. Cited by 23 papers in the map (data through 14 Oct 2026). Builds on 32 of its 60 references that are in the map. First posted on arXiv 4 Oct 2023.
Builds on
32 of 60 references.
References · 32
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
- 2015High-Dimensional Continuous Control Using Generalized Advantage Estimationcited by 2,026
- 2017Reinforcement Learning with Deep Energy-Based Policiescited by 669
- 2019A Theory of Regularized Markov Decision Processescited by 286
- 2020On the Global Convergence Rates of Softmax Policy Gradient Methodscited by 264
- 2018Global Convergence of Policy Gradient Methods for the Linear Quadratic Regulatorcited by 250
- 2019On the Theory of Policy Gradient Methods: Optimality, Approximation, and Distribution Shiftcited by 177
- 2020Fast Global Convergence of Natural Policy Gradient Methods with Entropy Regularizationcited by 176
- 2019Neural Policy Gradient Methods: Global Optimality and Rates of Convergencecited by 159
- 2021Policy Mirror Descent for Reinforcement Learning: Linear Convergence, New Sampling Complexity, and Generalized Problem Classescited by 156
- 2019Global Optimality Guarantees For Policy Gradient Methodscited by 150
- 2019Global Convergence of Policy Gradient Methods to (Almost) Locally Optimal Policiescited by 124
20 more
- 2022On the Convergence Rates of Policy Gradient Methodscited by 119
- 2020Mirror Descent Policy Optimizationcited by 106
- 2021Policy Mirror Descent for Regularized Reinforcement Learning: A Generalized Framework with Linear Convergencecited by 84
- 2023Stochastic Policy Gradient Methods: Improved Sample Complexity for Fisher-non-degenerate Policiescited by 58
- 2013The Information Geometry of Mirror Descentcited by 54
- 2022Linear Convergence of Natural Policy Gradient Methods with Log-Linear Policiescited by 46
- 2021Leveraging Non-uniformity in First-order Non-convex Optimizationcited by 44
- 2019LQR through the Lens of First Order Methods: Discrete-time Casecited by 40
- 2022Mirror Descent with Relative Smoothness in Measure Spaces, with application to Sinkhorn and EMcited by 34
- 2022Policy Optimization over General State and Action Spacescited by 23
- 2022Convergence of Policy Gradient for Entropy Regularized MDPs with Neural Network Approximation in the Mean-Field Regimecited by 21
- 2020Global optimality of softmax policy gradient with single hidden layer neural networks in the mean-field regimecited by 17
- 2021On the Sample Complexity and Metastability of Heavy-tailed Policy Search in Continuous Controlcited by 12
- 2022On the Hidden Biases of Policy Mirror Ascent in Continuous Action Spacescited by 11
- 2021Remarks on input to state stability of perturbed gradient flows, motivated by model-free feedback control learningcited by 8
- 2020Policy Gradient using Weak Derivatives for Reinforcement Learningcited by 7
- 2021Wasserstein Flow Meets Replicator Dynamics: A Mean-Field Analysis of Representation Learning in Actor-Criticcited by 7
- 2022Towards a Theoretical Foundation of Policy Optimization for Learning Control Policiescited by 6
- 2021Convergence and Optimality of Policy Gradient Methods in Weakly Smooth Settingscited by 5
- 2023Policy Gradient Converges to the Globally Optimal Policy for Nearly Linear-Quadratic Regulatorscited by 5
Built on by
Citing papers per year: 2024: 8; 2025: 5; 2026: 10.
23 papers.
Sort:
By area:
- 2024Fisher-Rao Gradient Flow: Geodesic Convexity and Functional Inequalitiescited by 13
- 2024Entropy annealing for policy mirror descent in continuous time and spacecited by 12
- 2024Stein transport for Bayesian inferencecited by 11
- 2024Fisher-Rao Gradient Flows of Linear Programs and State-Action Natural Policy Gradientscited by 5
- 2025PPO in the Fisher-Rao geometrycited by 4
- 2025Efficient Learning for Entropy-Regularized Markov Decision Processes via Multilevel Monte Carlocited by 3
- 2024Mirror Descent-Ascent for mean-field min-max problemscited by 2
- 2025Non-convex entropic mean-field optimization via Best Response flowcited by 2
- 2026A note on convergence of Wasserstein policy optimizationcited by 2
- 2024Optimal Rates of Convergence for Entropy Regularization in Discounted Markov Decision Processescited by 1
- 2024Convex Regularization and Convergence of Policy Gradient Flows under Safety Constraintscited by 1
- 2025Policy Optimization for Continuous-time Linear-Quadratic Graphon Mean Field Gamescited by 1
11 more
- 2025Convergence of an actor-critic gradient flow for entropy regularised MDPs in general spacescited by 1
- 2026Refined Analysis of Entropy-Regularized Actor-Criticcited by 1
- 2024Inference of Utilities and Time Preference in Sequential Decision-Makingcited by 0
- 2026Mirror descent actor-critic methods for entropy regularised MDPs in general spaces: stability and convergencecited by 0
- 2026On propagation of chaos for the Fisher-Rao gradient flow in entropic mean-field optimizationcited by 0
- 2026Global linear convergence of entropy-regularized softmax policy gradient beyond tabular MDPscited by 0
- 2026Policy Gradient for Continuous-Time Robust Markov Decision Processescited by 0
- 2026Randomized Optimal Switching Problem and Related Mirror Descent Flowcited by 0
- 2026Entropy Regularization Improves Policy Robustness in Continuous-Time Reinforcement Learningcited by 0
- 2026An analysis of Mirror-Descent Soft Actor-Criticcited by 0
- 2026Amortized Score-Hamiltonian Policy Iteration: A Grid-Free Scheme for Relaxed Stochastic Control Problemscited by 0
- 2026Amortized Score-Hamiltonian Policy Iteration: A Grid-Free Scheme for Relaxed Stochastic Control Problems3 Oct 2026
- 2026An analysis of Mirror-Descent Soft Actor-Critic28 Sep 2026
- 2026Entropy Regularization Improves Policy Robustness in Continuous-Time Reinforcement Learning3 Jul 2026
- 2026Randomized Optimal Switching Problem and Related Mirror Descent Flow11 Jun 2026
- 2026Policy Gradient for Continuous-Time Robust Markov Decision Processes3 Jun 2026
- 2026Global linear convergence of entropy-regularized softmax policy gradient beyond tabular MDPs24 May 2026
- 2026Refined Analysis of Entropy-Regularized Actor-Critic23 May 2026
- 2026A note on convergence of Wasserstein policy optimization21 May 2026
- 2026On propagation of chaos for the Fisher-Rao gradient flow in entropic mean-field optimization16 Feb 2026
- 2026Mirror descent actor-critic methods for entropy regularised MDPs in general spaces: stability and convergence11 Feb 2026
- 2025Convergence of an actor-critic gradient flow for entropy regularised MDPs in general spaces16 Oct 2025
- 2025Policy Optimization for Continuous-time Linear-Quadratic Graphon Mean Field Games6 Jun 2025
11 more
- 2025PPO in the Fisher-Rao geometry4 Jun 2025
- 2025Non-convex entropic mean-field optimization via Best Response flow28 May 2025
- 2025Efficient Learning for Entropy-Regularized Markov Decision Processes via Multilevel Monte Carlo27 Mar 2025
- 2024Convex Regularization and Convergence of Policy Gradient Flows under Safety Constraints28 Nov 2024
- 2024Stein transport for Bayesian inference2 Sep 2024
- 2024Fisher-Rao Gradient Flow: Geodesic Convexity and Functional Inequalities22 Jul 2024
- 2024Optimal Rates of Convergence for Entropy Regularization in Discounted Markov Decision Processes6 Jun 2024
- 2024Entropy annealing for policy mirror descent in continuous time and space30 May 2024
- 2024Inference of Utilities and Time Preference in Sequential Decision-Making24 May 2024
- 2024Fisher-Rao Gradient Flows of Linear Programs and State-Action Natural Policy Gradients28 Mar 2024
- 2024Mirror Descent-Ascent for mean-field min-max problems12 Feb 2024
- 2024Fisher-Rao Gradient Flows of Linear Programs and State-Action Natural Policy Gradientscited by 5
- 2025PPO in the Fisher-Rao geometrycited by 4
- 2026A note on convergence of Wasserstein policy optimizationcited by 2
- 2024Optimal Rates of Convergence for Entropy Regularization in Discounted Markov Decision Processescited by 1
- 2024Convex Regularization and Convergence of Policy Gradient Flows under Safety Constraintscited by 1
- 2025Policy Optimization for Continuous-time Linear-Quadratic Graphon Mean Field Gamescited by 1
- 2025Convergence of an actor-critic gradient flow for entropy regularised MDPs in general spacescited by 1
- 2026Refined Analysis of Entropy-Regularized Actor-Criticcited by 1
- 2026Mirror descent actor-critic methods for entropy regularised MDPs in general spaces: stability and convergencecited by 0
- 2026Global linear convergence of entropy-regularized softmax policy gradient beyond tabular MDPscited by 0
- 2026Policy Gradient for Continuous-Time Robust Markov Decision Processescited by 0
- 2026Randomized Optimal Switching Problem and Related Mirror Descent Flowcited by 0
12 most cited of 14 All 14 (JSON)
- 2026An analysis of Mirror-Descent Soft Actor-Critic28 Sep 2026
- 2026Entropy Regularization Improves Policy Robustness in Continuous-Time Reinforcement Learning3 Jul 2026
- 2026Randomized Optimal Switching Problem and Related Mirror Descent Flow11 Jun 2026
- 2026Policy Gradient for Continuous-Time Robust Markov Decision Processes3 Jun 2026
- 2026Global linear convergence of entropy-regularized softmax policy gradient beyond tabular MDPs24 May 2026
- 2026Refined Analysis of Entropy-Regularized Actor-Critic23 May 2026
- 2026A note on convergence of Wasserstein policy optimization21 May 2026
- 2026Mirror descent actor-critic methods for entropy regularised MDPs in general spaces: stability and convergence11 Feb 2026
- 2025Convergence of an actor-critic gradient flow for entropy regularised MDPs in general spaces16 Oct 2025
- 2025Policy Optimization for Continuous-time Linear-Quadratic Graphon Mean Field Games6 Jun 2025
- 2025PPO in the Fisher-Rao geometry4 Jun 2025
- 2024Convex Regularization and Convergence of Policy Gradient Flows under Safety Constraints28 Nov 2024
12 newest of 14 All 14 (JSON)
- 2024Fisher-Rao Gradient Flow: Geodesic Convexity and Functional Inequalitiescited by 13
- 2024Entropy annealing for policy mirror descent in continuous time and spacecited by 12
- 2024Stein transport for Bayesian inferencecited by 11
- 2025Efficient Learning for Entropy-Regularized Markov Decision Processes via Multilevel Monte Carlocited by 3
- 2024Mirror Descent-Ascent for mean-field min-max problemscited by 2
- 2025Non-convex entropic mean-field optimization via Best Response flowcited by 2
- 2024Inference of Utilities and Time Preference in Sequential Decision-Makingcited by 0
- 2026On propagation of chaos for the Fisher-Rao gradient flow in entropic mean-field optimizationcited by 0
- 2026Amortized Score-Hamiltonian Policy Iteration: A Grid-Free Scheme for Relaxed Stochastic Control Problemscited by 0
- 2026Amortized Score-Hamiltonian Policy Iteration: A Grid-Free Scheme for Relaxed Stochastic Control Problems3 Oct 2026
- 2026On propagation of chaos for the Fisher-Rao gradient flow in entropic mean-field optimization16 Feb 2026
- 2025Non-convex entropic mean-field optimization via Best Response flow28 May 2025
- 2025Efficient Learning for Entropy-Regularized Markov Decision Processes via Multilevel Monte Carlo27 Mar 2025
- 2024Stein transport for Bayesian inference2 Sep 2024
- 2024Fisher-Rao Gradient Flow: Geodesic Convexity and Functional Inequalities22 Jul 2024
- 2024Entropy annealing for policy mirror descent in continuous time and space30 May 2024
- 2024Inference of Utilities and Time Preference in Sequential Decision-Making24 May 2024
- 2024Mirror Descent-Ascent for mean-field min-max problems12 Feb 2024