arrow
Return

PHY: A performance-driven hybrid communication compression method for distributed training

delete2023-10-01
delete1
PRE
AI
C
Chen-Chun Chen
Y
Yu-Min Chou
J
Jerry Chou *
DOI:10.1016/j.jpdc.2023.104719delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Distributed training is needed to shorten the training time of deep neural networks. However, the communication overhead often hurts performance efficiency, especially in a distributed computing environment with limited network bandwidth. Hence, gradient compression techniques have been proposed to reduce communication time. But, compression also has the risk of causing lower model accuracy and longer training time due to compression loss and compression time. As a result, compression may not consistently achieve desired results, and there are limited discussions on when and which compression should be used. To address this problem, we propose a performance-driven hybrid compression solution. We make three main contributions. (1) We describe a hybrid compression strategy that chooses the compression method for individual model gradients. (2) We build an offline performance estimator and an online loss monitor to ensure the compression decision can minimize training time without sacrificing mode accuracy. (3) Our implementation can be imported to existing deep learning frameworks and applicable to a wide range of compression methods. Up to 3.6x training performance speedup was observed compared to other state-of-the-art methods.& COPY; 2023 Elsevier Inc. All rights reserved.
Keywords:
Deep learning
Distributed computing
Data compression
Performance
Optimization

Journal

Journal of Parallel and Distributed Computing cover
Journal of Parallel and Distributed Computing
IF:
4
Papers:
3.8K
Citations:
4.8K

Organization

No organization information available
Cited Papers

Cited Papers

Hardware-Centric AutoML for Mixed-Precision Quantization
err2020-06-11
err8
errOAAI
errWang, Kuan; Liu, Zhijian; Lin, Yujun; Lin, Ji; Han, Song
errShare
errSave
ImageNet Large Scale Visual Recognition Challenge
err2015-04-11
err2.7W
PREAI
errRussakovsky, Olga; Deng, Jia; Su, Hao; Krause, Jonathan; Satheesh, Sanjeev; Ma, Sean; Huang, Zhiheng; Karpathy, Andrej; Khosla, Aditya; Bernstein, Michael; Berg, Alexander C.; Fei-Fei, Li
errShare
errSave
Deep Neural Networks for Acoustic Modeling in Speech Recognition
err2012-11-01
err8.2K
PREAI
errHinton, Geoffrey; Deng, Li; Yu, Dong; Dahl, George E.; Mohamed, Abdel-rahman; Jaitly, Navdeep; Senior, Andrew; Vanhoucke, Vincent; Patrick Nguyen; Sainath, Tara N.; Kingsbury, Brian
errShare
errSave
Porous Materials for Applications in Energy and Environment
err2021-07-11
err0
PREAI
errTzipatly A. Esquivel-Castro; Antonia Martínez-Luévanos; Sofía Estrada-Flores; Lucía F. Cano-Salazar
errShare
errSave
no more