Primal Method for ERM with Flexible Mini-batching Schemes and Non-convex Losses

By Dominik Csiba, Peter Richtárik. 2015

23 citations recorded by the source corpus.

In this work we develop a new algorithm for regularized empirical risk minimization. Our method extends recent techniques of Shalev-Shwartz [02/2015], which enable a dual-free analysis of SDCA, to arbitrary mini-batching schemes. Moreover, our method is able to better utilize the information in the data dening the ERM problem. For convex loss functions, our complexity results match those of QUARTZ, which is a primal-dual method also allowing for arbitrary mini-batching schemes. The advantage of a dual-free analysis comes from the fact that it guarantees convergence even for non-convex loss functions, as long as the average loss is convex. We illustrate through experiments the utility of being able to design arbitrary mini-batching schemes.

Subfields: Artificial Intelligence, Computational Mechanics, Statistics and Probability

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. Jakub Konecný
  3. Samuel Horváth
  4. Dmitry Kovalev
  5. Dominik Csiba
  6. Nicolas Loizou
  7. Filip Hanzely
  8. Martin Takác

In Computational Mechanics

  1. Peter Richtárik
  2. Jakub Konecný
  3. Dominik Csiba
  4. Samuel Horváth
  5. Dmitry Kovalev
  6. Nicolas Loizou
  7. Martin Takác
  8. Zheng Qu

In Statistics and Probability

  1. Jakub Konecný
  2. Samuel Horváth
  3. Peter Richtárik
  4. Dmitry Kovalev
  5. Dominik Csiba
  6. Filip Hanzely
  7. Nicolas Loizou
  8. Martin Takác

Citations in the corpus