Less than a Single Pass: Stochastically Controlled Stochastic Gradient

[edit]

Lihua Lei, Michael Jordan ;
Proceedings of the 20th International Conference on Artificial Intelligence and Statistics, PMLR 54:148-156, 2017.

Abstract

We develop and analyze a procedure for gradient-based optimization that we refer to as stochastically controlled stochastic gradient (SCSG). As a member of the SVRG family of algorithms, SCSG makes use of gradient estimates at two scales. Unlike most existing algorithms in this family, both the computation cost and the communication cost of SCSG do not necessarily scale linearly with the sample size n; indeed, these costs are independent of n when the target accuracy is small. An experimental evaluation of SCSG on the MNIST dataset shows that it can yield accurate results on this dataset on a single commodity machine with a memory footprint of only 2.6MB and only eight disk accesses.

Related Material