English

Avoiding Synchronization in First-Order Methods for Sparse Convex Optimization

Distributed, Parallel, and Cluster Computing 2017-12-19 v1 Machine Learning Optimization and Control Machine Learning

Abstract

Parallel computing has played an important role in speeding up convex optimization methods for big data analytics and large-scale machine learning (ML). However, the scalability of these optimization methods is inhibited by the cost of communicating and synchronizing processors in a parallel setting. Iterative ML methods are particularly sensitive to communication cost since they often require communication every iteration. In this work, we extend well-known techniques from Communication-Avoiding Krylov subspace methods to first-order, block coordinate descent methods for Support Vector Machines and Proximal Least-Squares problems. Our Synchronization-Avoiding (SA) variants reduce the latency cost by a tunable factor of ss at the expense of a factor of ss increase in flops and bandwidth costs. We show that the SA-variants are numerically stable and can attain large speedups of up to 5.1×5.1\times on a Cray XC30 supercomputer.

Keywords

Cite

@article{arxiv.1712.06047,
  title  = {Avoiding Synchronization in First-Order Methods for Sparse Convex Optimization},
  author = {Aditya Devarakonda and Kimon Fountoulakis and James Demmel and Michael W. Mahoney},
  journal= {arXiv preprint arXiv:1712.06047},
  year   = {2017}
}