cs.LGSep 26, 2026

Distributionally Robust Average-Reward Reinforcement Learning: Finite-Sample Guarantees under Weak Communication

Authors: Chenyu Lu, Zijun Chen, Nian Si

Organizations: The Hong Kong University of Science and Technology

Abstract

We study distributionally robust reinforcement learning (DR-RL) in the average-reward setting under weak communication. Our main result provides finite-sample guarantees for estimating the robust optimal average reward and learning a near-optimal policy, covering both SA-rectangular and S-rectangular structures with divergence-based and distance-based uncertainty sets. Specifically, for Kullback--Leibler and fkf_k-divergence balls, we establish explicit radius conditions under which the robust average-reward Bellman equation admits a constant-gain solution, while for total variation and Wasserstein balls, any positive radius suffices without requiring the nominal MDP to be weakly communicating. Our algorithm is prior-knowledge-free and achieves sample complexities of O~(∣S∣∣A∣p∧−1Span⁡2(uδ∗)ε−2)\widetilde O(|\mathcal{S}||\mathcal{A}|p_{\wedge}^{-1}\operatorname{Span}^{2}(u_δ^{\ast})ε^{-2}) for estimating the robust optimal average reward and O~(∣S∣∣A∣p∧−2Span⁡2(uδ∗)ε−2)\widetilde O(|\mathcal{S}||\mathcal{A}|p_{\wedge}^{-2}\operatorname{Span}^{2}(u_δ^{\ast})ε^{-2}) for learning an εε-optimal policy. Here, p∧p_{\wedge} is the smallest positive nominal transition probability and uδ∗u_δ^{\ast} is a robust optimal bias function. We further provide an almost-tight explicit upper bound on Span⁡(uδ∗)\operatorname{Span}(u_δ^{\ast}). Finally, we validate the predicted n−1/2n^{-1/2} convergence rate through numerical experiments.

Explore similar work

CardsList
  1. Model-Free Robust Average-Reward Reinforcement Learning with Sample Complexity Analysis

    May 18, 2025Zachary Roch, George Atia, Yue WangReinforcement LearningAverage-Reward RL

  2. Robust Average-Reward Markov Decision Processes: Minimax-Optimal Learning via Plug-in Reductions

    Aug 6, 2026Yuepeng Yang, Yuxin Chen, Yuejie ChiRobust Markov Decision ProcessesAverage-Reward RL

  3. Central Limit Theorem for Two-Time-Scale Approximate Distributionally Robust RL

    May 8, 2026Shengbo Wang, Zexi ZhangStochastic ApproximationDistributionally Robust RL