Multicore Federated Learning for Mobile-Edge Computing Platforms

Yang Bai

doi:10.1109/JIOT.2022.3224239

2023-04-01

PRE

DOI：10.1109/JIOT.2022.3224239

原文链接

原文求助

摘要

En 中文

With increasingly strict data privacy regulations, federated learning (FL) has become one of the most often heard machine learning techniques due to its privacy-preserving trait. To efficiently implement the FL intelligence, researchers recently resort to a newly emerged computing paradigm, mobile-edge computing (MEC), and bring about a burst of works. However, most existing works neglect practical issues in MEC systems, e.g., device heterogeneity, unstable channel conditions, and unknown user mobility. Any of them, if not handled properly, can cause fatal failures to FL. This article proposed a novel FL framework, called multicore FL (MC-FL), to help FL intelligence land successfully on realistic MEC systems. A distinct feature of MC-FL is maintaining and training multiple global models (GMs) that exhibit different tradeoffs between learning performances and computational complexity. While this modification seems simple, it can effectively handle the device heterogeneity and device status variations, and improve the compatibility and robustness of FL. Furthermore, MC-FL employs a partial client participation scheme that allows participating clients to vary across time. This enables MC-FL to function under uncertain mobile environments. We rigorously prove the convergence of the designed MC-FL framework. In particular, we propose an online client scheduling scheme for MC-FL to judiciously schedule clients for training multiple GMs in a manner that minimizes the completion time of MC-FL. We also provide a service provisioning scenario with MC-FL to show how service subscribers could benefit from multiple GMs and improve their Quality of Experience (QoE). We evaluate our method on real-world data sets, and the results show that MC-FL outperforms state-of-the-art benchmarks.

Keyword:

Training

Convergence

Performance evaluation

Federated learning

Servers

Standards

Processor scheduling

Client scheduling

federated learning (FL)

mobile-edge computing (MEC)