arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2203.02645cs.LGcs.AI
收藏

在本地训练中加速联邦学习并减轻遗忘

Acceleration of Federated Learning with Alleviated Forgetting in Local Training

Tsinghua University(清华大学);BNRIST(北京信息科学与技术国家研究中心)

Chencheng Xu, Zhiwei Hong, Minlie Huang, Tao Jiang

摘要:

联邦学习(FL)通过在每个客户端上独立训练本地模型,然后在中央服务器上聚合参数,从而产生有效的全局模型,从而在保护隐私的同时实现机器学习模型的分布式优化。虽然已经提出了各种FL算法,但是当数据不是独立同分布(非i.i.d.)时,它们的训练效率仍然很低。在不同的客户。我们观察到,现有方法的缓慢收敛速度是(至少部分地)由每个单独客户端上的本地训练阶段期间的灾难性遗忘问题引起的,这导致关于其他客户端处的先前训练数据的损失函数的大幅增加。在这里,我们提出了FedReg,这是一种加速FL的算法,通过在生成的伪数据丢失的情况下正则化本地训练参数,从而减轻局部训练阶段的知识遗忘,其中伪数据对全局模型学习到的先前训练数据的知识进行编码。我们的综合实验表明,FedReg不仅显著提高了FL的收敛速度,特别是当神经网络架构较深且客户端数据极不一致时,而且能够在分类问题中更好地保护隐私,并且对梯度反转攻击更鲁棒。该代码可从以下网址获取:https://github.com/Zoesgithub/FedReg。

英文摘要:

Federated learning (FL) enables distributed optimization of machine learning models while protecting privacy by independently training local models on each client and then aggregating parameters on a central server, thereby producing an effective global model. Although a variety of FL algorithms have been proposed, their training efficiency remains low when the data are not independently and identically distributed (non-i.i.d.) across different clients. We observe that the slow convergence rates of the existing methods are (at least partially) caused by the catastrophic forgetting issue during the local training stage on each individual client, which leads to a large increase in the loss function concerning the previous training data at the other clients. Here, we propose FedReg, an algorithm to accelerate FL with alleviated knowledge forgetting in the local training stage by regularizing locally trained parameters with the loss on generated pseudo data, which encode the knowledge of previous training data learned by the global model. Our comprehensive experiments demonstrate that FedReg not only significantly improves the convergence rate of FL, especially when the neural network architecture is deep and the clients' data are extremely non-i.i.d., but is also able to protect privacy better in classification problems and more robust against gradient inversion attacks. The code is available at: https://github.com/Zoesgithub/FedReg.

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

↑