Scaling Up Quasi-newton Algorithms: Communication Efficient Distributed SR1

By Majid Jahani, M. Nazari, Sergey G. Rusakov, Albert S. Berahas, Martin Tak'avc. 2019

17 citations recorded by the source corpus.

In this paper, we present a scalable distributed implementation of the sampled LSR1 (S-LSR1) algorithm. First, we show that a naive distributed implementation of S-LSR1 requires multiple rounds of expensive communications at every iteration and thus is inefficient. We then propose DS-LSR1, a communication-efficient variant of the S-LSR1 method, that drastically reduces the amount of data communicated at every iteration, that has favorable work-load balancing across nodes and that is matrix-free and inverse-free. The proposed method scales well in terms of both the dimension of the problem and the number of data points. Finally, we illustrate the performance of DS-LSR1 on standard neural network training tasks.

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. Chenxin Ma
  2. Aryan Mokhtari
  3. Alejandro Ribeiro
  4. Peter Richtárik
  5. Jakub Konecný
  6. Martin Takác
  7. Majid Jahani
  8. Xi He

In Computational Mechanics

  1. Aryan Mokhtari
  2. Chenxin Ma
  3. Majid Jahani
  4. Martin Takác
  5. Alejandro Ribeiro
  6. Peter Richtárik
  7. Dheevatsa Mudigere
  8. Jakub Konecný

Citations in the corpus