arrow
返回

Multiendpoint DAG-Driven Joint Partitioning-Offloading and Scheduling Optimization for DNN Inference

delete2025-10-01
delete1
PRE
AI
X
Xiukun Yan
X
Xuexue Zhang
K
Kai Zeng
柏粉花 封面图
柏粉花 (Fenhua Bai)
沈韬 封面图
沈韬 (Tao Shen) *
B
Bin Cao
DOI:10.1109/JIOT.2025.3591531delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Model partitioning techniques, which decompose and collaboratively execute subtasks of deep neural networks (DNNs), have emerged as a critical strategy for enhancing distributed inference efficiency. However, in mobile edge computing (MEC), dynamic load fluctuations at edge nodes and the complexity of cross-node task dependencies make the delay minimization problem extremely challenging. Existing studies predominantly adopt a decoupled optimization framework that separately addresses partitioning-offloading and pipeline scheduling, neglecting their inherent cyclic state-dependent coupling. This oversight leads to suboptimal solutions, such as pipeline stagnation caused by mismatched computation and communication timestamps. To address these challenges, we propose a multiendpoint directed acyclic graph (DAG)-driven cooperative optimization approach, enabling partitioning-offloading and pipeline scheduling in MEC. Specifically, the approach involves two core steps: 1) Dynamic prescheduling: We propose an improved DNN scheduling algorithm for constrained subtasks, which simulates node-level queuing delays and pipeline stalls under real-world constraints, translating runtime states into latency objectives. 2) Partitioning and offloading solution retrieval: Based on latency objectives, we introduce a novel multiendpoint DAG structure and design a multinode collaborative optimization retrieval algorithm, enabling adaptive partitioning-offloading remapping of subtasks. Experiments demonstrate the superiority of the proposed method over other advanced methods, reducing the time overhead by an average of 24% and 75% in two different scenarios, respectively. The resource code can be found at: https://github.com/aiheiheiheii/Partition_Scheduling.git.
Keyword:
Internet of Things
Computational modeling
Servers
Pipelines
Delays
Dynamic scheduling
Collaboration
Artificial neural networks
Optimization
Resource management
Constrained scheduling
distributed computing
deep neural network (DNN) inference
model partitioning
multiendpoint directed acyclic graph (DAG)

期刊

IEEE Internet of Things Journal 封面图
IEEE Internet of Things Journal
IF:
8.9
论文数:
1.4W
被引数:
7.8W

机构

B
beijing university of posts & telecommunications
学者数:
1.4W
论文数: 1.2W
被引数: 9
引用论文

引用论文

A Robust Privacy-Preserving Data Aggregation Scheme for Edge-Supported IIoT
err2024-03-01
err8
PREAI
errShang, Shuai; Li, Xiong; Gu, Ke; Li, Lei; Zhang, Xiaosong; Pandi, Vijayakumar
err分享
err收藏
An efficient DNN splitting scheme for edge-AI enabled smart manufacturing
err2023-08-01
err10
PREAI
errGauttam, Himanshu; Pattanaik, K. K.; Bhadauria, Saumya; Nain, Garima; Prakash, Putta Bhanu
err分享
err收藏
Low-Cost Federated Broad Learning for Privacy-Preserved Knowledge Sharing in the RIS-Aided Internet of Vehicles
err2024-02-01
err4
errOAAI
errYuan, Xiaoming; Chen, Jiahui; Zhang, Ning; Ye, Qiang (John); Li, Changle; Zhu, Chunsheng; Shen, Xuemin Sherman
err分享
err收藏
err分享
err收藏
err分享
err收藏
学者 查看更多内容