Timezone: »
Spotlight
Distributionally Robust $Q$-Learning
Zijian Liu · Zhengqing Zhou · Perry Dong · Jerry Bai · Jose Blanchet · Wei Xu · Zhengyuan Zhou
Reinforcement learning (RL) has demonstrated remarkable achievements in simulated environments. However, carrying this success to real environments requires the important attribute of robustness, which the existing RL algorithms often lack as they assume the future deployment environment is the same as the training environment (i.e. simulator) in which the policy is learned, an assumption that 1) often does not hold due to the discrepancy between the simulator and the real environment and 2) renders the learned policy fragile as a result.In this paper, we aim to make initial progress in addressing the robustness problem. In particular, we propose a novel distributionally robust $Q$-learning algorithm that learns the best policy in the worst distributional perturbation of the environment. Our algorithm first transforms the infinite-dimensional learning problem (since the environment MDP perturbation lies in an infinite-dimensional space) into a finite-dimensional dual problem and subsequently uses a multi-level Monte-Carlo scheme to approximate the dual value using samples from the simulator. Despite the complexity, we show that the resulting distributionally robust $Q$-learning algorithm asymptotically converges to optimal worst-case policy, thus making it robust to future environment changes. Simulation results further demonstrate its empirical robustness.
Author Information
Zijian Liu (Boston University)
Zhengqing Zhou (Stanford University)
Perry Dong (University of California, Berkeley)
Jerry Bai (Horizon Robotics)
Jose Blanchet (Stanford University)
Wei Xu (Horizon Robotics)
Zhengyuan Zhou (Arena Technologies & NYU)
Related Events (a corresponding poster, oral, or spotlight)
-
2022 Poster: Distributionally Robust $Q$-Learning »
Wed. Jul 20th through Thu the 21st Room Hall E
More from the Same Authors
-
2022 Poster: Adaptive Accelerated (Extra-)Gradient Methods with Variance Reduction »
Zijian Liu · Ta Duy Nguyen · Alina Ene · Huy Nguyen -
2022 Poster: Doubly Robust Distributionally Robust Off-Policy Evaluation and Learning »
Kaiwen Wang · Xiaojie Mao · Zhengyuan Zhou · Nathan Kallus -
2022 Spotlight: Adaptive Accelerated (Extra-)Gradient Methods with Variance Reduction »
Zijian Liu · Ta Duy Nguyen · Alina Ene · Huy Nguyen -
2022 Spotlight: Doubly Robust Distributionally Robust Off-Policy Evaluation and Learning »
Kaiwen Wang · Xiaojie Mao · Zhengyuan Zhou · Nathan Kallus -
2021 Poster: Testing Group Fairness via Optimal Transport Projections »
Nian Si · Karthyek Murthy · Jose Blanchet · Viet Anh Nguyen -
2021 Spotlight: Testing Group Fairness via Optimal Transport Projections »
Nian Si · Karthyek Murthy · Jose Blanchet · Viet Anh Nguyen -
2021 Poster: Sequential Domain Adaptation by Synthesizing Distributionally Robust Experts »
Bahar Taskesen · Man-Chung Yue · Jose Blanchet · Daniel Kuhn · Viet Anh Nguyen -
2021 Oral: Sequential Domain Adaptation by Synthesizing Distributionally Robust Experts »
Bahar Taskesen · Man-Chung Yue · Jose Blanchet · Daniel Kuhn · Viet Anh Nguyen -
2021 Poster: Generative Particle Variational Inference via Estimation of Functional Gradients »
Neale Ratzlaff · Jerry Bai · Fuxin Li · Wei Xu -
2021 Spotlight: Generative Particle Variational Inference via Estimation of Functional Gradients »
Neale Ratzlaff · Jerry Bai · Fuxin Li · Wei Xu -
2020 Poster: Implicit Generative Modeling for Efficient Exploration »
Neale Ratzlaff · Qinxun Bai · Fuxin Li · Wei Xu -
2020 Poster: Robust Bayesian Classification Using An Optimistic Score Ratio »
Viet Anh Nguyen · Nian Si · Jose Blanchet -
2020 Poster: Distributionally Robust Policy Evaluation and Learning in Offline Contextual Bandits »
Nian Si · Fan Zhang · Zhengyuan Zhou · Jose Blanchet -
2019 Poster: Probability Functional Descent: A Unifying Perspective on GANs, Variational Inference, and Reinforcement Learning »
Casey Chu · Jose Blanchet · Peter Glynn -
2019 Oral: Probability Functional Descent: A Unifying Perspective on GANs, Variational Inference, and Reinforcement Learning »
Casey Chu · Jose Blanchet · Peter Glynn