Personalized client-edge-cloud hierarchical federated learning in mobile edge computing

Abstract Mobile edge computing aims to deploy mobile applications at the edge of wireless networks. Federated learning in mobile edge computing is a forward-looking distributed framework for deploying deep learning algorithms in many application scenarios. One challenge of federated learning in mobi...

Full description

Saved in:
Bibliographic Details
Main Authors: Chunmei Ma, Xiangqian Li, Baogui Huang, Guangshun Li, Fengyin Li
Format: Article
Language:English
Published: SpringerOpen 2024-12-01
Series:Journal of Cloud Computing: Advances, Systems and Applications
Subjects:
Online Access:https://doi.org/10.1186/s13677-024-00721-w
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1841559092762509312
author Chunmei Ma
Xiangqian Li
Baogui Huang
Guangshun Li
Fengyin Li
author_facet Chunmei Ma
Xiangqian Li
Baogui Huang
Guangshun Li
Fengyin Li
author_sort Chunmei Ma
collection DOAJ
description Abstract Mobile edge computing aims to deploy mobile applications at the edge of wireless networks. Federated learning in mobile edge computing is a forward-looking distributed framework for deploying deep learning algorithms in many application scenarios. One challenge of federated learning in mobile edge computing is data heterogeneity since the unified model of federated learning performs poorly when client data are non-independent and identically distributed. Personalized federated learning can obtain amazing models in scenarios where client data are non-independent and identically distributed. This is because the personalized model captures the features of users’ local data more accurately than the unified model. The personalized federated learning problem under two-tier (server-client) federated learning structures has been widely studied and applied. However, a lot of research results exhibit three distinct limitations: 1) suboptimal communication efficiency, 2) slow model convergence, and 3) underutilization of the relationships within user data, resulting in lower accuracy of personalized models. In this paper, we present the first personalized federated learning algorithm based on the client-edge-cloud structure. The edge server is responsible for model personalization and employs a learnable mixing parameter to mix the local model and the global model. We also utilize two learnable normalization parameters trained by clients to improve the performance of personalized models. Furthermore, in order to facilitate the collaboration among edge servers, we propose a similarity aggregation method to assign aggregation weights based on the Tanimoto coefficients between models. The experimental results show that the proposed algorithm not only increases the convergence speed of personalized models but also improves their testing accuracy.
format Article
id doaj-art-bd5c75031a0448069b33e90b2432ae75
institution Kabale University
issn 2192-113X
language English
publishDate 2024-12-01
publisher SpringerOpen
record_format Article
series Journal of Cloud Computing: Advances, Systems and Applications
spelling doaj-art-bd5c75031a0448069b33e90b2432ae752025-01-05T12:46:16ZengSpringerOpenJournal of Cloud Computing: Advances, Systems and Applications2192-113X2024-12-0113111710.1186/s13677-024-00721-wPersonalized client-edge-cloud hierarchical federated learning in mobile edge computingChunmei Ma0Xiangqian Li1Baogui Huang2Guangshun Li3Fengyin Li4School of Computer Science, Qufu Normal UniversitySchool of Computer Science, Qufu Normal UniversitySchool of Computer Science, Qufu Normal UniversitySchool of Computer Science, Qufu Normal UniversitySchool of Computer Science, Qufu Normal UniversityAbstract Mobile edge computing aims to deploy mobile applications at the edge of wireless networks. Federated learning in mobile edge computing is a forward-looking distributed framework for deploying deep learning algorithms in many application scenarios. One challenge of federated learning in mobile edge computing is data heterogeneity since the unified model of federated learning performs poorly when client data are non-independent and identically distributed. Personalized federated learning can obtain amazing models in scenarios where client data are non-independent and identically distributed. This is because the personalized model captures the features of users’ local data more accurately than the unified model. The personalized federated learning problem under two-tier (server-client) federated learning structures has been widely studied and applied. However, a lot of research results exhibit three distinct limitations: 1) suboptimal communication efficiency, 2) slow model convergence, and 3) underutilization of the relationships within user data, resulting in lower accuracy of personalized models. In this paper, we present the first personalized federated learning algorithm based on the client-edge-cloud structure. The edge server is responsible for model personalization and employs a learnable mixing parameter to mix the local model and the global model. We also utilize two learnable normalization parameters trained by clients to improve the performance of personalized models. Furthermore, in order to facilitate the collaboration among edge servers, we propose a similarity aggregation method to assign aggregation weights based on the Tanimoto coefficients between models. The experimental results show that the proposed algorithm not only increases the convergence speed of personalized models but also improves their testing accuracy.https://doi.org/10.1186/s13677-024-00721-wMobile edge computingFederated learningClient-edge-cloudPersonalized modelNon-independent and identically distributed
spellingShingle Chunmei Ma
Xiangqian Li
Baogui Huang
Guangshun Li
Fengyin Li
Personalized client-edge-cloud hierarchical federated learning in mobile edge computing
Journal of Cloud Computing: Advances, Systems and Applications
Mobile edge computing
Federated learning
Client-edge-cloud
Personalized model
Non-independent and identically distributed
title Personalized client-edge-cloud hierarchical federated learning in mobile edge computing
title_full Personalized client-edge-cloud hierarchical federated learning in mobile edge computing
title_fullStr Personalized client-edge-cloud hierarchical federated learning in mobile edge computing
title_full_unstemmed Personalized client-edge-cloud hierarchical federated learning in mobile edge computing
title_short Personalized client-edge-cloud hierarchical federated learning in mobile edge computing
title_sort personalized client edge cloud hierarchical federated learning in mobile edge computing
topic Mobile edge computing
Federated learning
Client-edge-cloud
Personalized model
Non-independent and identically distributed
url https://doi.org/10.1186/s13677-024-00721-w
work_keys_str_mv AT chunmeima personalizedclientedgecloudhierarchicalfederatedlearninginmobileedgecomputing
AT xiangqianli personalizedclientedgecloudhierarchicalfederatedlearninginmobileedgecomputing
AT baoguihuang personalizedclientedgecloudhierarchicalfederatedlearninginmobileedgecomputing
AT guangshunli personalizedclientedgecloudhierarchicalfederatedlearninginmobileedgecomputing
AT fengyinli personalizedclientedgecloudhierarchicalfederatedlearninginmobileedgecomputing