arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2106.13239cs.LGcs.AIcs.CVcs.DC
收藏

联合噪声客户端学习

Federated Noisy Client Learning

University of Macau(澳门大学);Hong Kong Baptist University(香港浸会大学);Agency for Science, Technology and Research (A*STAR)(新加坡科技研究局)

Kahou Tam, Li Li, Bo Han, Chengzhong Xu, Huazhu Fu

摘要:

联邦学习(FL)根据多个本地客户端协作训练共享的全局模型,同时保持训练数据分散,以保护数据隐私。然而,标准FL方法忽略了嘈杂的客户端问题,这可能会损害共享模型的整体性能。我们首先调查FL中嘈杂客户端引起的关键问题,并根据不同层学习的表示来量化嘈杂客户端的负面影响。我们有以下两个关键观察结果:(1)嘈杂的客户端可能严重影响FL中全局模型的收敛性和性能,以及(2)嘈杂的客户端可能会在全局模型的较深层中引起比前几层更大的偏差。基于上述观察,我们提出了Fed-NCL,一个框架,进行鲁棒的联邦学习与嘈杂的客户端。具体地说,Fed-NCL首先通过估计数据质量和模型分歧来识别噪声客户端。然后,鲁棒分层聚合提出了自适应聚合每个客户端的本地模型,以处理噪声客户端造成的数据异构性。我们进一步对噪声客户端执行标签校正,以提高全局模型的泛化能力。在各种数据集上的实验结果表明,我们的算法提高了不同的国家的最先进的系统与嘈杂的客户端的性能。我们的代码可在https://github.com/TKH666/Fed-NCL上获得

英文摘要:

Federated learning (FL) collaboratively trains a shared global model depending on multiple local clients, while keeping the training data decentralized in order to preserve data privacy. However, standard FL methods ignore the noisy client issue, which may harm the overall performance of the shared model. We first investigate critical issue caused by noisy clients in FL and quantify the negative impact of the noisy clients in terms of the representations learned by different layers. We have the following two key observations: (1) the noisy clients can severely impact the convergence and performance of the global model in FL, and (2) the noisy clients can induce greater bias in the deeper layers than the former layers of the global model. Based on the above observations, we propose Fed-NCL, a framework that conducts robust federated learning with noisy clients. Specifically, Fed-NCL first identifies the noisy clients through well estimating the data quality and model divergence. Then robust layer-wise aggregation is proposed to adaptively aggregate the local models of each client to deal with the data heterogeneity caused by the noisy clients. We further perform the label correction on the noisy clients to improve the generalization of the global model. Experimental results on various datasets demonstrate that our algorithm boosts the performances of different state-of-the-art systems with noisy clients. Our code is available on https://github.com/TKH666/Fed-NCL

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

↑