Communication Optimization Strategies for Distributed Deep Learning: A Survey
Preprint
Abstract
Recent trends in high-performance computing and deep learning lead to a proliferation of studies on large-scale deep neural network (DNN) training. However, the frequent communication requirements among computation nodes drastically slow down the overall training speed, which makes the bottleneck in distributed training, particularly in clusters with limited network bandwidth. To mitigate the drawbacks of distributed communication, researchers have proposed various optimization strategies. In this paper, we give a comprehensive survey of communication strategies from both algorithm and computer network perspectives. Algorithm optimizations focus on reducing the amount of communication in distributed training, while network optimizations focus on speeding up the communication between distributed devices. At the algorithm level, we describe how to reduce the number of communication rounds and transmitted bits per round, besides we shed light on how to overlap computation and communication. At the network level, we discuss the effect caused by network infrastructures, including communication schemes, network protocols, and topology. Finally, we extrapolate potential challenges and research directions for communication acceleration in distributed DNN training.Keywords
All Related Versions
This publication has 12 references indexed in Scilit:
- A Hitchhiker’s Guide On Distributed Training Of Deep Neural NetworksJournal of Parallel and Distributed Computing, 2020
- A survey of techniques for optimizing deep learning on GPUsJournal of Systems Architecture, 2019
- Fast Deep Neural Network Training on Distributed Systems and Cloud TPUsIEEE Transactions on Parallel and Distributed Systems, 2019
- Improving the Performance of Distributed MXNet with RDMAInternational Journal of Parallel Programming, 2019
- Improving the Performance of Distributed TensorFlow with RDMAInternational Journal of Parallel Programming, 2017
- Bandwidth optimal all-reduce algorithms for clusters of workstationsJournal of Parallel and Distributed Computing, 2009
- A scalable, commodity data center network architectureACM SIGCOMM Computer Communication Review, 2008
- Transmission of IP over InfiniBand (IPoIB)2006
- Gradient-based learning applied to document recognitionProceedings of the IEEE, 1998
- Universal codeword sets and representations of the integersIEEE Transactions on Information Theory, 1975