Federated Learning for Online Resource Allocation in Mobile Edge Computing
A Deep Reinforcement Learning Approach
Jingjing Zheng (CISTER Research Centre)
Kai Li (CISTER Research Centre)
N. Mhaisen (TU Delft - Networked Systems)
Wei Ni (CSIRO: Commonwealth Scientific and Industrial Research Organisation)
Eduardo Tovar (CISTER Research Centre)
Mohsen Guizani (Mohamed Bin Zayed University of Artificial Intelligence)
More Info
expand_more
Other than for strictly personal use, it is not permitted to download, forward or distribute the text or part of it, without the consent of the author(s) and/or copyright holder(s), unless the work is under an open content license such as Creative Commons.
Abstract
Federated learning (FL) is increasingly considered to circumvent the disclosure of private data in mobile edge computing (MEC) systems. Training with large data can enhance FL learning accuracy, which is associated with non-negligible energy use. Scheduled edge devices with small data save energy but decrease FL learning accuracy due to a reduction in energy consumption. A trade-off between the energy consumption of edge devices and the learning accuracy of FL is formulated in this proposed work. The FL-enabled twin-delayed deep deterministic policy gradient (FL-TD3) framework is proposed as a solution to the formulated problem because its state and action spaces are large in a continuous domain. This framework provides the maximum accuracy ratio of FL divided by the device’s energy consumption. A comparison of the numerical results with the state-of-the-art demonstrates that the ratio has been improved significantly.