Model-Free Robust Average-Reward Reinforcement Learning with Sample Complexity Analysis
Zachary Roch, George Atia, Yue Wang
Abstract
Robust reinforcement learning (RL) under the average-reward criterion is essential for long-term decision-making, particularly when the environment may differ from its training dynamics. However, most existing studies focus on model-based settings and provide only asymptotic guarantees, hindering their principled understanding and practical deployment, especially in data-limited scenarios. We aim to close this gap by proposing a model-free algorithm, Robust Halpern Iteration (RHI). We first design our algorithm based on a black-box sampling oracle, which can estimate the worst-case performance accurately. We then derive the finite sample complexity of RHI under the generative model setting, assuming the sampling oracle. To concretely design such an oracle, we propose a -order multi-level Monte-Carlo estimator, which is shown to have a lower bias compared to prior methods. We further instantiate our design for multiple uncertainty models, including KL and divergence sets, and show that our RHI algorithm achieves an -optimal robust policy with a sample complexity of , where are the number of states and actions, and is the robust optimal span. Our result asymptotically matches the best complexity in robust average reward RL.
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 70eb6398-aa3f-44c8-a989-647605bff6ecBuilds on22
- Off-Dynamics Reinforcement Learning: Training for Transfer with Domain ClassifiersBenjamin Eysenbach, Shreyas Chaudhari, Swapnil Asawa, Sergey Levine et al.ICLR 2021 · 120 citations
- Distributionally Robust Q-LearningZijian Liu, Qinxun Bai, Jose H. Blanchet, Perry Dong et al.ICML 2022 · 72 citations
- Twice regularized MDPs and the equivalence between robustness and regularizationEsther Derman, Matthieu Geist, Shie MannorNeurIPS 2021 · 68 citations
- Exact Optimal Accelerated Complexity for Fixed-Point IterationsJisun Park, Ernest K. RyuICML 2022 · 50 citations
- Finite Sample Analysis of Average-Reward TD Learning and -LearningSheng Zhang, Zhe Zhang, Siva Theja MaguluriNeurIPS 2021 · 48 citations
Related papers
- Combining Pessimism with Optimism for Robust and Efficient Model-Based Deep Reinforcement LearningSebastian Curi, Ilija Bogunovic, Andreas KrauseICML 2021 · 20 citations
- Robust Reinforcement Learning using Offline DataKishan Panaganti, Zaiyan Xu, Dileep Kalathil, Mohammad GhavamzadehNeurIPS 2022 · 130 citations
- Near-Optimal Sample Complexity for MDPs via AnchoringJongmin Lee, Mario Bravo, Roberto CominettiICML 2025
- Model-Free Robust Average-Reward Reinforcement LearningYue Wang, Alvaro Velasquez, George K. Atia, Ashley Prater-Bennette et al.ICML 2023 · 25 citations
- Sample Complexity of Distributionally Robust Average-Reward Reinforcement LearningZijun Chen, Shengbo Wang, Nian SiNeurIPS 2025 · 9 citations
