ByteComp: Revisiting Gradient Compression in Distributed Training
Not on the map. Why?
Abstract (arXiv)
Gradient compression (GC) is a promising approach to addressing the communication bottleneck in distributed deep learning (DDL). However, it is challenging to find the optimal compression strategy for applying GC to DDL because of the intricate interactions among tensors. To fully unleash the benefits of GC, two questions must be addressed: 1) How to express all compression strategies and the corresponding interactions among tensors of any DDL training job? 2) How to quickly select a near-optimal compression strategy? In this paper, we propose ByteComp to answer these questions. It first designs a decision tree abstraction to express all the compression strategies and develops empirical models to timeline tensor computation, communication, and compression to enable ByteComp to derive the intricate interactions among tensors. It then designs a compression decision algorithm that analyzes tensor interactions to eliminate and prioritize strategies and optimally offloads compression to CPUs. Experimental evaluations show that ByteComp can improve the training throughput over the start-of-the-art compression-enabled system by up to 77% for representative DDL training jobs. Moreover, the computational time needed to select the compression strategy is measured in milliseconds, and the selected strategy is only a few percent from optimal.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 46 of its 76 references that are in the map. First posted on arXiv 28 May 2022.
Builds on
46 of 76 references.
References · 46
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2015Rethinking the Inception Architecture for Computer Visioncited by 5,475
- 2016Pointer Sentinel Mixture Modelscited by 3,096
- 2018Know What You Don't Know: Unanswerable Questions for SQuADcited by 1,554
- 2017In-Datacenter Performance Analysis of a Tensor Processing Unitcited by 618
- 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 571
- 2017Deep Gradient Compression: Reducing the Communication Bandwidth for Distributed Trainingcited by 478
- 2018Sparsified SGD with Memorycited by 376
- 2015Skip-Thought Vectorscited by 322
34 more
- 2019Error Feedback Fixes SignSGD and other Gradient Compression Schemescited by 310
- 2018Horovod: fast and easy distributed deep learning in TensorFlowcited by 279
- 2017TernGrad: Ternary Gradients to Reduce Communication in Distributed Deep Learningcited by 256
- 2017Sparse Communication for Distributed Gradient Descentcited by 249
- 2014cuDNN: Efficient Primitives for Deep Learningcited by 245
- 2017Regularizing and Optimizing LSTM Language Modelscited by 205
- 2017Gradient Sparsification for Communication-Efficient Distributed Optimizationcited by 201
- 2020Ansor: Generating High-Performance Tensor Programs for Deep Learningcited by 201
- 2015MXNet: A Flexible and Efficient Machine Learning Library for Heterogeneous Distributed Systemscited by 195
- 2019U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translationcited by 186
- 2016QSGD: Communication-Efficient SGD via Gradient Quantization and Encodingcited by 147
- 2017Revisiting Distributed Synchronous SGDcited by 142
- 2018Error Compensated Quantized SGD and its Applications to Large-scale Distributed Optimizationcited by 76
- 2018Glow: Graph Lowering Compiler Techniques for Neural Networkscited by 75
- 2019Scaling Distributed Machine Learning with In-Network Aggregationcited by 65
- 2018TVM: An Automated End-to-End Optimizing Compiler for Deep Learningcited by 62
- 2017Poseidon: An Efficient Communication Architecture for Distributed Deep Learning on GPU Clusterscited by 57
- 20158-Bit Approximations for Parallelism in Deep Learningcited by 56
- 2019Communication-Efficient Distributed Blockwise Momentum SGD with Error-Feedbackcited by 50
- 2019Blink: Fast and Generic Collectives for Distributed MLcited by 36
- 2018SparCML: High-Performance Sparse Communication for Machine Learningcited by 34
- 2017AdaComp : Adaptive Residual Gradient Compression for Data-Parallel Distributed Trainingcited by 33
- 2018TicTac: Accelerating Distributed Deep Learning with Communication Schedulingcited by 27
- 2021On the Utility of Gradient Compression in Distributed Training Systemscited by 26
- 2020Towards Scalable Distributed Training of Deep Learning on Public Cloud Clusterscited by 23
- 2019Priority-based Parameter Propagation for Distributed DNN Trainingcited by 21
- 2020Is Network the Bottleneck of Distributed Training?cited by 20
- 2018Parameter Hub: a Rack-Scale Parameter Server for Distributed Deep Neural Network Trainingcited by 17
- 2019CROSSBOW: Scaling Deep Learning with Small Batch Sizes on Multi-GPU Serverscited by 14
- 2019Optimizing Network Performance for Distributed DNN Training on GPU Clusters: ImageNet/AlexNet Training in 1.5 Minutescited by 13
- 2016DLAU: A Scalable Deep Learning Accelerator Unit on FPGAcited by 12
- 2021Compressed Communication for Distributed Training: Adaptive Methods and Systemcited by 5
- 2021MergeComp: A Compression Scheduler for Scalable Communication-Efficient Distributed Trainingcited by 1
- 2021MAPA: Multi-Accelerator Pattern Allocation Policy for Multi-Tenant GPU Serverscited by 1
Built on by
Citing papers per year: 2023: 1.
1 paper.