Multicore Federated Learning for Mobile-Edge Computing Platforms
Yang Bai,Lixing Chen,Jianhua Li,Jun Wu,Pan Zhou,Zichuan Xu,Jie Xu
DOI: https://doi.org/10.1109/jiot.2022.3224239
IF: 10.6
2023-03-25
IEEE Internet of Things Journal
Abstract: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.
computer science, information systems,telecommunications,engineering, electrical & electronic