Communication efficient decentralized learning over bipartite graphs
Ben Issaid, Chaouki; Elgabli, Anis; Park, Jihong; Bennis, Mehdi; Debbah, Mérouane (2021-11-17)
C. Ben Issaid, A. Elgabli, J. Park, M. Bennis and M. Debbah, "Communication Efficient Decentralized Learning Over Bipartite Graphs," in IEEE Transactions on Wireless Communications, vol. 21, no. 6, pp. 4150-4167, June 2022, doi: 10.1109/TWC.2021.3126859.
© 2021 IEEE. This work is licensed under a Creative Commons Attribution 4.0 License. For more information, see https://creativecommons.org/licenses/by/4.0/
https://creativecommons.org/licenses/by/4.0/
https://urn.fi/URN:NBN:fi-fe2022020717882
Tiivistelmä
Abstract
In this paper, we propose a communication-efficiently decentralized machine learning framework that solves a consensus optimization problem defined over a network of inter-connected workers. The proposed algorithm, Censored and Quantized Generalized GADMM (CQ-GGADMM), leverages the worker grouping and decentralized learning ideas of Group Alternating Direction Method of Multipliers (GADMM), and pushes the frontier in communication efficiency by extending its applicability to generalized network topologies, while incorporating link censoring for negligible updates after quantization. We theoretically prove that CQ-GGADMM achieves the linear convergence rate when the local objective functions are strongly convex under some mild assumptions. Numerical simulations corroborate that CQ-GGADMM exhibits higher communication efficiency in terms of the number of communication rounds and transmit energy consumption without compromising the accuracy and convergence speed, compared to the censored decentralized ADMM, and the worker grouping method of GADMM.
Kokoelmat
- Avoin saatavuus [32130]