arrow
Return

Multi-Exit DNN Inference Acceleration Based on Multi-Dimensional Optimization for Edge Intelligence

delete2022-01-01
delete26
PRE
AI
东
东方 (Fang Dong)
H
Hui‐Tian Wang *
沈典 cover
沈典 (Dian Shen)
Z
Zhaowu Huang
Q
Qiang He
张
张竞慧 (Jinghui Zhang)
T
Tingting Zhang
DOI:10.1109/TMC.2022.3172402delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Edge intelligence, as a prospective paradigm for accelerating DNN inference, is mostly implemented by model partitioning which inevitably incurs the large transmission overhead of DNN's intermediate data. A popular solution introduces multi-exit DNNs to reduce latency by enabling early exits. However, existing work ignores the correlation between exit settings and synergistic inference, causing incoordination of device-to-edge. To address this issue, this paper first investigates the bottlenecks of executing multi-exit DNNs in edge computing and builds a novel model for inference acceleration with exit selection, model partition, and resource allocation. To tackle the intractable coupling subproblems, we propose a Multi-exit DNN inference Acceleration framework based on Multi-dimensional Optimization (MAMO). In MAMO, the exit selection subproblem is first extracted from the original problem. Then, bidirectional dynamic programming is employed to determine the optimal exit setting for an arbitrary multi-exit DNN. Finally, based on the optimal exit setting, a DRL-based policy is developed to learn joint decisions of model partition and resource allocation. We deploy MAMO on a real-world testbed and evaluate its performance in various scenarios. Extensive experiments show that it can adapt to heterogeneous tasks and dynamic networks, and accelerate DNN inference by up to 13:7x compared with the state-of-the-art.
Keywords:
Edge intelligence
exit selection
inference acceleration
model partition
multi-exit DNN
resource allocation

Journal

IEEE Transactions on Mobile Computing cover
IEEE Transactions on Mobile Computing
IF:
9.2
Papers:
5.9K
Citations:
1.8W

Organization

S
Swinburne University of Technology
Scholars:
9.3K
Papers: 1.2W
Citations: 2.0W
S
southeast university - china
Scholars:
5.3W
Papers: 4.9W
Citations: 57
C
China Mobile
Scholars:
939
Papers: 701
Citations: 2
researcher View more organizations
Cited Papers

Cited Papers

Performance Study of Screen-Printed Textile Antennas after Repeated Washing
err2014-06-17
err0
errOAAI
errI. Kazani; F. Declercq; M. L. Scarpello; C. Hertleer; H. Rogier; D. Vande Ginste; G. De Mey; G. Guxho; L. Van Langenhove
errShare
errSave
Edge Intelligence: Paving the Last Mile of Artificial Intelligence With Edge Computing
err2019-08-01
err1.2K
errOAAI
errZhou, Zhi; Chen, Xu; Li, En; Zeng, Liekang; Luo, Ke; Zhang, Junshan
errShare
errSave
Federated Machine Learning: Concept and Applications
err2019-01-28
err5.5K
PREAI
errYang, Qiang; Liu, Yang; Chen, Tianjian; Tong, Yongxin
errShare
errSave
In-Edge AI: Intelligentizing Mobile Edge Computing, Caching and Communication by Federated Learning
err2019-09-01
err690
errOAAI
errWang, Xiaofei; Han, Yiwen; Wang, Chenyang; Zhao, Qiyang; Chen, Xu; Chen, Min
errShare
errSave
What Comes After Telepresence? Embodiment, Social Presence and Transporting One’s Functional and Social Self
err2022-10-09
err0
errOAAI
errJan B.F. Van Erp; Camille Sallaberry; Christiaan Brekelmans; Douwe Dresscher; Frank Ter Haar; Gwenn Englebienne; Jeanine Van Bruggen; Joachim De Greeff; Leonor Fermoselle Silva Pereira; Alexander Toet; Nirul Hoeba; Robin Lieftink; Sara Falcone; Tycho Brug
errShare
errSave
researcher View more