English

Near-Optimal Sample Complexities of Divergence-based S-rectangular Distributionally Robust Reinforcement Learning

Machine Learning 2026-04-29 v3 Machine Learning

Abstract

Distributionally robust reinforcement learning (DR-RL) has recently gained significant attention as a principled approach that addresses discrepancies between training and testing environments. To balance robustness, conservatism, and computational traceability, the literature has introduced DR-RL models with SA-rectangular and S-rectangular adversaries. While most existing statistical analyses focus on SA-rectangular models, owing to their algorithmic simplicity and the optimality of deterministic policies, S-rectangular models more accurately capture distributional discrepancies in many real-world applications and often yield more effective robust randomized policies. In this paper, we study the empirical value iteration algorithm for divergence-based S-rectangular DR-RL and establish near-optimal sample complexity bounds of O~(SA(1γ)4ε2)\widetilde{O}(|\mathcal{S}||\mathcal{A}|(1-\gamma)^{-4}\varepsilon^{-2}), where ε\varepsilon is the target accuracy, S|\mathcal{S}| and A|\mathcal{A}| denote the cardinalities of the state and action spaces, and γ\gamma is the discount factor. To the best of our knowledge, these are the first sample complexity results for divergence-based S-rectangular models that achieve optimal dependence on S|\mathcal{S}|, A|\mathcal{A}|, and ε\varepsilon simultaneously. We further validate this theoretical dependence through numerical experiments on a robust inventory control problem and a theoretical worst-case example, demonstrating the fast learning performance of our proposed algorithm.

Keywords

Cite

@article{arxiv.2505.12202,
  title  = {Near-Optimal Sample Complexities of Divergence-based S-rectangular Distributionally Robust Reinforcement Learning},
  author = {Zhenghao Li and Shengbo Wang and Nian Si},
  journal= {arXiv preprint arXiv:2505.12202},
  year   = {2026}
}
R2 v1 2026-07-01T02:19:06.932Z