SubFlow: A Dynamic Induced-Subgraph Strategy Toward Real-Time DNN Inference and Training

被引:35
作者
Lee, Seulki [1 ]
Nirjon, Shahriar [1 ]
机构
[1] Univ N Carolina, Dept Comp Sci, Chapel Hill, NC 27599 USA
来源
2020 IEEE REAL-TIME AND EMBEDDED TECHNOLOGY AND APPLICATIONS SYMPOSIUM (RTAS 2020) | 2020年
关键词
NEURAL-NETWORKS; LEARNING CAPABILITY; BACKPROPAGATION;
D O I
10.1109/RTAS48715.2020.00-20
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
We introduce SubFlow-a dynamic adaptation and execution strategy for a deep neural network (DNN), which enables real-time DNN inference and training. The goal of SubFlow is to complete the execution of a DNN task within a timing constraint that may dynamically change while ensuring comparable performance to executing the full network by executing a subset of the DNN at run-time. To this end, we propose two online algorithms that enable SubFlow: 1) dynamic construction of a sub-network which constructs the best sub-network of the DNN in terms of size and configuration, and 2) time-bound execution which executes the sub-network within a given time budget either for inference or training. We implement and open-source SubFlow by extending TensorFlow with full compatibility by adding SubFlow operations for convolutional and fully-connected layers of a DNN. We evaluate SubFlow with three popular DNN models (LeNet-5, AlexNet, and KWS), which shows that it provides flexible run-time execution and increases the utility of a DNN under dynamic timing constraints, e.g., 1x-6.7x range of dynamic execution speed with average -3% of performance (inference accuracy) difference. We also implement an autonomous robot as an example system that uses SubFlow and demonstrate that its obstacle detection DNN is flexibly executed to meet a range of deadlines that varies depending on its running speed.
引用
收藏
页码:15 / 29
页数:15
相关论文
共 138 条
  • [1] Abadi M, 2016, PROCEEDINGS OF OSDI'16: 12TH USENIX SYMPOSIUM ON OPERATING SYSTEMS DESIGN AND IMPLEMENTATION, P265
  • [2] Al-Qizwini M, 2017, IEEE INT VEH SYM, P89, DOI 10.1109/IVS.2017.7995703
  • [3] [Anonymous], 2013, COMPUTER ORG DESIGN
  • [4] [Anonymous], 2016, P 14 ACM C EMB NETW
  • [5] [Anonymous], 2017, ARXIV170603912
  • [6] [Anonymous], 2014, DICT STAT 3E
  • [7] Arfken G., 1999, MATH METHODS PHYS
  • [8] Augonnet C., 2010, THESIS INRIA
  • [9] Bachman D., 2007, Advanced calculus demystified
  • [10] Badreddin E., 1992, IFAC Proceedings Volumes, V25, P325