Skip to main content
Article
PerFED-GAN: Personalized Federated Learning via Generative Adversarial Networks
IEEE Internet of Things Journal
  • Xingjian Cao, University of Electronic Science and Technology of China
  • Gang Sun, University of Electronic Science and Technology of China & The Agile and Intelligent Computing Key Laboratory of Sichuan Province, Chengdu, China
  • Hongfang Yu, University of Electronic Science and Technology of China & The Pengcheng Laboratory, Shenzhen, China
  • Mohsen Guizani, Mohamed bin Zayed University of Artificial Intelligence
Document Type
Article
Abstract

Federated learning is gaining popularity as a distributed machine learning method that can be used to deploy AI-dependent IoT applications while protecting client data privacy and security. Due to the differences of clients, a single global model may not perform well on all clients, so the personalized federated learning method, which trains a personalized model for each client that better suits its individual needs, becomes a research hotspot. Most personalized federated learning research, however, focuses on data heterogeneity while ignoring the need for model architecture heterogeneity. Most existing federated learning methods uniformly set the model architecture of all clients participating in federated learning, which is inconvenient for each client’s individual model and local data distribution requirements, and also increases the risk of client model leakage. This paper proposes a federated learning method based on co-training and generative adversarial networks(GANs) that allows each client to design its own model to participate in federated learning training independently without sharing any model architecture or parameter information with other clients or a center. In our experiments, the proposed method outperforms the existing methods in mean test accuracy by 42% when the client’s model architecture and data distribution vary significantly. Copyright © 2022, The Authors. All rights reserved.

DOI
10.1109/JIOT.2022.3172114
Publication Date
5-3-2022
Keywords
  • Data models,
  • Collaborative work,
  • Training,
  • Computer architecture,
  • Task analysis,
  • Machine learning,
  • Computational modeling,
  • federated learning,
  • personalized model,
  • Non-IID data,
  • co-training,
  • generative adversarial networks
Comments

IR Deposit conditions:

OA version (pathway a): Accepted version

No embargo

When accepted for publication, set statement to accompany deposit (see policy)

Must link to publisher version with DOI

Publisher copyright and source must be acknowledged

Citation Information
X. Cao, G. Sun, H. Yu, M. Guizani, "PerFED-GAN: personalized federated learning via generative adversarial networks," in IEEE Internet of Things Journal, p. 1-1, May 2022, doi: 10.1109/JIOT.2022.3172114