Communication-Efficient Distributed Learning with Differential Privacy
Authors: Xiaoxing Ren, Yuwen Ma, Nicola Bastianello, Karl H. Johansson, Thomas Parisini, Andreas A. Malikopoulos
Organizations: School of Civil and Environmental Engineering, Cornell University, Ithaca, New York, United States. · Department of Electronic and Electrical Engineering, University College London, London WC1E 7JE, United Kingdom. · School of Electrical Engineering and Computer Science, and Digital Futures, KTH Royal Institute of Technology, Stockholm, Sweden. · Department of Electrical and Electronic Engineering, Imperial College London, London, United Kingdom · Department of Electronic Systems, Aalborg University, Denmark · Department of Engineering and Architecture, University of Trieste, Trieste, Italy. · Applied Mathematics, Systems Engineering, Mechanical Engineering, Electrical & Computer Engineering, Cornell University, Ithaca, New York, United States.
We address nonconvex learning problems over undirected networks. In particular, we focus on the challenge of designing an algorithm that is both communication-efficient and that guarantees the privacy of the agents' data. The first goal is achieved through a local training approach, which reduces communication frequency. The second goal is achieved by perturbing gradients during local training, specifically through gradient clipping and additive noise. We prove that the resulting algorithm converges to a stationary point of the problem within a bounded distance. Additionally, we provide theoretical privacy guarantees within a differential privacy framework that ensure agents' training data cannot be inferred from the trained model shared over the network. We show the algorithm's superior performance on a classification task under the same privacy budget, compared with state-of-the-art methods.
Figures & tables
Algorithm [Ref.]
Time
PORTER [ 21 ]
τ(tg+2tc)
PriSMA [ 22 ]
τ(2tg+tc)
LT-ADMM-DP
τtg+tc
TABLE I : Computation time over τ iterations.
Fig. 1 : Numerical results for three algorithms with differential privacy.
Communication-efficient distributed training reduces the data exchanged among devices so models can be trained faster and over limited-bandwidth networks.