English

Efficient $Q$-Learning and Actor-Critic Methods for Robust Average Reward Reinforcement Learning

Machine Learning 2025-12-11 v3 Artificial Intelligence Machine Learning

Abstract

We present a non-asymptotic convergence analysis of QQ-learning and actor-critic algorithms for robust average-reward Markov Decision Processes (MDPs) under contamination, total-variation (TV) distance, and Wasserstein uncertainty sets. A key ingredient of our analysis is showing that the optimal robust QQ operator is a strict contraction with respect to a carefully designed semi-norm (with constant functions quotiented out). This property enables a stochastic approximation update that learns the optimal robust QQ-function using O~(ϵ2)\tilde{\mathcal{O}}(\epsilon^{-2}) samples. We also provide an efficient routine for robust QQ-function estimation, which in turn facilitates robust critic estimation. Building on this, we introduce an actor-critic algorithm that learns an ϵ\epsilon-optimal robust policy within O~(ϵ2)\tilde{\mathcal{O}}(\epsilon^{-2}) samples. We provide numerical simulations to evaluate the performance of our algorithms.

Keywords

Cite

@article{arxiv.2506.07040,
  title  = {Efficient $Q$-Learning and Actor-Critic Methods for Robust Average Reward Reinforcement Learning},
  author = {Yang Xu and Swetha Ganesh and Vaneet Aggarwal},
  journal= {arXiv preprint arXiv:2506.07040},
  year   = {2025}
}

Comments

Updated the main text of the draft

R2 v1 2026-07-01T03:05:25.945Z