Sample complexity of variance-reduced policy gradient: weaker assumptions and lower bounds

Abstract:Several variance-reduced versions of REINFORCE based on importance sampling achieve an improved $O(\epsilon^{-3})$ sample complexity to find an $\epsilon$-stationary point, under an unrealistic assumption on the variance of the importance weights. In this paper, we propose the \algo (Defensive Policy Gradient) algorithm, based on defensive importance sampling, which achieves the same rate without any assumption on the variance of ordinary importance weights. We also establish lower bounds in a generalized black-box policy-optimization model that hides states and actions and permits parameter-dependent rewards. In this model, the optimal rates are $\Theta(\epsilon^{-4})$ with bounded-variance one-policy feedback and $\Theta(\epsilon^{-3})$ with mean-square-smooth coupled two-policy feedback. Under standard policy-regularity conditions, REINFORCE and \algo realize the corresponding oracle conditions and attain the $O(\epsilon^{-4})$ and $O(\epsilon^{-3})$ upper bounds, respectively. Although the lower bounds do not apply directly to the classical MDP interaction model in which these algorithms operate, this correspondence provides oracle-level evidence that the faster rate of \algo is optimal and genuinely separated from that of vanilla policy gradient.

Machine Learning (cs.LG)

arXiv:2610.03165 [cs.LG]

(or arXiv:2610.03165v1 [cs.LG] for this version)

https://doi.org/10.48550/arXiv.2610.03165 arXiv-issued DOI via DataCite (pending registration)

Paczolay, G., Papini, M., Metelli, A.M. et al. Sample complexity of variance-reduced policy gradient: weaker assumptions and lower bounds. Mach Learn 113, 6475-6510 (2024)

https://doi.org/10.1007/s10994-024-06573-4 DOI(s) linking to related resources