Distributed Resource Scheduling for Large-Scale MEC Systems: A Multi-Agent Ensemble Deep Reinforcement Learning with Imitation Acceleration

Feibo Jiang, Li Dong*, Kezhi Wang*, Kun Yang, Cunhua Pan

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

52 Citations (Scopus)
60 Downloads (Pure)

Abstract

In large-scale mobile edge computing (MEC) systems, the task latency and energy consumption are important for massive resource-consuming and delay-sensitive Internet of things devices (IoTDs). Against this background, we propose a distributed intelligent resource scheduling (DIRS) framework to minimize the sum of task latency and energy consumption for all IoTDs, which can be formulated as a mixed integer nonlinear programming. The DIRS framework includes centralized training relying on the global information and distributed decision making by each agent deployed in each MEC server. Specifically, we first introduce a novel multi-agent ensemble-assisted distributed deep reinforcement learning (DRL) architecture, which can simplify the overall neural network structure of each agent by partitioning the state space and also improve the performance of a single agent by combining decisions of all the agents. Secondly, we apply action refinement to enhance the exploration ability of the proposed DIRS framework, where the near-optimal state-action pairs are obtained by a novel Levy flight search. Finally, an imitation acceleration scheme is presented to pre-train all the agents, which can significantly accelerate the learning process of the proposed framework through learning the professional experience from a small amount of demonstration data. The simulation results in three typical scenarios demonstrate that the proposed DIRS framework is efficient and outperforms the existing benchmark schemes.
Original languageEnglish
Pages (from-to)6597-6610
Number of pages13
JournalIEEE Internet of Things Journal
Volume9
Issue number9
Early online date20 Sept 2021
DOIs
Publication statusPublished - 1 May 2022

Keywords

  • Multi-agent reinforcement learning
  • Distributed deep reinforcement learning
  • Imitation learning
  • Resource scheduling
  • Levy flight

Fingerprint

Dive into the research topics of 'Distributed Resource Scheduling for Large-Scale MEC Systems: A Multi-Agent Ensemble Deep Reinforcement Learning with Imitation Acceleration'. Together they form a unique fingerprint.

Cite this