arrow
Return

Robust communication-efficient decentralized learning with heterogeneity

delete2023-08-01
delete3
PRE
AI
X
Xiao Zhang
王旸旸 cover
王旸旸 (Yangyang Wang)
S
Shuzhen Chen
C
Cui Wang
D
Dongxiao Yu *
成秀珍 (Xiuzhen Cheng)
DOI:10.1016/j.sysarc.2023.102900delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
In this paper, we propose a robust communication-efficient decentralized learning algorithm, named RCEDL, to address data heterogeneity, communication heterogeneity and communication efficiency simultaneously in real -world scenarios. To the best of our knowledge, this is the first work to address the above challenges in a united framework. In detail, we design a compressed cross-gradient aggregation mechanism with delay to resolve the Non-IID issues, a blocking-resilient mechanism which allows receiving delayed parameters and gradients, and a communication-efficient mechanism including parameters compression and adaptive neighbors selection methods to reduce the communication cost as much as possible. In addition, we also provide convergence analysis of RCEDL and prove its convergence rate O( & RADIC;1NK ) same with the state-of-the-art decentralized learning algorithms. Finally, we conduct extensive experiments to evaluate RCEDL algorithm on two widely used datasets CIFAR-10 and MNIST under different experimental settings. Compared with the state-of-the-art baseline methods, the proposed RCEDL is much more robust with higher accuracy and at least 3.4 x communication cost reduction under the heterogeneous environment.
Keywords:
Decentralized learning
Communication-efficient
Heterogeneity

Journal

Journal of Systems Architecture cover
Journal of Systems Architecture
IF:
4.1
Papers:
3.0K
Citations:
4.2K

Organization

S
shandong university
Scholars:
9.4W
Papers: 6.4W
Citations: 94