面向异构IoT设备协作的DNN推断加速研究

孙胜; 李叙晶; 刘敏; 杨博; 过晓冰

doi:10.7544/issn1000-1239.2020.20190863

面向异构IoT设备协作的DNN推断加速研究

孙胜^1,2,
李叙晶^1,2,
刘敏^1,2,
杨博^1,2,
过晓冰³

¹(中国科学院计算技术研究所北京 100190)
²(中国科学院大学北京 100049)
³(联想研究院北京 100085) (sunsheng@ict.ac.cn)

基金项目: 国家自然科学基金项目(61732017,61872028)

详细信息

中图分类号: TP391
计量
- 文章访问数: 1343
- HTML全文浏览量: 6
- PDF下载量: 591
出版历程
- 发布日期: 2020-03-31

DNN Inference Acceleration via Heterogeneous IoT Devices Collaboration

Sun Sheng^1,2,
Li Xujing^1,2,
Liu Min^1,2,
Yang Bo^1,2,
Guo Xiaobing³

¹(Institute of Computing Technology, Chinese Academy of Sciences, Beijing 100190)
²(University of Chinese Academy of Sciences, Beijing 100049)
³(Lenovo Research, Beijing 100085)

Funds: This work was supported by the National Natural Science Foundation of China (61732017, 61872028).

摘要

摘要: 深度神经网络(deep neural network, DNN)已经广泛应用于各种智能应用，如图像和视频识别.然而，由于DNN任务计算量大，资源受限的物联网(Internet of things, IoT)设备难以本地单独执行DNN推断任务.现有云协助方法容易受到通信延迟无法预测和远程服务器性能不稳定等因素的影响.一种非常有前景的方法是利用IoT设备协作实现分布式、可扩展DNN任务推断.然而，现有工作仅研究IoT设备同构情况下的静态拆分策略.因此，迫切需要研究如何在能力异构且资源受限的IoT设备间自适应地拆分DNN任务，协作执行任务推断.上述研究问题面临2个重要挑战:1)DNN任务多层推断延迟难以准确预测;2)难以在异构动态的多设备环境中实时智能调整协作推断策略.为此，首先提出细粒度可解释的多层延迟预测模型.进一步，利用进化增强学习(evolutionary reinforcement learning, ERL)自适应确定DNN推断任务的近似最优拆分策略.实验结果表明：该方法能够在异构动态环境中实现显著DNN推断加速.
- 深度神经网络推断加速 /
- 异构设备协作 /
- 进化增强学习 /
- 多层预测模型 /
- 拆分策略
Abstract: Deep neural networks (DNNs) have been intensively deployed in a variety of intelligent applications (e.g., image and video recognition). Nevertheless, due to DNNs’ heavy computation burden, resource-constrained IoT devices are unsuitable to locally execute DNN inference tasks. Existing cloud-assisted approaches are severely affected by unpredictable communication latency and unstable performance of remote servers. As a countermeasure, it is a promising paradigm to leverage collaborative IoT devices to achieve distributed and scalable DNN inference. However, existing works only consider homogeneous IoT devices with static partition. Thus, there is an urgent need for a novel framework to adaptively partition DNN tasks and orchestrate distributed inference among heterogeneous resource-constrained IoT devices. There are two main challenges in this framework. First, it is difficult to accurately profile the DNNs’ multi-layer inference latency. Second, it is difficult to learn the collaborative inference strategy adaptively and in real-time in the heterogeneous environments. To this end, we first propose an interpretable multi-layer prediction model to abstract complex layer parameters. Furthermore, we leverage the evolutionary reinforcement learning (ERL) to adaptively determine the near-optimal partitioning strategy for DNN inference tasks. Real-world experiments based on Raspberry Pi are implemented, showing that our proposed method can significantly accelerate the inference speed in dynamic and heterogeneous environments.
- DNN inference acceleration /
- heterogeneous device collaboration /
- evolutionary reinforce-ment learning /
- multi-layer prediction model /
- partitioning strategy

HTML全文

参考文献(0)

施引文献(14)

期刊类型引用(6)

1.	郭晓龙，牛晋宇，杜永萍. 基于树莓派的高效卷积优化方法. 计算机技术与发展. 2023(05): 96-104 . 百度学术
2.	辛明勇，祝健杨，徐长宝，姚浩，刘德宏. 基于循环神经网络的多核处理器层次化存储技术. 电子设计工程. 2023(22): 121-124+129 . 百度学术
3.	王利伟，玄志武，徐洪洲，刘学. Windows环境下遥测数据并行拼接处理方法研究. 电子设计工程. 2021(02): 10-15 . 百度学术
4.	孟慧玲，王耀彬，李凌，杨洋，王欣夷，刘志勤. TACLeBench中内核程序循环级推测并行性分析. 计算机应用. 2021(09): 2652-2657 . 百度学术
5.	于海心，王晶，李晓锋. 基于改进RMS算法的多核嵌入式系统总线周期调度表优化设计. 火炮发射与控制学报. 2021(03): 71-75 . 百度学术
6.	丁艳，张海文，孙永彦. 基于多网格技术的电网工程造价数据信息分析方法研究. 电子设计工程. 2021(19): 35-39 . 百度学术