Energy-Efficient and Reliable Federated Learning in Heterogeneous Mobile-Edge Computing
Yan Xu, Liying Li, Peijin Cong, Junlong Zhou · 2022
As a decentralized machine learning paradigm, Federated Learning (FL) is an emerging technique to protect user data privacy in Mobile Edge Computing (MEC). FL adopts the idea of distributed privacy computing, enabling the terminal devices to train machine learning models required by servers locally and upload model parameters to servers for aggregation. However, in the process of coupling FL with MEC systems, device heterogeneity, malicious use of poisoned data by users and limited system resources will seriously affect FL training accuracy, system reliability and energy efficiency. In this paper, a highly reliable and energy-efficient FL scheme is designed to solve these issues. Specifically, we first propose a calculation method for estimating the reliability of heterogeneous devices. Then, we design a highly reliable device selection scheme based on devices’ reliability, computing power, and participation times. Finally, we utilize the slack time in FL training to dynamically adjust the voltage and frequency of devices for reducing the energy consumption of training. Experimental results show that our proposed scheme improves the accuracy by up to 36.45% on average and saves energy by up to 48.63% when compared to the baseline and benchmarking methods.