arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.01548cs.LGcs.AIcs.DC
收藏

PFA:保护隐私的联邦自适应有效模型个性化

PFA: Privacy-preserving Federated Adaptation for Effective Model Personalization

Peking University(北京大学)

Bingyan Liu, Yao Guo, Xiangqun Chen

摘要:

联邦学习(FL)已经成为一种流行的分布式机器学习范例,具有改进的隐私性。在学习之后,应该针对每个不同的客户机进一步个性化所得到的联邦模型。虽然已经提出了几种方法来实现个性化,但它们通常限于单个本地设备,这可能导致偏差或过拟合,因为单个设备中的数据非常有限。在本文中,我们试图实现超越单一客户端的个性化。其动机是,在FL期间,可能存在许多具有相似数据分布的客户端,因此,如果这些相似的客户端可以相互协作,则可以显著提高个性化性能。受此启发,本文引入了一个新的概念,称为联邦自适应,旨在以联邦的方式适应训练模型,以实现更好的个性化效果。然而,联邦自适应的关键挑战是,由于隐私问题,我们在自适应期间不能从客户端外包任何原始数据。在本文中,我们提出了PFA,一个框架来实现隐私保护联邦适应。PFA利用神经网络的稀疏性来生成隐私保护表示,并使用它们来有效地识别具有相似数据分布的客户端。根据分组结果,PFA对联邦模型进行分组FL过程以完成自适应。为了进行评估,我们手动构建了几个实际的FL数据集的基础上,公共数据集,以模拟类不平衡和背景差异的条件。在这些数据集和流行的模型架构上进行的大量实验证明了PFA的有效性,在确保用户隐私的同时,大大优于其他最先进的方法。我们将在https://github.com/lebyni/PFA上发布我们的代码。

英文摘要:

Federated learning (FL) has become a prevalent distributed machine learning paradigm with improved privacy. After learning, the resulting federated model should be further personalized to each different client. While several methods have been proposed to achieve personalization, they are typically limited to a single local device, which may incur bias or overfitting since data in a single device is extremely limited. In this paper, we attempt to realize personalization beyond a single client. The motivation is that during FL, there may exist many clients with similar data distribution, and thus the personalization performance could be significantly boosted if these similar clients can cooperate with each other. Inspired by this, this paper introduces a new concept called federated adaptation, targeting at adapting the trained model in a federated manner to achieve better personalization results. However, the key challenge for federated adaptation is that we could not outsource any raw data from the client during adaptation, due to privacy concerns. In this paper, we propose PFA, a framework to accomplish Privacy-preserving Federated Adaptation. PFA leverages the sparsity property of neural networks to generate privacy-preserving representations and uses them to efficiently identify clients with similar data distributions. Based on the grouping results, PFA conducts an FL process in a group-wise way on the federated model to accomplish the adaptation. For evaluation, we manually construct several practical FL datasets based on public datasets in order to simulate both the class-imbalance and background-difference conditions. Extensive experiments on these datasets and popular model architectures demonstrate the effectiveness of PFA, outperforming other state-of-the-art methods by a large margin while ensuring user privacy. We will release our code at: https://github.com/lebyni/PFA.

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

↑