PowerSGD: Practical Low-Rank Gradient Compression for Distributed Optimization
We study lossy gradient compression methods to alleviate the communication bottleneck in data-parallel distributed optimization. Despite the significant attention received, current compression schemes either do not scale well, or fail to achieve the target test accuracy. We propose a new low-rank gradient compressor based on power iteration that can i) compress gradients rapidly, ii) efficiently aggregate the compressed gradients using all-reduce, and iii) achieve test performance on par with SGD. The proposed algorithm is the only method evaluated that achieves consistent wall-clock speedups when benchmarked against regular SGD using highly optimized off-the-shelf tools for distributed communication. We demonstrate reduced training times for convolutional networks as well as LSTMs on common datasets. Our code is available at https://github.com/epfml/powersgd.
Cite this item
Vogels, T., Karinireddy, S. P., & Jaggi, M. (2019). PowerSGD: Practical Low-Rank Gradient Compression for Distributed Optimization. Advances in Neural Information Processing Systems 32 (NeurIPS 2019), Advances in Neural Information Processing Systems, 32. https://infoscience.epfl.ch/handle/20.500.14299/169957
Click Copy to copy the citation, or select the text manually.
NeurIPS-2019-powersgd-practical-low-rank-gradient-compression-for-distributed-optimization-Paper.pdf
Publisher's Version
openaccess
CC BY
487.23 KB
Adobe PDF
84e367aa93dea0738f2a6eacc3189394