跳到主要导航 跳到搜索 跳到主要内容

DFDG: Data-Free Dual-Generator Adversarial Distillation for One-Shot Federated Learning

  • East China Normal University
  • Naval Medical University
  • Shanghai Normal University

科研成果: 书/报告/会议事项章节会议稿件同行评审

摘要

Federated Learning (FL) is a distributed machine learning scheme in which clients jointly participate in the collaborative training of a global model by sharing model information rather than their private datasets. In light of concerns associated with communication and privacy, one-shot FL with a single communication round has emerged as a de facto promising solution. However, existing one-shot FL methods either require public datasets, focus on model homogeneous settings, or distill limited knowledge from local models, making it difficult or even impractical to train a robust global model. To address these limitations, we propose a new data-free dual-generator adversarial distillation method (namely DFDG) for one-shot FL, which can explore a broader local models' training space via training dual generators. DFDG is executed in an adversarial manner and comprises two parts: dual-generator training and dual-model distillation. In dual-generator training, we delve into each generator concerning fidelity, transferability and diversity to ensure its utility, and additionally tailor the cross-divergence loss to lessen the overlap of dual generators' output spaces. In dual-model distillation, the trained dual generators work together to provide the training data for updates of the global model. At last, our extensive experiments on various image classification tasks show that DFDG achieves significant performance gains in accuracy compared to SOTA baselines. We provide our code here: https://anonymous.4open.science/r/DFDG-7BDB.

源语言英语
主期刊名Proceedings - 24th IEEE International Conference on Data Mining, ICDM 2024
编辑Elena Baralis, Kun Zhang, Ernesto Damiani, Merouane Debbah, Panos Kalnis, Xindong Wu
出版商Institute of Electrical and Electronics Engineers Inc.
281-290
页数10
ISBN(电子版)9798331506681
DOI
出版状态已出版 - 2024
活动24th IEEE International Conference on Data Mining, ICDM 2024 - Abu Dhabi, 阿拉伯联合酋长国
期限: 9 12月 202412 12月 2024

出版系列

姓名Proceedings - IEEE International Conference on Data Mining, ICDM
ISSN(印刷版)1550-4786

会议

会议24th IEEE International Conference on Data Mining, ICDM 2024
国家/地区阿拉伯联合酋长国
Abu Dhabi
时期9/12/2412/12/24

指纹

探究 'DFDG: Data-Free Dual-Generator Adversarial Distillation for One-Shot Federated Learning' 的科研主题。它们共同构成独一无二的指纹。

引用此