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

利用预训练扩散模型探索一次性半监督联邦学习

Exploring One-shot Semi-supervised Federated Learning with A Pre-trained Diffusion Model

Mingzhao Yang, Shangchao Su, Bin Li, Xiangyang Xue

AI总结:

本文提出 FedDISC,利用预训练扩散模型在单轮通信内实现半监督联邦学习,通过原型伪标签和领域表示生成合成数据训练全局模型,无需客户端本地训练,性能超越现有 SOTA 方法。

摘要:

近年来,半监督联邦学习(semi-FL)被提出用于处理现实世界中常见的场景:服务器上拥有带标签数据,而客户端上只有无标签数据。然而,现有方法面临通信成本高、数据异构性强以及客户端设备训练压力大等挑战。为了解决这些问题,我们将强大的扩散模型(DM)引入半监督联邦学习,并提出 FedDISC,一种联邦扩散启发的半监督协同训练方法。具体而言,我们首先提取带标签服务器数据的原型,并利用这些原型为客户端数据预测伪标签。对于每个类别,我们计算聚类中心和领域特定表示,以表征其分布在语义和风格上的信息。加入噪声后,这些表示被发送回服务器,服务器利用预训练的扩散模型生成符合客户端分布的合成数据集,并在其上训练全局模型。借助扩散模型中蕴含的大量知识,合成数据集具有与客户端图像相当的质量和多样性,从而使训练出的全局模型能够达到甚至超越有监督集中式训练的性能上限。FedDISC 仅需一轮通信即可完成,不需要任何本地训练,并且只涉及极少量的信息上传,极大地增强了其实用性。在三个大规模数据集上的大量实验表明,FedDISC 有效解决了非独立同分布(non-IID)客户端上的半监督联邦学习问题,并优于对比的最先进(SOTA)方法。充分的可视化实验也表明,FedDISC 生成的合成数据集在多样性和质量上与原始客户端数据集相当,且泄露客户端隐私敏感信息的可能性可以忽略不计。

英文摘要:

Recently, semi-supervised federated learning (semi-FL) has been proposed to handle the commonly seen real-world scenarios with labeled data on the server and unlabeled data on the clients. However, existing methods face several challenges such as communication costs, data heterogeneity, and training pressure on client devices. To address these challenges, we introduce the powerful diffusion models (DM) into semi-FL and propose FedDISC, a Federated Diffusion-Inspired Semi-supervised Co-training method. Specifically, we first extract prototypes of the labeled server data and use these prototypes to predict pseudo-labels of the client data. For each category, we compute the cluster centroids and domain-specific representations to signify the semantic and stylistic information of their distributions. After adding noise, these representations are sent back to the server, which uses the pre-trained DM to generate synthetic datasets complying with the client distributions and train a global model on it. With the assistance of vast knowledge within DM, the synthetic datasets have comparable quality and diversity to the client images, subsequently enabling the training of global models that achieve performance equivalent to or even surpassing the ceiling of supervised centralized training. FedDISC works within one communication round, does not require any local training, and involves very minimal information uploading, greatly enhancing its practicality. Extensive experiments on three large-scale datasets demonstrate that FedDISC effectively addresses the semi-FL problem on non-IID clients and outperforms the compared SOTA methods. Sufficient visualization experiments also illustrate that the synthetic dataset generated by FedDISC exhibits comparable diversity and quality to the original client dataset, with a neglectable possibility of leaking privacy-sensitive information of the clients.

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

↑