Policy Gradient Algorithms in Average-Reward Multichain MDPs

Abstract

While there is an extensive body of research analyzing policy gradient methods for discounted cumulative-reward MDPs, prior work on policy gradient methods for average-reward MDPs has been limited, with most existing results restricted to ergodic or unichain settings. In this work, we first establish a policy gradient theorem for average-reward multichain MDPs based on the invariance of the classification of recurrent and transient states. Building on this foundation, we develop refined analyses and obtain a collection of convergence and sample-complexity results that advance the understanding of this setting. In particular, we show that the proposed $\hat{I}\pm$-clipped policy mirror ascent algorithm attains an $\hat{I}\mu$-optimal policy with respect to positive policies.

arXiv admin note: text overlap with arXiv:2510.18340