Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. PowerSGD: Practical Low-Rank Gradient Compression for Distributed Optimization
 
conference paper

PowerSGD: Practical Low-Rank Gradient Compression for Distributed Optimization

Vogels, Thijs  
•
Karinireddy, Sai Praneeth
•
Jaggi, Martin  
January 1, 2019
Advances In Neural Information Processing Systems 32 (NeurIPS 2019)
33rd Conference on Neural Information Processing Systems (NeurIPS)

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.

  • Files
  • Details
  • Metrics
Loading...
Thumbnail Image
Name

NeurIPS-2019-powersgd-practical-low-rank-gradient-compression-for-distributed-optimization-Paper.pdf

Type

Publisher's Version

Version

http://purl.org/coar/version/c_970fb48d4fbd8a85

Access type

openaccess

License Condition

CC BY

Size

487.23 KB

Format

Adobe PDF

Checksum (MD5)

84e367aa93dea0738f2a6eacc3189394

Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés