Convergence Theorems for Entropy-Regularized and Distributional Reinforcement Learning

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Jhaveri, Yash, Wiltzer, Harley, Shafto, Patrick, Bellemare, Marc G., Meger, David
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866916999805796352
author Jhaveri, Yash
Wiltzer, Harley
Shafto, Patrick
Bellemare, Marc G.
Meger, David
author_facet Jhaveri, Yash
Wiltzer, Harley
Shafto, Patrick
Bellemare, Marc G.
Meger, David
contents In the pursuit of finding an optimal policy, reinforcement learning (RL) methods generally ignore the properties of learned policies apart from their expected return. Thus, even when successful, it is difficult to characterize which policies will be learned and what they will do. In this work, we present a theoretical framework for policy optimization that guarantees convergence to a particular optimal policy, via vanishing entropy regularization and a temperature decoupling gambit. Our approach realizes an interpretable, diversity-preserving optimal policy as the regularization temperature vanishes and ensures the convergence of policy derived objects--value functions and return distributions. In a particular instance of our method, for example, the realized policy samples all optimal actions uniformly. Leveraging our temperature decoupling gambit, we present an algorithm that estimates, to arbitrary accuracy, the return distribution associated to its interpretable, diversity-preserving optimal policy.
format Preprint
id arxiv_https___arxiv_org_abs_2510_08526
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Convergence Theorems for Entropy-Regularized and Distributional Reinforcement Learning
Jhaveri, Yash
Wiltzer, Harley
Shafto, Patrick
Bellemare, Marc G.
Meger, David
Machine Learning
In the pursuit of finding an optimal policy, reinforcement learning (RL) methods generally ignore the properties of learned policies apart from their expected return. Thus, even when successful, it is difficult to characterize which policies will be learned and what they will do. In this work, we present a theoretical framework for policy optimization that guarantees convergence to a particular optimal policy, via vanishing entropy regularization and a temperature decoupling gambit. Our approach realizes an interpretable, diversity-preserving optimal policy as the regularization temperature vanishes and ensures the convergence of policy derived objects--value functions and return distributions. In a particular instance of our method, for example, the realized policy samples all optimal actions uniformly. Leveraging our temperature decoupling gambit, we present an algorithm that estimates, to arbitrary accuracy, the return distribution associated to its interpretable, diversity-preserving optimal policy.
title Convergence Theorems for Entropy-Regularized and Distributional Reinforcement Learning
topic Machine Learning
url https://arxiv.org/abs/2510.08526