EventGraD: Event-Triggered Communication in Parallel Machine Learning

Abstract

Communication in parallel systems imposes significant overhead which often turns out to be a bottleneck in parallel machine learning. To relieve some of this overhead, in this paper, we present EventGraD - an algorithm with event-triggered communication for stochastic gradient descent in parallel machine learning. The main idea of this algorithm is to modify the requirement of communication at every iteration in standard implementations of stochastic gradient descent in parallel machine learning to communicating only when necessary at certain iterations. We provide theoretical analysis of convergence of our proposed algorithm. We also implement the proposed algorithm for data-parallel training of a popular residual neural network used for training the CIFAR-10 dataset and show that EventGraD can reduce the communication load by up to 60% while retaining the same level of accuracy. In addition, EventGraD can be combined with other approaches such as Top-K sparsification to decrease communication further while maintaining accuracy.

arXiv commentPublished in Neurocomputing, Nov 2021

References in corpus (13)

• HOGWILD!: A Lock-Free Approach to Parallelizing Stochastic Gradient Descent
• Deep Gradient Compression: Reducing the Communication Bandwidth for Distributed Training
• Mixed Precision Training
• QSGD: Communication-Efficient SGD via Gradient Quantization and Encoding
• Communication-Efficient Federated Deep Learning with Asynchronous Model Update and Temporally Weighted Aggregation
• Can Decentralized Algorithms Outperform Centralized Algorithms? A Case Study for Decentralized Parallel Stochastic Gradient Descent
• Asynchronous Parallel Stochastic Gradient for Nonconvex Optimization
• GossipGraD: Scalable Deep Learning using Gossip Communication based Asynchronous Gradient Descent
• Deep learning with Elastic Averaging SGD
• Gossip training for deep learning
• The Convergence of Sparsified Gradient Methods
• Ternary Compression for Communication-Efficient Federated Learning
• Distributed Deep Learning with Event-Triggered Communication