Skip to main content

Federated Learning With Differential Privacy: Algorithms and Performance Analysis

Author(s): Wei, Kang; Li, Jun; Ding, Ming; Ma, Chuan; Yang, Howard H; et al

To refer to this page use:
Abstract: Federated learning (FL), as a type of distributed machine learning, is capable of significantly preserving clients’ private data from being exposed to adversaries. Nevertheless, private information can still be divulged by analyzing uploaded parameters from clients, e.g., weights trained in deep neural networks. In this paper, to effectively prevent information leakage, we propose a novel framework based on the concept of differential privacy (DP), in which artificial noise is added to parameters at the clients’ side before aggregating, namely, noising before model aggregation FL (NbAFL). First, we prove that the NbAFL can satisfy DP under distinct protection levels by properly adapting different variances of artificial noise. Then we develop a theoretical convergence bound on the loss function of the trained FL model in the NbAFL. Specifically, the theoretical bound reveals the following three key properties: 1) there is a tradeoff between convergence performance and privacy protection levels, i.e., better convergence performance leads to a lower protection level; 2) given a fixed privacy protection level, increasing the number $N$ of overall clients participating in FL can improve the convergence performance; and 3) there is an optimal number aggregation times (communication rounds) in terms of convergence performance for a given protection level. Furthermore, we propose a $K$ -client random scheduling strategy, where $K$ ( $1\leq K< N$ ) clients are randomly selected from the $N$ overall clients to participate in each aggregation. We also develop a corresponding convergence bound for the loss function in this case and the $K$ -client random scheduling strategy also retains the above three properties. Moreover, we find that there is an optimal $K$ that achieves the best convergence performance at a fixed privacy level. Evaluations demonstrate that our theoretical results are consistent with simulations, thereby facilitating the design of various privacy-preserving FL algorithms with different tradeoff requirements on convergence performance and privacy levels.
Publication Date: 17-Apr-2020
Citation: Wei, Kang, Li, Jun, Ding, Ming, Ma, Chuan, Yang, Howard H, Farokhi, Farhad, Jin, Shi, Quek, Tony QS, Vincent Poor, H. (2020). Federated Learning With Differential Privacy: Algorithms and Performance Analysis. IEEE Transactions on Information Forensics and Security, 15 (3454 - 3469. doi:10.1109/tifs.2020.2988575
DOI: doi:10.1109/tifs.2020.2988575
ISSN: 1556-6013
EISSN: 1556-6021
Pages: 3454 - 3469
Type of Material: Journal Article
Journal/Proceeding Title: IEEE Transactions on Information Forensics and Security
Version: Author's manuscript

Items in OAR@Princeton are protected by copyright, with all rights reserved, unless otherwise indicated.