English

Finite-Sample Analysis of Policy Evaluation for Robust Average Reward Reinforcement Learning

Machine Learning 2025-12-11 v4 Machine Learning

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 O~(ϵ2)\tilde{\mathcal{O}}(\epsilon^{-2}) for robust policy evaluation and robust average reward estimation, marking a significant advancement in robust reinforcement learning theory.

Keywords

Cite

@article{arxiv.2502.16816,
  title  = {Finite-Sample Analysis of Policy Evaluation for Robust Average Reward Reinforcement Learning},
  author = {Yang Xu and Washim Uddin Mondal and Vaneet Aggarwal},
  journal= {arXiv preprint arXiv:2502.16816},
  year   = {2025}
}

Comments

The 39th Conference on Neural Information Processing Systems (NeurIPS 2025)

R2 v1 2026-06-28T21:54:56.907Z