State Augmented Constrained Reinforcement Learning: Overcoming the Limitations of Learning with Rewards

Miguel Calvo-Fullana,Santiago Paternain,Luiz F. O. Chamon,Alejandro Ribeiro
2023-09-21
Abstract:A common formulation of constrained reinforcement learning involves multiple rewards that must individually accumulate to given thresholds. In this class of problems, we show a simple example in which the desired optimal policy cannot be induced by any weighted linear combination of rewards. Hence, there exist constrained reinforcement learning problems for which neither regularized nor classical primal-dual methods yield optimal policies. This work addresses this shortcoming by augmenting the state with Lagrange multipliers and reinterpreting primal-dual methods as the portion of the dynamics that drives the multipliers evolution. This approach provides a systematic state augmentation procedure that is guaranteed to solve reinforcement learning problems with constraints. Thus, as we illustrate by an example, while previous methods can fail at finding optimal policies, running the dual dynamics while executing the augmented policy yields an algorithm that provably samples actions from the optimal policy.
Machine Learning,Robotics,Optimization and Control
What problem does this paper attempt to address?