Federated Learning with GAN-Based Data Synthesis for Non-IID Clients

Zijian Li, Jiawei Shao, Yuyi Mao, Jessie Hui Wang, Jun Zhang

Research output: Chapter in book / Conference proceedingConference article published in proceeding or bookAcademic researchpeer-review

7 Citations (Scopus)

Abstract

Federated learning (FL) has recently emerged as a popular privacy-preserving collaborative learning paradigm. However, it suffers from the non-independent and identically distributed (non-IID) data among clients. In this chapter, we propose a novel framework, named Synthetic Data Aided Federated Learning (SDA-FL), to resolve this non-IID challenge by sharing synthetic data. Specifically, each client pretrains a local generative adversarial network (GAN) to generate differentially private synthetic data, which are uploaded to the parameter server (PS) to construct a global shared synthetic dataset. To generate confident pseudo labels for the synthetic dataset, we also propose an iterative pseudo labeling mechanism performed by the PS. The assistance of the synthetic dataset with confident pseudo labels significantly alleviates the data heterogeneity among clients, which improves the consistency among local updates and benefits the global aggregation. Extensive experiments evidence that the proposed framework outperforms the baseline methods by a large margin in several benchmark datasets under both the supervised and semi-supervised settings.

Original languageEnglish
Title of host publicationTrustworthy Federated Learning - First International Workshop, FL 2022, Held in Conjunction with IJCAI 2022, Revised Selected Papers
EditorsRandy Goebel, Han Yu, Boi Faltings, Lixin Fan, Zehui Xiong
PublisherSpringer Science and Business Media Deutschland GmbH
Pages17-32
Number of pages16
ISBN (Print)9783031289958
DOIs
Publication statusPublished - 2023
Event1st International Workshop on Trustworthy Federated Learning in Conjunction with International Joint Conference on AI, FL-IJCAI 2022 - Vienna, Austria
Duration: 23 Jul 202223 Jul 2022

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume13448 LNAI
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference1st International Workshop on Trustworthy Federated Learning in Conjunction with International Joint Conference on AI, FL-IJCAI 2022
Country/TerritoryAustria
CityVienna
Period23/07/2223/07/22

Keywords

  • Federated Learning
  • Generative Adversarial Network (GAN)
  • Non-Independent and Identically Distributed (non-IID) Problem

ASJC Scopus subject areas

  • Theoretical Computer Science
  • General Computer Science

Fingerprint

Dive into the research topics of 'Federated Learning with GAN-Based Data Synthesis for Non-IID Clients'. Together they form a unique fingerprint.

Cite this