Entropy-Preserving Reinforcement Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Petrenko, Aleksei, Lipkin, Ben, Chen, Kevin, Wijmans, Erik, Cusumano-Towner, Marco, Giryes, Raja, Krähenbühl, Philipp
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912963171975168
author Petrenko, Aleksei
Lipkin, Ben
Chen, Kevin
Wijmans, Erik
Cusumano-Towner, Marco
Giryes, Raja
Krähenbühl, Philipp
author_facet Petrenko, Aleksei
Lipkin, Ben
Chen, Kevin
Wijmans, Erik
Cusumano-Towner, Marco
Giryes, Raja
Krähenbühl, Philipp
contents Policy gradient algorithms have driven many recent advancements in language model reasoning. An appealing property is their ability to learn from exploration on their own trajectories, a process crucial for fostering diverse and creative solutions. As we show in this paper, many policy gradient algorithms naturally reduce the entropy -- and thus the diversity of explored trajectories -- as part of training, yielding a policy increasingly limited in its ability to explore. In this paper, we argue that entropy should be actively monitored and controlled throughout training. We formally analyze the contributions of leading policy gradient objectives on entropy dynamics, identify empirical factors (such as numerical precision) that significantly impact entropy behavior, and propose explicit mechanisms for entropy control. These include REPO, a family of algorithms that modify the advantage function to regulate entropy, and ADAPO, an adaptive asymmetric clipping approach. Models trained with our entropy-preserving methods maintain diversity throughout training, yielding final policies that are more performant and retain their trainability for sequential learning in new environments.
format Preprint
id arxiv_https___arxiv_org_abs_2603_11682
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Entropy-Preserving Reinforcement Learning
Petrenko, Aleksei
Lipkin, Ben
Chen, Kevin
Wijmans, Erik
Cusumano-Towner, Marco
Giryes, Raja
Krähenbühl, Philipp
Machine Learning
Artificial Intelligence
Policy gradient algorithms have driven many recent advancements in language model reasoning. An appealing property is their ability to learn from exploration on their own trajectories, a process crucial for fostering diverse and creative solutions. As we show in this paper, many policy gradient algorithms naturally reduce the entropy -- and thus the diversity of explored trajectories -- as part of training, yielding a policy increasingly limited in its ability to explore. In this paper, we argue that entropy should be actively monitored and controlled throughout training. We formally analyze the contributions of leading policy gradient objectives on entropy dynamics, identify empirical factors (such as numerical precision) that significantly impact entropy behavior, and propose explicit mechanisms for entropy control. These include REPO, a family of algorithms that modify the advantage function to regulate entropy, and ADAPO, an adaptive asymmetric clipping approach. Models trained with our entropy-preserving methods maintain diversity throughout training, yielding final policies that are more performant and retain their trainability for sequential learning in new environments.
title Entropy-Preserving Reinforcement Learning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2603.11682