English

Provably Sample-Efficient Robust Reinforcement Learning with Average Reward

Machine Learning 2025-09-26 v2 Machine Learning

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 specification. However, a significant gap exists in understanding the finite-sample complexity of these methods, as most existing work provides only asymptotic guarantees. This limitation hinders their principled understanding and practical deployment, especially in data-limited scenarios. We close this gap by proposing \textbf{Robust Halpern Iteration (RHI)}, a new algorithm designed for robust Markov Decision Processes (MDPs) with transition uncertainty characterized by p\ell_p-norm and contamination models. Our approach offers three key advantages over previous methods: (1). Weaker Structural Assumptions: RHI only requires the underlying robust MDP to be communicating, a less restrictive condition than the commonly assumed ergodicity or irreducibility; (2). No Prior Knowledge: Our algorithm operates without requiring any prior knowledge of the robust MDP; (3). State-of-the-Art Sample Complexity: To learn an ϵ\epsilon-optimal robust policy, RHI achieves a sample complexity of O~(SAH2ϵ2)\tilde{\mathcal O}\left(\frac{SA\mathcal H^{2}}{\epsilon^{2}}\right), where SS and AA denote the numbers of states and actions, and H\mathcal H is the robust optimal bias span. This result represents the tightest known bound. Our work hence provides essential theoretical understanding of sample efficiency of robust average reward RL.

Keywords

Cite

@article{arxiv.2505.12462,
  title  = {Provably Sample-Efficient Robust Reinforcement Learning with Average Reward},
  author = {Zachary Roch and Chi Zhang and George Atia and Yue Wang},
  journal= {arXiv preprint arXiv:2505.12462},
  year   = {2025}
}

Comments

Preprint, work in progress

R2 v1 2026-07-01T02:19:54.423Z