ORVIT: Near-Optimal Online Distributionally Robust Reinforcement Learning
Debamita Ghosh, George K. Atia, Yue Wang
Abstract
We investigate reinforcement learning (RL) in the presence of distributional mismatch between training and deployment, where policies trained in simulators often underperform in practice due to mismatches between training and deployment conditions, and thereby reliable guarantees on real-world performance are essential. Distributionally robust RL addresses this issue by optimizing worst-case performance over an uncertainty set of environments and providing an optimized lower bound on deployment performance. However, existing studies typically assume access to either a generative model or offline datasets with broad coverage of the deployment environment-assumptions that limit their practicality in unknown environments without prior knowledge. In this work, we study a more practical and challenging setting: online distributionally robust RL, where the agent interacts only with a single unknown training environment while seeking policies that are robust with respect to an uncertainty set around this nominal model. We consider general f -divergence-based ambiguity sets, including χ 2 and KL divergence balls, and design a computationally efficient algorithm that achieves sublinear regret for the robust control objective under minimal assumptions, without requiring generative or offline data access. Moreover, we establish a corresponding minimax lower bound on the regret of any online algorithm, demonstrating the near-optimality of our method. Experiments across diverse environments with model misspecification show that our approach consistently improves worst-case performance and aligns with the theoretical guarantees.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext eaec4c4f-0975-4adb-ba6a-f962cadc6f14Cited by top-tier papers2
- Sample-Efficient Distributionally Robust Multi-Agent Reinforcement Learning via Online InteractionZain Ulabedeen Farhat, Debamita Ghosh, George K. Atia, Yue WangICLR 2026 · 5 citations
- Model-Free Robust Average-Reward Reinforcement Learning with Sample Complexity AnalysisZachary Roch, George Atia, Yue WangICML 2026 · 1 citation
Builds on30
- Training language models to follow instructions with human feedbackLong Ouyang, Jeffrey Wu, Xu Jiang, Diogo Almeida et al.NeurIPS 2022 · 24,707 citations
- Training Diffusion Models with Reinforcement LearningKevin Black, Michael Janner, Yilun Du, Ilya Kostrikov et al.ICLR 2024 · 816 citations
- Bellman Eluder Dimension: New Rich Classes of RL Problems, and Sample-Efficient AlgorithmsChi Jin, Qinghua Liu, Sobhan MiryoosefiNeurIPS 2021 · 264 citations
- Guiding Pretraining in Reinforcement Learning with Large Language ModelsYuqing Du, Olivia Watkins, Zihan Wang, Cédric Colas et al.ICML 2023 · 257 citations
- Almost Optimal Model-Free Reinforcement Learningvia Reference-Advantage DecompositionZihan Zhang, Yuan Zhou, Xiangyang JiNeurIPS 2020 · 183 citations
Related papers
- Online Robust Reinforcement Learning with General Function ApproximationDebamita Ghosh, George Atia, Yue WangICML 2026
- A Unified Principle of Pessimism for Offline Reinforcement Learning under Model MismatchYue Wang, Zhongchang Sun, Shaofeng ZouNeurIPS 2024 · 11 citations
- Sample Complexity of Distributionally Robust Off-Dynamics Reinforcement Learning with Online InteractionYiting He, Zhishuai Liu, Weixin Wang, Pan XuICML 2025
- Distributionally Robust Q-LearningZijian Liu, Qinxun Bai, Jose H. Blanchet, Perry Dong et al.ICML 2022 · 72 citations
- 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
