arrow
返回

A high-performance dataflow-centric optimization framework for deep learning inference on the edge

delete2024-07-01
delete1
PRE
AI
R
Runhua Zhang
H
Hongxu Jiang *
J
Jinkun Geng
F
Fangzheng Tian
Y
Yuhang Ma
H
Haojie Wang
DOI:10.1016/j.sysarc.2024.103180delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Edge computing has been emerging as a popular scenario for model inference. However, the inference performance on edge devices (e.g., Multi-Core DSP, FGPA, etc.) suffers from inefficiency due to the lack of highly optimized inference frameworks. Previous model inference frameworks are mainly developed in an operator-centric way, which provides insufficient acceleration to edge-based inference. Besides, the operator-centric framework incurs significant costs for continuous development and maintenance. Targeting the existing drawbacks of operator-centric frameworks, we design Xenos, which can automatically conduct dataflow-centric optimization of the computation graph and accelerate inference in two dimensions. Vertically, Xenos develops operator linking technique to improve data locality by restructuring the inter-operator dataflow. Horizontally, Xenos develops DSP-aware operator split technique to enable higher parallelism across multiple DSP units. Our evaluation demonstrates the effectiveness of vertical and horizontal dataflow optimization, which reduce the inference time by 15.0%-84.9% and 17.9%-89.9% , respectively. Besides, Xenos also outperforms the widely-used TVM by 1.1x-1.9x. Moreover, we extend Xenos to a distributed solution, which we call d-Xenos. d-Xenos employs multiple edge devices to jointly conduct the inference task and achieves a speedup of 3.68x-3.78x compared with the single device.
Keyword:
Edge computing
Model inference
Dataflow-centric
Computation graph
Data locality

期刊

Journal of Systems Architecture 封面图
Journal of Systems Architecture
IF:
4.1
论文数:
3.0K
被引数:
4.2K

机构

T
tsinghua university
学者数:
11.9W
论文数: 10.0W
被引数: 137
B
Beihang University
学者数:
5.2W
论文数: 4.1W
被引数: 37
S
Stanford University
学者数:
9.6W
论文数: 8.2W
被引数: 17.0W
学者 查看更多机构
引用论文

引用论文

Edge-Cloud Polarization and Collaboration: A Comprehensive Survey for AI边缘云极化与协作: 人工智能的综合调查
err2022-01-01
err41
errOAAI
errYao, Jiangchao; Zhang, Shengyu; Yao, Yang; Wang, Feng; Ma, Jianxin; Zhang, Jianwei; Chu, Yunfei; Ji, Luo; Jia, Kunyang; Shen, Tao; Wu, Anpeng; Zhang, Fengda; Tan, Ziqi; Kuang, Kun; Wu, Chao; Wu, Fei; Zhou, Jingren; Yang, Hongxia
err分享
err收藏
err分享
err收藏
err分享
err收藏
Latent mean differences in executive function in at-risk preterm children: The delay-deficit dilemma.
err2014-01-01
err0
PREAI
errIda Sue Baron; Brandi A. Weiss; Fern R. Litman; Margot D. Ahronovich; Robin Baker
err分享
err收藏
学者 查看更多内容