CSER: Communication-efficient SGD with Error Reset

By Cong Xie, Shuai Zheng, Oluwasanmi Koyejo, Indranil Gupta, Mu Li, Haibin Lin. 2020

48 citations recorded by the source corpus.

The scalability of Distributed Stochastic Gradient Descent (SGD) is today limited by communication bottlenecks. We propose a novel SGD variant: Communication-efficient SGD with Error Reset, or CSER. The key idea in CSER is first a new technique called "error reset" that adapts arbitrary compressors for SGD, producing bifurcated local models with periodic reset of resulting local residual errors. Second we introduce partial synchronization for both the gradients and the models, leveraging advantages from them. We prove the convergence of CSER for smooth non-convex problems. Empirical results show that when combined with highly aggressive compressors, the CSER algorithms: i) cause no loss of accuracy, and ii) accelerate the training by nearly $10\times$ for CIFAR-100, and by $4.5\times$ for ImageNet.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

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. Cong Xie
  2. Oluwasanmi Koyejo
  3. Indranil Gupta
  4. Haibin Lin
  5. Lifeng Lai
  6. Xinyang Cao
  7. Rachid Guerraoui
  8. El-Mahdi El-Mhamdi

In Computer Vision and Pattern Recognition

  1. Indranil Gupta
  2. Oluwasanmi Koyejo
  3. Cong Xie
  4. Haibin Lin

Citations in the corpus