Dependent Task Offloading in Edge Computing Using GNN and Deep Reinforcement Learning
Zequn Cao,Xiaoheng Deng,Sheng Yue,Ping Jiang,Ju Ren,Jinsong Gui
DOI: https://doi.org/10.1109/jiot.2024.3374969
IF: 10.6
2024-01-01
IEEE Internet of Things Journal
Abstract:Task offloading is a widely used technology in Edge Computing (EC), which declines the makespan of user task with the aid of resourceful edge servers. How to solve the competition for computation and communication resources among tasks is a fundamental issue in task offloading. Besides, real-life user tasks often comprise multiple interdependent subtasks. Dependencies among subtasks significantly raises the complexity of task offloading, and makes it difficult to propose generalized approaches for scenarios of different size. In this paper, we study the Dependent Task Offloading (DTO) problem within both single-user single-edge and multi-user multi-edge scenario. First, we use Directed Acyclic Graph (DAG) to model dependent task, where nodes and directed edges represent the subtasks and their interdependencies respectively. Then, we propose a task scheduling method based on Graph Attention Network (GAT) and Deep Reinforcement Learning (DRL) to minimize the makespan of user tasks. More specifically, our method introduces a multi-discrete action DRL scheduler that simultaneously determines which subtask to consider and whether it should be offloaded at each step, and employs GAT to encode the graph-based state representation. To stabilize and speed up DRL scheduler training, we pretrain GAT encoder with unsupervised learning. Extensive experiments demonstrate that our proposed approach can be applied to various environments and outperforms prior methods.
computer science, information systems,telecommunications,engineering, electrical & electronic