Q-Gadmm: Quantized Group Admm For Communication Efficient Decentralized Machine Learning
Anis Elgabli, Jihong Park, Amrit S. Bedi, Mehdi Bennis, Vaneet Aggarwal
-
SPS
IEEE Members: $11.00
Non-members: $15.00Length: 14:54
In this paper, we propose a communication-efficient decentralized machine learning (ML) algorithm, coined quantized group ADMM (Q-GADMM). Every worker in Q-GADMM communicates only with two neighbors, and updates its model via the group alternating direct method of multiplier (GADMM), thereby ensuring fast convergence while reducing the number of communication rounds. Furthermore, each worker quantizes its model updates before transmissions, thereby decreasing the communication payload sizes. We prove that Q-GADMM converges to the optimal solution for convex loss functions, and numerically show that Q-GADMM yields $7$x less communication cost while achieving almost the same accuracy and convergence speed compared to GADMM without quantization.