arXiv · 2502.11033
Convergence of Policy Mirror Descent Beyond Compatible Function Approximation
Abstract
Modern policy optimization methods roughly follow the policy mirror descent (PMD) algorithmic template, for which there are by now numerous theoretical convergence results. However, most of these either target tabular environments, or can be applied effectively only when the class of policies being optimized over satisfies strong closure conditions, which is typically not the case when working with parametric policy classes in large-scale environments. In this work, we develop a theoretical framework for PMD for general policy classes where we replace the closure conditions with a strictly weaker variational gradient dominance assumption, and obtain upper bounds on the rate of convergence to the best-in-class policy. Our main result leverages a novel notion of smoothness with respect to a local norm induced by the occupancy measure of the current policy, and casts PMD as a particular instance of smooth non-convex optimization in non-Euclidean space.
Explore related subjects
Keep this discovery
Uri Sherman, Tomer Koren, Yishay Mansour. 2025-02-16. Convergence of Policy Mirror Descent Beyond Compatible Function Approximation. https://arxiv.org/abs/2502.11033
Cite the original work for its findings. Save a collection to share your selection of sources.