arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.19953cs.CV
收藏

FedDEO:基于扩散模型的描述增强一次性联邦学习

FedDEO: Description-Enhanced One-Shot Federated Learning with Diffusion Models

School of Computer Science, Fudan University(复旦大学计算机科学学院)

Mingzhao Yang, Shangchao Su, Bin Li, Xiangyang Xue

AI总结:

提出FedDEO,通过在客户端训练本地描述作为知识传输媒介,指导服务器端扩散模型生成合成数据以训练聚合模型,在通信与隐私保护上优于现有方法。

摘要:

近年来,一次性联邦学习(OSFL)受到的关注由其最小化通信的能力所驱动。随着扩散模型(DM)的发展,几种方法将DM用于OSFL,利用模型参数、图像特征或文本提示作为媒介,将本地客户端知识传输到服务器。然而,这些媒介通常需要公共数据集或统一特征提取器,显著限制了其实用性。在本文中,我们提出FedDEO,一种基于DM的描述增强一次性联邦学习方法,为在OSFL中利用DM提供了新颖探索。我们方法的核心思想涉及在客户端上训练本地描述,作为将分布式客户端知识传输到服务器的媒介。首先,我们在客户端数据上训练本地描述以捕获客户端分布的特征,随后将其上传至服务器。在服务器上,描述被用作条件以指导DM生成符合各种客户端分布的合成数据集,从而实现聚合模型的训练。在三个大规模真实世界数据集上的理论分析及充分的定量和可视化实验表明,通过训练本地描述,服务器能够生成具有高质量和多样性的合成数据集。因此,凭借在通信和隐私保护方面的优势,聚合模型相比FL或基于扩散的OSFL方法表现更优,且在某些客户端上超越了集中式训练的性能上限。

英文摘要:

In recent years, the attention towards One-Shot Federated Learning (OSFL) has been driven by its capacity to minimize communication. With the development of the diffusion model (DM), several methods employ the DM for OSFL, utilizing model parameters, image features, or textual prompts as mediums to transfer the local client knowledge to the server. However, these mediums often require public datasets or the uniform feature extractor, significantly limiting their practicality. In this paper, we propose FedDEO, a Description-Enhanced One-Shot Federated Learning Method with DMs, offering a novel exploration of utilizing the DM in OSFL. The core idea of our method involves training local descriptions on the clients, serving as the medium to transfer the knowledge of the distributed clients to the server. Firstly, we train local descriptions on the client data to capture the characteristics of client distributions, which are then uploaded to the server. On the server, the descriptions are used as conditions to guide the DM in generating synthetic datasets that comply with the distributions of various clients, enabling the training of the aggregated model. Theoretical analyses and sufficient quantitation and visualization experiments on three large-scale real-world datasets demonstrate that through the training of local descriptions, the server is capable of generating synthetic datasets with high quality and diversity. Consequently, with advantages in communication and privacy protection, the aggregated model outperforms compared FL or diffusion-based OSFL methods and, on some clients, outperforms the performance ceiling of centralized training.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑