Less than a Single Pass: Stochastically Controlled Stochastic Gradient

By Lihua Lei, Michael I. Jordan. 2016

103 citations recorded by the source corpus.

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, with the number of updates at the faster scale being governed by a geometric random variable. 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 low. An experimental evaluation on real datasets confirms the effectiveness of SCSG.

Subfields: Artificial Intelligence, Computational Mechanics

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Peter Richtárik
  2. Filip Hanzely
  3. Jakub Konecný
  4. Michael I. Jordan
  5. Zeyuan Allen-Zhu
  6. Nicolas Loizou
  7. L. Eon Bottou
  8. Simon Lacoste-Julien

In Computational Mechanics

  1. Peter Richtárik
  2. Filip Hanzely
  3. Zeyuan Allen-Zhu
  4. L. Eon Bottou
  5. Aaron Defazio
  6. Michael I. Jordan
  7. Jakub Konecný
  8. Ashok Cutkosky

Citations in the corpus