Federated learning(FL)is an emerging privacy-preserving distributed computing paradigm,enabling numer-ous clients to collaboratively train machine learning models without the necessity of transmitting clients'private datasets to the central server.Unlike most existing research where the local datasets of clients are assumed to be unchanged over time throughout the whole FL process,our study addresses such scenarios in this paper where clients'datasets need to be updated periodically,and the server can incentivize clients to employ as fresh as possible datasets for local model training.Our primary objective is to design a client selection strategy to minimize the loss of the global model for FL loss within a constrained budget.To this end,we introduce the concept of"Age of Information"(AoI)to quantitatively assess the freshness of local datasets and conduct a theoretical analysis of the convergence bound in our AoI-aware FL system.Based on the convergence bound,we further formulate our problem as a restless multi-armed bandit(RMAB)problem.Next,we relax the RMAB problem and apply the Lagrangian Dual approach to decouple it into multiple subproblems.Finally,we propose a Whittle's Index Based Client Selection(WICS)algorithm to determine the set of selected clients.In addition,comprehensive simulations substantiate that the proposed algorithm can effectively reduce training loss and enhance the learning accuracy compared with some state-of-the-art methods.
federated learningAge of Informationrestless multi-armed banditWhittle's index
徐殷、肖明军、吴晨、吴杰、周津锐、孙贺
展开 >
School of Computer Science and Technology,University of Science and Technology of China,Hefei 230026,China
Suzhou Institute for Advanced Research,University of Science and Technology of China,Suzhou 215123,China
School of Data Science,University of Science and Technology of China,Hefei 230026,China
Department of Computer and Information Sciences,Temple University,Philadelphia,PA 19122,U.S.A.
展开 >
National Natural Science Foundation of ChinaNatural Science Foundation of Jiangsu Province of ChinaTeaching Research Project of the Education Department of Anhui Province of China
万方数据
维普
