Finite-Sample Analysis of Policy Evaluation for Robust Average Reward Reinforcement Learning
Yang Xu, Washim Uddin Mondal, Vaneet Aggarwal
Abstract
We present the first finite-sample analysis of policy evaluation in robust average-reward Markov Decision Processes (MDPs). Prior work in this setting have established only asymptotic convergence guarantees, leaving open the question of sample complexity. In this work, we address this gap by showing that the robust Bellman operator is a contraction under a carefully constructed semi-norm, and developing a stochastic approximation framework with controlled bias. Our approach builds upon Multi-Level Monte Carlo (MLMC) techniques to estimate the robust Bellman operator efficiently. To overcome the infinite expected sample complexity inherent in standard MLMC, we introduce a truncation mechanism based on a geometric distribution, ensuring a finite expected sample complexity while maintaining a small bias that decays exponentially with the truncation level. Our method achieves the order-optimal sample complexity of for robust policy evaluation and robust average reward estimation, marking a significant advancement in robust reinforcement learning theory.
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 papers2
- Model-Free Robust Average-Reward Reinforcement Learning with Sample Complexity AnalysisZachary Roch, George Atia, Yue WangICML 2026 · 1 citation
- Distributionally Robust Markov Games with Average RewardZachary Roch, Yue WangICML 2026
Builds on12
- Finite-Sample Analysis of Contractive Stochastic Approximation Using Smooth Convex EnvelopesZaiwei Chen, Siva Theja Maguluri, Sanjay Shakkottai, Karthikeyan ShanmugamNeurIPS 2020 · 66 citations
- Policy Gradient for Rectangular Robust Markov Decision ProcessesNavdeep Kumar, Esther Derman, Matthieu Geist, Kfir Y. Levy et al.NeurIPS 2023 · 45 citations
- First-Order Methods for Wasserstein Distributionally Robust MDPJulien Grand-Clément, Christian KroerICML 2021 · 32 citations
- Learning Robust Policy against Disturbance in Transition Dynamics via State-Conservative Policy OptimizationYufei Kuang, Miao Lu, Jie Wang, Qi Zhou et al.AAAI 2022 · 29 citations
- Model-Free Robust Average-Reward Reinforcement LearningYue Wang, Alvaro Velasquez, George K. Atia, Ashley Prater-Bennette et al.ICML 2023 · 25 citations
Related papers
- Robust Reinforcement Learning using Least Squares Policy Iteration with Provable Performance GuaranteesKishan Panaganti Badrinath, Dileep KalathilICML 2021 · 78 citations
- A Reduction Framework for Distributionally Robust Reinforcement Learning under Average RewardZachary Roch, George K. Atia, Yue WangICML 2025
- Truncating Trajectories in Monte Carlo Policy Evaluation: an Adaptive ApproachRiccardo Poiani, Nicole Nobili, Alberto Maria Metelli, Marcello RestelliNeurIPS 2023 · 3 citations
- Sample Complexity of Distributionally Robust Average-Reward Reinforcement LearningZijun Chen, Shengbo Wang, Nian SiNeurIPS 2025 · 9 citations
- A Sharper Global Convergence Analysis for Average Reward Reinforcement Learning via an Actor-Critic ApproachSwetha Ganesh, Washim Uddin Mondal, Vaneet AggarwalICML 2025
