共 24 条
GA-DRL: Graph Neural Network-Augmented Deep Reinforcement Learning for DAG Task Scheduling Over Dynamic Vehicular Clouds
被引:4
|作者:
Liu, Zhang
[1
]
Huang, Lianfen
[1
]
Gao, Zhibin
[2
]
Luo, Manman
[3
]
Hosseinalipour, Seyyedali
[4
]
Dai, Huaiyu
[5
]
机构:
[1] Xiamen Univ, Dept Informat & Commun Engn, Xiamen 361005, Peoples R China
[2] Jimei Univ, Nav Inst, Xiamen 361021, Peoples R China
[3] Xiamen Univ, Dept Elect Engn, Xiamen 361005, Peoples R China
[4] Univ Buffalo SUNY, Dept Elect Engn, Buffalo, NY 14260 USA
[5] North Carolina State Univ, Dept Elect & Comp Engn, Raleigh, NC 26795 USA
来源:
IEEE TRANSACTIONS ON NETWORK AND SERVICE MANAGEMENT
|
2024年
/
21卷
/
04期
基金:
中国国家自然科学基金;
关键词:
Task analysis;
Dynamic scheduling;
Topology;
Vehicle dynamics;
Processor scheduling;
Heuristic algorithms;
Feature extraction;
Vehicular cloud;
DAG scheduling;
deep reinforcement learning;
graph neural network;
ALGORITHM;
D O I:
10.1109/TNSM.2024.3387707
中图分类号:
TP [自动化技术、计算机技术];
学科分类号:
0812 ;
摘要:
Vehicular Clouds (VCs) are modern platforms for processing of computation-intensive tasks over vehicles. Such tasks are often represented as Directed Acyclic Graphs (DAGs) consisting of interdependent vertices/subtasks and directed edges. However, efficient scheduling of DAG tasks over VCs presents significant challenges, mainly due to the dynamic service provisioning of vehicles within VCs and non-Euclidean representation of DAG tasks' topologies. In this paper, we propose a Graph neural network-Augmented Deep Reinforcement Learning scheme (GA-DRL) for the timely scheduling of DAG tasks over dynamic VCs. In doing so, we first model the VC-assisted DAG task scheduling as a Markov decision process. We then adopt a multi-head Graph ATtention network (GAT) to extract the features of DAG subtasks. Our developed GAT enables a two-way aggregation of the topological information in a DAG task by simultaneously considering predecessors and successors of each subtask. We further introduce non-uniform DAG neighborhood sampling through codifying the scheduling priority of different subtasks, which makes our developed GAT generalizable to completely unseen DAG task topologies. Finally, we augment GAT into a double deep Q-network learning module to conduct subtask-to-vehicle assignment according to the extracted features of subtasks, while considering the dynamics and heterogeneity of the vehicles in VCs. Through simulating various DAG tasks under real-world movement traces of vehicles, we demonstrate that GA-DRL outperforms existing benchmarks in terms of DAG task completion time.
引用
收藏
页码:4226 / 4242
页数:17
相关论文