Robust Reinforcement Learning using Offline Data
Kishan Panaganti, Zaiyan Xu, Dileep Kalathil, Mohammad Ghavamzadeh
Abstract
The goal of robust reinforcement learning (RL) is to learn a policy that is robust against the uncertainty in model parameters. Parameter uncertainty commonly occurs in many real-world RL applications due to simulator modeling errors, changes in the real-world system dynamics over time, and adversarial disturbances. Robust RL is typically formulated as a max-min problem, where the objective is to learn the policy that maximizes the value against the worst possible models that lie in an uncertainty set. In this work, we propose a robust RL algorithm called Robust Fitted Q-Iteration (RFQI), which uses only an offline dataset to learn the optimal robust policy. Robust RL with offline data is significantly more challenging than its non-robust counterpart because of the minimization over all models present in the robust Bellman operator. This poses challenges in offline data collection, optimization over the models, and unbiased estimation. In this work, we propose a systematic approach to overcome these challenges, resulting in our RFQI algorithm. We prove that RFQI learns a near-optimal robust policy under standard assumptions and demonstrate its superior performance on standard benchmark problems.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers49
- The Curious Price of Distributional Robustness in Reinforcement Learning with a Generative ModelLaixi Shi, Gen Li, Yuting Wei, Yuxin Chen et al.NeurIPS 2023 · 66 citations
- Double Pessimism is Provably Efficient for Distributionally Robust Offline Reinforcement Learning: Generic Algorithm and Robust Partial CoverageJose H. Blanchet, Miao Lu, Tong Zhang, Han ZhongNeurIPS 2023 · 58 citations
- Natural Actor-Critic for Robust Reinforcement Learning with Function ApproximationRuida Zhou, Tao Liu, Min Cheng, Dileep Kalathil et al.NeurIPS 2023 · 55 citations
- Robust Multi-Agent Reinforcement Learning via Adversarial Regularization: Theoretical Foundation and Stable AlgorithmsAlexander Bukharin, Yan Li, Yue Yu, Qingru Zhang et al.NeurIPS 2023 · 55 citations
- Policy Gradient in Robust MDPs with Global Convergence GuaranteeQiuhao Wang, Chin Pang Ho, Marek PetrikICML 2023 · 43 citations
Builds on12
- Conservative Q-Learning for Offline Reinforcement LearningAviral Kumar, Aurick Zhou, George Tucker, Sergey LevineNeurIPS 2020 · 2,881 citations
- MOPO: Model-based Offline Policy OptimizationTianhe Yu, Garrett Thomas, Lantao Yu, Stefano Ermon et al.NeurIPS 2020 · 989 citations
- Robust Deep Reinforcement Learning against Adversarial Perturbations on State ObservationsHuan Zhang, Hongge Chen, Chaowei Xiao, Bo Li et al.NeurIPS 2020 · 437 citations
- Bellman-consistent Pessimism for Offline Reinforcement LearningTengyang Xie, Ching-An Cheng, Nan Jiang, Paul Mineiro et al.NeurIPS 2021 · 339 citations
- What are the Statistical Limits of Offline RL with Linear Function Approximation?Ruosong Wang, Dean P. Foster, Sham M. KakadeICLR 2021 · 172 citations
Related papers
- Model-Free Robust ϕ-Divergence Reinforcement Learning Using Both Offline and Online DataKishan Panaganti, Adam Wierman, Eric MazumdarICML 2024 · 12 citations
- Robust Reinforcement Learning using Least Squares Policy Iteration with Provable Performance GuaranteesKishan Panaganti Badrinath, Dileep KalathilICML 2021 · 78 citations
- Corruption-Robust Offline Reinforcement Learning with General Function ApproximationChenlu Ye, Rui Yang, Quanquan Gu, Tong ZhangNeurIPS 2023 · 37 citations
- Online Robust Reinforcement Learning with Model UncertaintyYue Wang, Shaofeng ZouNeurIPS 2021 · 157 citations
- Robust Policy Learning over Multiple Uncertainty SetsAnnie Xie, Shagun Sodhani, Chelsea Finn, Joelle Pineau et al.ICML 2022 · 25 citations
