No Cover Image

Journal article 63 views 15 downloads

APCSMA: Adaptive Personalized Client-Selection and Model-Aggregation Algorithm for Federated Learning in Edge Computing Scenarios

Xueting Ma Orcid Logo, Guorui Ma, Yang Liu Orcid Logo, Shuhan Qi Orcid Logo

Entropy, Volume: 26, Issue: 8, Start page: 712

Swansea University Author: Yang Liu Orcid Logo

  • 67498.VoR.pdf

    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)

Check full text

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...

Full description

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