nn-Meter: towards accurate latency prediction of deep-learning model inference on diverse edge devices
nn-Meter: towards accurate latency prediction of deep-learning model inference on diverse edge devices
复制标题
nn-Meter:在各种边缘设备上实现深度学习模型推理的准确延迟预测
DOI:
10.1145/3458864.3467882
复制
发表时间:
2021
期刊:
影响因子:
--
通讯作者:
Yunxin Liu
中科院分区:
文献类型:
--
作者:
L. Zhang;S. Han;Jianyu Wei;Ningxin Zheng;Ting Cao;Yuqing Yang;Yunxin Liu
With the recent trend of on-device deep learning, inference latency has become a crucial metric in running Deep Neural Network (DNN) models on various mobile and edge devices. To this end, latency prediction of DNN model inference is highly desirable for many tasks where measuring the latency on real devices is infeasible or too costly, such as searching for efficient DNN models with latency constraints from a huge model-design space. Yet it is very challenging and existing approaches fail to achieve a high accuracy of prediction, due to the varying model-inference latency caused by the runtime optimizations on diverse edge devices. In this paper, we propose and develop nn-Meter, a novel and efficient system to accurately predict the inference latency of DNN models on diverse edge devices. The key idea of nn-Meter is dividing a whole model inference into kernels, i.e., the execution units on a device, and conducting kernel-level prediction. nn-Meter builds atop two key techniques: (i) kernel detection to automatically detect the execution unit of model inference via a set of well-designed test cases; and (ii) adaptive sampling to efficiently sample the most beneficial configurations from a large space to build accurate kernel-level latency predictors. Implemented on three popular platforms of edge hardware (mobile CPU, mobile GPU, and Intel VPU) and evaluated using a large dataset of 26,000 models, nn-Meter significantly outperforms the prior state-of-the-art.
影响因子:
6.2
作者:
Adams, Andrew;Ma, Karima;Ragan-Kelley, Jonathan
通讯作者:
Ragan-Kelley, Jonathan