arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2011.01783cs.LGcs.DC

一种具有公平性保证的联邦学习效率提升客户端选择方案

An Efficiency-boosting Client Selection Scheme for Federated Learning with Fairness Guarantee

South China University of Technology(华南理工大学);the University of Warwick(华威大学);State University of New York, New Paltz(纽约州立大学新帕尔兹分校);The University of Sydney(悉尼大学)

Tiansheng Huang, Weiwei Lin, Wentai Wu, Ligang He, Keqin Li, Albert Y. Zomaya

AI总结:

针对联邦学习中客户端选择影响效率与公平性的问题,提出基于李雅普诺夫优化和C2MAB的RBCS-F算法,以估计模型交换时间并保证公平,其遗憾有界且经实验验证。

摘要:

集中式人工智能模型训练过程中潜在的隐私泄露问题已引起公众的深切关注。一种被称为联邦学习(FL)的并行与分布式计算(PDC)方案应运而生,它作为一种新的范式,通过允许客户端在本地进行模型训练,无需上传个人敏感数据,从而应对隐私问题。在联邦学习中,客户端的数量可能非常庞大,但用于模型分发和重新上传的带宽却相当有限,因此只让部分志愿者参与训练过程是明智之举。客户端选择策略对于联邦学习过程中的训练效率、最终模型质量以及公平性至关重要。在本文中,我们将保证公平性的客户端选择建模为李雅普诺夫优化问题,并提出一种基于C2MAB的方法来估计每个客户端与服务器之间的模型交换时间,在此基础上,我们设计了一种名为RBCS-F的保证公平性的算法来解决问题。RBCS-F的遗憾被严格限制在一个有限常数内,这证明了其理论可行性。除了理论结果之外,我们在公共数据集上的实际训练实验还可以得出更多经验数据。

英文摘要:

The issue of potential privacy leakage during centralized AI's model training has drawn intensive concern from the public. A Parallel and Distributed Computing (or PDC) scheme, termed Federated Learning (FL), has emerged as a new paradigm to cope with the privacy issue by allowing clients to perform model training locally, without the necessity to upload their personal sensitive data. In FL, the number of clients could be sufficiently large, but the bandwidth available for model distribution and re-upload is quite limited, making it sensible to only involve part of the volunteers to participate in the training process. The client selection policy is critical to an FL process in terms of training efficiency, the final model's quality as well as fairness. In this paper, we will model the fairness guaranteed client selection as a Lyapunov optimization problem and then a C2MAB-based method is proposed for estimation of the model exchange time between each client and the server, based on which we design a fairness guaranteed algorithm termed RBCS-F for problem-solving. The regret of RBCS-F is strictly bounded by a finite constant, justifying its theoretical feasibility. Barring the theoretical results, more empirical data can be derived from our real training experiments on public datasets.

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

↑