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

FedDC: Federated Learning with Non-IID Data via Local Drift Decoupling and Correction

National University of Defense Technology(国防科技大学);IHPC, ASTAR(新加坡科技研究局高性能计算研究所);University of Macau(澳门大学)

Liang Gao, Huazhu Fu, Li Li, Yingwen Chen, Ming Xu, Cheng-Zhong Xu

英文摘要:

Federated learning (FL) allows multiple clients to collectively train a high-performance global model without sharing their private data. However, the key challenge in federated learning is that the clients have significant statistical heterogeneity among their local data distributions, which would cause inconsistent optimized local models on the client-side. To address this fundamental dilemma, we propose a novel federated learning algorithm with local drift decoupling and correction (FedDC). Our FedDC only introduces lightweight modifications in the local training phase, in which each client utilizes an auxiliary local drift variable to track the gap between the local model parameter and the global model parameters. The key idea of FedDC is to utilize this learned local drift variable to bridge the gap, i.e., conducting consistency in parameter-level. The experiment results and analysis demonstrate that FedDC yields expediting convergence and better performance on various image classification tasks, robust in partial participation settings, non-iid data, and heterogeneous clients.

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

↑