TY - GEN
T1 - State Entropy Regularization for Robust Reinforcement Learning
AU - Ashlag, Yonatan
AU - Koren, Uri
AU - Mutti, Mirco
AU - Derman, Esther
AU - Bacon, Pierre Luc
AU - Mannor, Shie
N1 - Publisher Copyright: © 2025 Neural information processing systems foundation. All rights reserved.
PY - 2025
Y1 - 2025
N2 - State entropy regularization has empirically shown better exploration and sample complexity in reinforcement learning (RL). However, its theoretical guarantees have not been studied. In this paper, we show that state entropy regularization improves robustness to structured and spatially correlated perturbations. These types of variation are common in transfer learning but often overlooked by standard robust RL methods, which typically focus on small, uncorrelated changes. We provide a comprehensive characterization of these robustness properties, including formal guarantees under reward and transition uncertainty, as well as settings where the method performs poorly. Much of our analysis contrasts state entropy with the widely used policy entropy regularization, highlighting their different benefits. Finally, from a practical standpoint, we illustrate that compared with policy entropy, the robustness advantages of state entropy are more sensitive to the number of rollouts used for policy evaluation.
AB - State entropy regularization has empirically shown better exploration and sample complexity in reinforcement learning (RL). However, its theoretical guarantees have not been studied. In this paper, we show that state entropy regularization improves robustness to structured and spatially correlated perturbations. These types of variation are common in transfer learning but often overlooked by standard robust RL methods, which typically focus on small, uncorrelated changes. We provide a comprehensive characterization of these robustness properties, including formal guarantees under reward and transition uncertainty, as well as settings where the method performs poorly. Much of our analysis contrasts state entropy with the widely used policy entropy regularization, highlighting their different benefits. Finally, from a practical standpoint, we illustrate that compared with policy entropy, the robustness advantages of state entropy are more sensitive to the number of rollouts used for policy evaluation.
UR - https://www.scopus.com/pages/publications/105048597242
M3 - Conference contribution
T3 - Advances in Neural Information Processing Systems
SP - 23814
EP - 23842
BT - 39th Conference on Neural Information Processing Systems, NeurIPS 2025
T2 - 39th Conference on Neural Information Processing Systems, NeurIPS 2025
Y2 - 2 December 2025 through 7 December 2025
ER -