Variance Reduction Based Experience Replay for Policy Optimization
Hua Zheng, Wei Xie, M. Ben Feng, Keilung Choy
TL;DR
This paper tackles low sample efficiency in policy-gradient RL by proposing Variance Reduction Experience Replay (VRER), a principled framework that selectively reuses informative step-based experiences to reduce gradient variance. VRER integrates with existing policy optimization methods to produce PG-VRER, and the authors provide a novel finite-time convergence analysis that accounts for Markovian noise and behavior-policy interdependence, revealing a bias-variance trade-off in reusing older samples. The framework relies on a variance-based sample selection rule, bias-mitigating downsampling, and a moving-block bootstrap variance estimator, together enabling practical, off-policy improvements with theoretical guarantees. Empirically, VRER consistently accelerates convergence and improves performance across standard Gym/PyBullet benchmarks for TRPO, PPO, and A2C variants, while analyzing reuse patterns and parameter sensitivity to guide practical deployment.
Abstract
Effective reinforcement learning (RL) for complex stochastic systems requires leveraging historical data collected in previous iterations to accelerate policy optimization. Classical experience replay treats all past observations uniformly and fails to account for their varying contributions to learning. To overcome this limitation, we propose Variance Reduction Experience Replay (VRER), a principled framework that selectively reuses informative samples to reduce variance in policy gradient estimation. VRER is algorithm-agnostic and integrates seamlessly with existing policy optimization methods, forming the basis of our sample-efficient off-policy algorithm, Policy Gradient with VRER (PG-VRER). Motivated by the lack of rigorous theoretical analysis of experience replay, we develop a novel framework that explicitly captures dependencies introduced by Markovian dynamics and behavior-policy interactions. Using this framework, we establish finite-time convergence guarantees for PG-VRER and reveal a fundamental bias-variance trade-off: reusing older experience increases bias but simultaneously reduces gradient variance. Extensive empirical experiments demonstrate that VRER consistently accelerates policy learning and improves performance over state-of-the-art policy optimization algorithms.
