Residential College | false |
Status | 已發表Published |
FedGCS: A Generative Framework for Efficient Client Selection in Federated Learning via Gradient-based Optimization | |
Ning, Zhiyuan1,2; Tian, Chunlin4; Xiao, Meng1,2; Fan, Wei5; Wang, Pengyang4; Li, Li4; Wang, Pengfei1,2; Zhou, Yuanchun1,2,3 | |
2024 | |
Conference Name | 33rd International Joint Conference on Artificial Intelligence, IJCAI 2024 |
Source Publication | Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence (IJCAI-24) |
Pages | 4760-4768 |
Conference Date | 3-9 August 2024 |
Conference Place | Jeju, South Korea |
Publisher | International Joint Conferences on Artificial Intelligence |
Abstract | Federated Learning faces significant challenges in statistical and system heterogeneity, along with high energy consumption, necessitating efficient client selection strategies. Traditional approaches, including heuristic and learning-based methods, fall short of addressing these complexities holistically. In response, we propose FedGCS, a novel generative client selection framework that innovatively recasts the client selection process as a generative task. Drawing inspiration from the methodologies used in large language models, FedGCS efficiently encodes abundant decision-making knowledge within a continuous representation space, enabling efficient gradient-based optimization to search for optimal client selection that will be finally output via generation. The framework comprises four steps: (1) automatic collection of diverse “selection-score” pair data using classical client selection methods; (2) training an encoder-evaluator-decoder framework on this data to construct a continuous representation space; (3) employing gradient-based optimization in this space for optimal client selection; (4) generating the final optimal client selection via using beam search for the well-trained decoder. FedGCS outperforms traditional methods by being more comprehensive, generalizable, and efficient, simultaneously optimizing for model performance, latency, and energy consumption. The effectiveness of FedGCS is proven through extensive experimental analyses. |
Keyword | Machine Learning Data Mining |
DOI | 10.24963/ijcai.2024/526 |
URL | View the original |
Language | 英語English |
Scopus ID | 2-s2.0-85204286352 |
Fulltext Access | |
Citation statistics | |
Document Type | Conference paper |
Collection | Faculty of Science and Technology THE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU) DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Li, Li; Wang, Pengfei |
Affiliation | 1.Computer Network Information Center, Chinese Academy of Sciences, China 2.University of Chinese Academy of Sciences, China 3.Hangzhou Institute for Advanced Study, UCAS, China 4.Department of Computer and Information Science, IOTSC, University of Macau, Macao 5.University of Oxford, United Kingdom |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Ning, Zhiyuan,Tian, Chunlin,Xiao, Meng,et al. FedGCS: A Generative Framework for Efficient Client Selection in Federated Learning via Gradient-based Optimization[C]:International Joint Conferences on Artificial Intelligence, 2024, 4760-4768. |
APA | Ning, Zhiyuan., Tian, Chunlin., Xiao, Meng., Fan, Wei., Wang, Pengyang., Li, Li., Wang, Pengfei., & Zhou, Yuanchun (2024). FedGCS: A Generative Framework for Efficient Client Selection in Federated Learning via Gradient-based Optimization. Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence (IJCAI-24), 4760-4768. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment