Multi-Objective Reinforcement Learning with Max-Min Criterion: A Game-Theoretic Approach
Woohyeon Byeon, Giseung Park, Jongseong Chae, Amir Leshem, Youngchul Sung
TL;DR
This work addresses MORL under a max-min fairness criterion by recasting the problem as a two-player, zero-sum, entropy-regularized game and solving it with single-loop updates. The proposed ERAM and ARAM algorithms combine mirror-descent/NPG updates for the policy with closed-form updates for the adversarial weights, enabling global last-iterate convergence and scalable runtime. Theoretical results establish convergence guarantees and complexity bounds under exact and approximate policy evaluation, while experiments show superior max-min performance, substantial memory savings, and faster training versus baselines in tabular MORL and traffic signal control tasks. The approach offers a practical path toward robust, fair, and scalable multi-objective RL in real-world resource allocation settings.
Abstract
In this paper, we propose a provably convergent and practical framework for multi-objective reinforcement learning with max-min criterion. From a game-theoretic perspective, we reformulate max-min multi-objective reinforcement learning as a two-player zero-sum regularized continuous game and introduce an efficient algorithm based on mirror descent. Our approach simplifies the policy update while ensuring global last-iterate convergence. We provide a comprehensive theoretical analysis on our algorithm, including iteration complexity under both exact and approximate policy evaluations, as well as sample complexity bounds. To further enhance performance, we modify the proposed algorithm with adaptive regularization. Our experiments demonstrate the convergence behavior of the proposed algorithm in tabular settings, and our implementation for deep reinforcement learning significantly outperforms previous baselines in many MORL environments.
