PreprintCited by 80Stochastic Gradient Optimization TechniquesParallel Computing and Optimization TechniquesDistributed Control Multi-Agent Systems

GossipGraD: Scalable Deep Learning using Gossip Communication based Asynchronous Gradient Descent

Open access
Daily, Jeff, Vishnu, Abhinav, Siegel, Charles, Warfel, Thomas, Amatya, Vinay·2018·arXiv

Abstract

In this paper, we present GossipGraD - a gossip communication protocol based Stochastic Gradient Descent (SGD) algorithm for scaling Deep Learning (DL) algorithms on large-scale systems. The salient features of GossipGraD are: 1) reduction in overall communication complexity from Θ(log(p)) for p compute nodes in well-studied SGD to O(1), 2) model diffusion such that compute nodes exchange their updates (gradients) indirectly after every log(p) steps, 3) rotation of communication partners for facilitating direct diffusion of gradients, 4) asynchronous distributed shuffle of samples during the feedforward phase in SGD to prevent over-fitting, 5) asynchronous communication of gradients for further reducing the communication cost of SGD and GossipGraD. We implement GossipGraD for GPU and CPU clusters and use NVIDIA GPUs (Pascal P100) connected with InfiniBand, and Intel Knights Landing (KNL) connected with Aries network. We evaluate GossipGraD using well-studied dataset ImageNet-1K (~250GB), and widely studied neural network topologies such as GoogLeNet and ResNet50 (current winner of ImageNet Large Scale Visualization Research Challenge (ILSVRC)). Our performance evaluation using both KNL and Pascal GPUs indicates that GossipGraD can achieve perfect efficiency for these datasets and their associated neural network topologies. Specifically, for ResNet50, GossipGraD is able to achieve ~100% compute efficiency using 128 NVIDIA Pascal P100 GPUs - while matching the top-1 classification accuracy published in literature.

Integrity signals

Registry

DataCite · 10.48550/arxiv.1803.05880

Cites

42 of 42 references verified0 citation errors · 0 not checkable

Statistics

Recomputing p-values needs the paper’s full text.

Claims the paper makes

We couldn’t retrieve this paper’s full text, so its claims and self-check weren’t run. The registry and citation checks still apply.

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

  • By deployment on large-scale systems, trivial gossip-based algorithms at scale lead to a communication imbalance, poor convergence, and heavy communication overhead.

    “By deployment on large-scale systems, Daily et al. [66] show that the trivial gossip-based algorithms at scale lead to a communication imbalance, poor convergence, and heavy communication overhead.”
  • GossipGrad reduces the overall communication complexity from Θ(log(p)) to O(1).

    “GossipGrad reduces the overall communication complexity from Θ(log(p)) to O(1) and considers diffusion such that computing nodes exchange their updates (gradients) indirectly after every log(p) steps.”

Cites

Flagged first, then by citations

Cited by

Sorted by citations

Authors
Daily, J., Vishnu, A., Siegel, C., Warfel, T., & Amatya, V.
Venue
arXiv
Publisher
arXiv
Published
2018
ISSN
2331-8422 (ISSN-L)
OpenAlex
W2790598069
Cited by
80
References
42
Citations by year
2025: 3 · 2024: 5 · 2023: 4 · 2022: 16 · 2021: 17 · 2020: 26 · 2019: 8 · 2018: 1
Type
posted content
Language
en
Open access
Locations
  • arXiv (Cornell University) (submittedVersion)
  • arXiv (Cornell University)
Topics
  • Physical Sciences › Computer Science › Artificial Intelligence › Stochastic Gradient Optimization Techniques
  • Physical Sciences › Computer Science › Hardware and Architecture › Parallel Computing and Optimization Techniques
  • Physical Sciences › Computer Science › Computer Networks and Communications › Distributed Control Multi-Agent Systems
Keywords
stochastic gradient descent, distributed deep learning, communication complexity, GPU clusters, Intel Knights Landing, ResNet50, ImageNet classification, scalability

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review