Abstract:In this paper, we propose a simple variant of the original stochastic variance reduction gradient (SVRG), where hereafter we refer to as the variance reduced stochastic gradient descent (VR-SGD). Different from the choices of the snapshot point and starting point in SVRG and its proximal variant, Prox-SVRG, the two vectors of each epoch in VR-SGD are set to the average and last iterate of the previous epoch, respectively. This setting allows us to use much larger learning rates or step sizes than SVRG, e.g., 3/(7L) for VR-SGD vs 1/(10L) for SVRG, and also makes our convergence analysis more challenging. In fact, a larger learning rate enjoyed by VR-SGD means that the variance of its stochastic gradient estimator asymptotically approaches zero more rapidly. Unlike common stochastic methods such as SVRG and proximal stochastic methods such as Prox-SVRG, we design two different update rules for smooth and non-smooth objective functions, respectively. In other words, VR-SGD can tackle non-smooth and/or non-strongly convex problems directly without using any reduction techniques such as quadratic regularizers. Moreover, we analyze the convergence properties of VR-SGD for strongly convex problems, which show that VR-SGD attains a linear convergence rate. We also provide the convergence guarantees of VR-SGD for non-strongly convex problems. Experimental results show that the performance of VR-SGD is significantly better than its counterparts, SVRG and Prox-SVRG, and it is also much better than the best known stochastic method, Katyusha.

Stochastic Variance Reduction for Deep Q-learning

Policy Optimization with Stochastic Mirror Descent.

Gradient Q : A Unified Algorithm with Function Approximation for Reinforcement Learning

Stochastic Gradient Descent with Variance Reduction Technique

Gradient Q(σ, Λ): A Unified Algorithm with Function Approximation for Reinforcement Learning

VR-SGD: A Simple Stochastic Variance Reduction Method for Machine Learning

Accelerated Stochastic ADMM with Variance Reduction

Adaptive Variance Reducing for Stochastic Gradient Descent.

Stochastic Sub-Sampled Newton Method with Variance Reduction

On the Reduction of Variance and Overestimation of Deep Q-Learning

Stochastic Variance-Reduced Policy Gradient

Variance Reduced Domain Randomization for Reinforcement Learning With Policy Gradient

A Coefficient Makes SVRG Effective

Variance aware reward smoothing for deep reinforcement learning

Larger is Better: The Effect of Learning Rates Enjoyed by Stochastic Optimization with Progressive Variance Reduction

On the Ineffectiveness of Variance Reduced Optimization for Deep Learning

Stochastic Zeroth-order Optimization Via Variance Reduction Method.

Biologically Plausible Variational Policy Gradient with Spiking Recurrent Winner-Take-All Networks

Amortized Variational Deep Q Network.

Parameter-Free Reduction of the Estimation Bias in Deep Reinforcement Learning for Deterministic Policy Gradients

Trading-off variance and complexity in stochastic gradient descent