mlsanityData to 7 Oct 2026

Communication-Efficient Large-Scale Distributed Deep Learning: A Comprehensive Survey

arXiv 2404.06114 ↗First posted 9 Apr 2024cs.DC cs.AIcited by 9

Not on the map. Why?

Abstract (arXiv)

With the rapid growth in the volume of data sets, models, and devices in the domain of deep learning, there is increasing attention on large-scale distributed deep learning. In contrast to traditional distributed deep learning, the large-scale scenario poses new challenges that include fault tolerance, scalability of algorithms and infrastructures, and heterogeneity in data sets, models, and resources. Due to intensive synchronization of models and sharing of data across GPUs and computing nodes during distributed training and inference processes, communication efficiency becomes the bottleneck for achieving high performance at a large scale. This article surveys the literature over the period of 2018-2023 on algorithms and technologies aimed at achieving efficient communication in large-scale distributed deep learning at various levels, including algorithms, frameworks, and infrastructures. Specifically, we first introduce efficient algorithms for model synchronization and communication data compression in the context of large-scale distributed training. Next, we introduce efficient strategies related to resource allocation and task scheduling for use in distributed training and inference. After that, we present the latest technologies pertaining to modern communication infrastructures used in distributed deep learning with a focus on examining the impact of the communication overhead in a large-scale and heterogeneous setting. Finally, we conduct a case study on the distributed training of large language models at a large scale to illustrate how to apply these technologies in real cases. This article aims to offer researchers a comprehensive understanding of the current landscape of large-scale distributed deep learning and to reveal promising future research directions toward communication-efficient solutions in this scope.

Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Builds on 146 of its 326 references that are in the map. First posted on arXiv 9 Apr 2024.

Builds on

146 of 326 references.

References · 146

  1. 2017Attention Is All You Needcited by 59,932
  2. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  3. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  4. 2014Adam: A Method for Stochastic Optimizationcited by 36,960
  5. 2020Language Models are Few-Shot Learnerscited by 32,199
  6. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  7. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  8. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
  9. 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
  10. 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,304
  11. 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,530
  12. 2020wav2vec 2.0: A Framework for Self-Supervised Learning of Speech Representationscited by 3,853
38 more
  1. 2018Federated Optimization in Heterogeneous Networkscited by 2,971
  2. 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,479
  3. 2022LaMDA: Language Models for Dialog Applicationscited by 1,218
  4. 2023BloombergGPT: A Large Language Model for Financecited by 923
  5. 2020Tackling the Objective Inconsistency Problem in Heterogeneous Federated Optimizationcited by 897
  6. 2022CodeGen: An Open Large Language Model for Code with Multi-Turn Program Synthesiscited by 876
  7. 2019Towards Federated Learning at Scale: System Designcited by 826
  8. 2019Large Batch Optimization for Deep Learning: Training BERT in 76 minutescited by 671
  9. 2021Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LMcited by 621
  10. 2018Local SGD Converges Fast and Communicates Littlecited by 595
  11. 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 561
  12. 2016DoReFa-Net: Training Low Bitwidth Convolutional Neural Networks with Low Bitwidth Gradientscited by 557
  13. 2022Using DeepSpeed and Megatron to Train Megatron-Turing NLG 530B, A Large-Scale Generative Language Modelcited by 557
  14. 2016Linear Convergence of Gradient and Proximal-Gradient Methods Under the Polyak-Lojasiewicz Conditioncited by 526
  15. 2017Can Decentralized Algorithms Outperform Centralized Algorithms? A Case Study for Decentralized Parallel Stochastic Gradient Descentcited by 496
  16. 2017Deep Gradient Compression: Reducing the Communication Bandwidth for Distributed Trainingcited by 471
  17. 2019Robust and Communication-Efficient Federated Learning from Non-IID Datacited by 415
  18. 2016Quantized Neural Networks: Training Neural Networks with Low Precision Weights and Activationscited by 413
  19. 2018Adaptive Federated Learning in Resource Constrained Edge Computing Systemscited by 408
  20. 2018Sparsified SGD with Memorycited by 373
  21. 2015Deep Learning with Limited Numerical Precisioncited by 368
  22. 2019Error Feedback Fixes SignSGD and other Gradient Compression Schemescited by 306
  23. 2018Parallel Restarted SGD with Faster Convergence and Less Communication: Demystifying Why Model Averaging Works for Deep Learningcited by 277
  24. 2018Characterizing Implicit Bias in Terms of Optimization Geometrycited by 259
  25. 2017TernGrad: Ternary Gradients to Reduce Communication in Distributed Deep Learningcited by 253
  26. 2017Sparse Communication for Distributed Gradient Descentcited by 243
  27. 2019PowerSGD: Practical Low-Rank Gradient Compression for Distributed Optimizationcited by 234
  28. 2018The Convergence of Sparsified Gradient Methodscited by 228
  29. 2019Edge Intelligence: Paving the Last Mile of Artificial Intelligence with Edge Computingcited by 225
  30. 2018Don't Use Large Mini-Batches, Use Local SGDcited by 201
  31. 2020ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learningcited by 201
  32. 2017Gradient Sparsification for Communication-Efficient Distributed Optimizationcited by 199
  33. 2016Clipper: A Low-Latency Online Prediction Serving Systemcited by 191
  34. 2018Measuring the Effects of Data Parallelism on Neural Network Trainingcited by 190
  35. 2019A Survey on Distributed Machine Learningcited by 187
  36. 2019Energy Efficient Federated Learning Over Wireless Communication Networkscited by 181
  37. 2019Qsparse-local-SGD: Distributed SGD with Quantization, Sparsification, and Local Computationscited by 179
  38. 2021Training Graph Neural Networks with 1000 Layerscited by 169

50 most cited of 146 All 146 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 2; 2025: 4; 2026: 3.

9 papers.