On Practical Robust Reinforcement Learning: Adjacent Uncertainty Set and Double-Agent Algorithm

• Published in: IEEE transaction on neural networks and learning systems Vol. 36; no. 4; pp. 7696 - 7710
• Main Authors: Hwang, Ukjo, Hong, Songnam
• Format: Journal Article
• Language: English
• Published: United States IEEE 01.04.2025
• Subjects: Optimization; Q-learning; Reinforcement learning (RL); robust RL (RRL); Robustness; Testing; Training; Uncertainty; uncertainty set
• ISSN: 2162-237X; 2162-2388; 2162-2388
• DOI: 10.1109/TNNLS.2024.3385234

Robust reinforcement learning (RRL) aims to seek a robust policy by optimizing the worst case performance over an uncertainty set. This set contains some perturbed Markov decision processes (MDPs) from a nominal MDP (N-MDP) that generate samples for training, which reflects some potential mismatches between the training simulator (i.e., N-MDP) and real-world settings (i.e., the testing environments). Unfortunately, existing RRL algorithms are only applied to the tabular setting and it is still an open problem to extend them into more general continuous state space. We contribute to this subject in the following ways. We first construct an elaborated uncertainty set, which contains plausible (perturbed) MDPs only compared with the existing sets. Based on this, we propose a sample-based RRL algorithm [named adjacent robust Q-learning (ARQ-Learning)] for the tabular setting and characterize its finite-time error bound. Also, it is proved that ARQ-Learning converges as fast as the standard Q-learning and robust Q-learning (Robust-Q) while guaranteeing better robustness. Our major contribution is to introduce an additional pessimistic agent that can address the major hurdle for the extension of ARQ-Learning into cases with large or continuous state spaces. Leveraging this double-agent approach, we for the first time develop (model-free) RRL algorithms for continuous state/action spaces. Via experiments, we demonstrate the effectiveness of our algorithms.