Journal article 26 views 10 downloads
APCSMA: Adaptive Personalized Client-Selection and Model-Aggregation Algorithm for Federated Learning in Edge Computing Scenarios
,
Guorui Ma,
Yang Liu ,
Shuhan Qi
Entropy, Volume: 26, Issue: 8, Start page: 712
Swansea University Author:
Yang Liu
-
PDF | Version of Record
© 2024 by the authors. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Download (20.75MB)
DOI (Published version): 10.3390/e26080712
Abstract
With the rapid advancement of the Internet and big data technologies, traditional centralized machine learning methods are challenged when dealing with large-scale datasets. Federated Learning (FL), as an emerging distributed machine learning paradigm, enables multiple clients to collaboratively tra...
| Published in: | Entropy |
|---|---|
| ISSN: | 1099-4300 |
| Published: |
MDPI AG
2024
|
| Online Access: |
Check full text
|
| URI: | https://cronfa.swan.ac.uk/Record/cronfa67498 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| Abstract: |
With the rapid advancement of the Internet and big data technologies, traditional centralized machine learning methods are challenged when dealing with large-scale datasets. Federated Learning (FL), as an emerging distributed machine learning paradigm, enables multiple clients to collaboratively train a global model while preserving privacy. Edge computing, also recognized as a critical technology for handling massive datasets, has garnered significant attention. However, the heterogeneity of clients in edge computing environments can severely impact the performance of the resultant models. This study introduces an Adaptive Personalized Client-Selection and Model-Aggregation Algorithm, APCSMA, aimed at optimizing FL performance in edge computing settings. The algorithm evaluates clients’ contributions by calculating the real-time performance of local models and the cosine similarity between local and global models, and it designs a ContriFunc function to quantify each client’s contribution. The server then selects clients and assigns weights during model aggregation based on these contributions. Moreover, the algorithm accommodates personalized needs in local model updates, rather than simply overwriting with the global model. Extensive experiments were conducted on the FashionMNIST and Cifar-10 datasets, simulating three data distributions with parameters dir = 0.1, 0.3, and 0.5. The accuracy improvements achieved were 3.9%, 1.9%, and 1.1% for the FashionMNIST dataset, and 31.9%, 8.4%, and 5.4% for the Cifar-10 dataset, respectively. |
|---|---|
| Keywords: |
edge computing; federated learning; client selection; model aggregation |
| College: |
Faculty of Science and Engineering |
| Funders: |
This research was funded by Guangdong Provincial Key Laboratory of Novel Security Intelligence Technologies (No. 2022B1212010005); Shenzhen Science and Technology Major Special Project (No. KJZD20230923114608017); National Natural Science Foundation of China (No. 62372139, No. 62376073); Natural Science Foundation of Guang-dong (No. 2024A1515030024); Shenzhen Stable Supporting Program (General Project) (No. GXWD20231130110352002); and Shenzhen Foundational Research Funding Under Grant (No. 20220818102414030, No. JCYJ20200109113427092). |
| Issue: |
8 |
| Start Page: |
712 |