Skip to main content
Article
Computation Offloading and Resource Allocation Based on DT-MEC-Assisted Federated Learning Framework
IEEE Transactions on Cognitive Communications and Networking
  • Yejun He, Shenzhen University
  • Mengna Yang, Shenzhen University
  • Zhou He, A. James Clark School of Engineering
  • Mohsen Guizani, Mohamed Bin Zayed University of Artificial Intelligence
Document Type
Article
Abstract

Traditional centralized machine learning uses a large amount of data for model training, which may face some privacy and security problems. On the other hand, federated learning (FL), which focuses on privacy protection, also faces challenges such as core network congestion and limited mobile device (MD) resources. The computation offloading technology of mobile edge computing (MEC) can effectively alleviate these challenges, but it ignores the effect of user mobility and the unpredictable MEC environment. In this paper, we first propose an architecture that combines digital twin (DT) and MEC technology with the FL framework, where the DT network can virtually imitate the statue of physical entities (PEs) and network topology to be used for real-time data analysis and network resource optimization. The computation offloading technology of MEC is used to alleviate resource constraints of MDs and the core network congestion. We further leverage the FL to construct DT models based on PEs’ running data. Then, we jointly optimize the problem of computation offloading and resource allocation to reduce the straggler effect in FL based on the framework. Since the solution of the objective function is a stochastic programming problem, we model a markov decision process (MDP), and use the deep deterministic policy gradient (DDPG) algorithm to solve this objective function. The simulation results prove the feasibility of the proposed scheme, and the scheme can significantly reduce the total cost by about 50% and improve the communication performance compared with baseline schemes.

DOI
10.1109/TCCN.2023.3298926
Publication Date
7-26-2023
Keywords
  • Computation offloading,
  • Computational modeling,
  • deep deterministic policy gradient (DDPG),
  • federated learning,
  • Internet of Things,
  • Optimization,
  • resource allocation,
  • Resource management,
  • Servers,
  • Task analysis,
  • Training
Comments

IR Deposit conditions:

OA version (pathway a) Accepted version

No embargo

When accepted for publication, set statement to accompany deposit (see policy)

Must link to publisher version with DOI

Publisher copyright and source must be acknowledged

Citation Information
Y. He, M. Yang, Z. He and M. Guizani, "Computation Offloading and Resource Allocation Based on DT-MEC-Assisted Federated Learning Framework," in IEEE Transactions on Cognitive Communications and Networking, vol. 9, no. 6, pp. 1707-1720, Dec. 2023, doi: 10.1109/TCCN.2023.3298926