Compressing Deep Model With Pruning and Tucker Decomposition for Smart Embedded Systems

被引:18
作者
Dai, Cheng [1 ]
Liu, Xingang [2 ]
Cheng, Hongqiang [2 ]
Yang, Laurence T. [3 ]
Deen, M. Jamal [4 ]
机构
[1] Sichuan Univ, Coll Comp Sci, Chengdu 610017, Peoples R China
[2] Univ Elect Sci & Technol China, Sch Informat & Commun Engn, Chengdu 611731, Peoples R China
[3] St Francis Xavier Univ, Dept Comp Sci, Antigonish, NS B2G 2W5, Canada
[4] McMaster Univ, Dept Elect Engn & Comp Sci, Hamilton, ON L8S 4K1, Canada
基金
中国国家自然科学基金;
关键词
Computational modeling; Deep learning; Bayes methods; Internet of Things; Data models; Streaming media; Edge computing; Deep model compression; parameter pruning; smart embedded systems; Tucker decomposition (TD); IOT;
D O I
10.1109/JIOT.2021.3116316
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Deep learning has been proved to be one of the most effective method in feature encoding for different intelligent applications such as video-based human action recognition. However, its nonconvex optimization mechanism leads large memory consumption, which hinders its deployment on the smart embedded systems with limited computational resources. To overcome this challenge, we propose a novel deep model compression technique for smart embedded systems, which realizes both the memory size reduction and inference complexity decrease within a small drop of accuracy. First, we propose an improved naive Bayes inference-based channel parameter pruning to obtain a sparse model with higher accuracy. Then, to improve the inference efficiency, the improved Tucker decomposition method is proposed, where an improved genetic algorithm is used to optimize the Tucker ranks. Finally, to elevate the effectiveness of our proposed method, extensive experiments are conducted. The experimental results show that our method can achieve the state-of-the-art performance compared with existing methods in terms of accuracy, parameter compression, and floating-point operations reduction.
引用
收藏
页码:14490 / 14500
页数:11
相关论文
共 56 条
[1]  
[Anonymous], 2018, Advances in neural information processing systems
[2]  
[Anonymous], 2015, INT C LEARNING REPRE
[3]   FlinkCL: An OpenCL-Based In-Memory Computing Architecture on Heterogeneous CPU-GPU Clusters for Big Data [J].
Chen, Cen ;
Li, Kenli ;
Ouyang, Aijia ;
Li, Keqin .
IEEE TRANSACTIONS ON COMPUTERS, 2018, 67 (12) :1765-1779
[4]   GFlink: An In-Memory Computing Architecture on Heterogeneous CPU-GPU Clusters for Big Data [J].
Chen, Cen ;
Li, Kenli ;
Ouyang, Aijia ;
Zeng, Zeng ;
Li, Keqin .
IEEE TRANSACTIONS ON PARALLEL AND DISTRIBUTED SYSTEMS, 2018, 29 (06) :1275-1288
[5]   GPU-Accelerated Parallel Hierarchical Extreme Learning Machine on Flink for Big Data [J].
Chen, Cen ;
Li, Kenli ;
Ouyang, Aijia ;
Tang, Zhuo ;
Li, Keqin .
IEEE TRANSACTIONS ON SYSTEMS MAN CYBERNETICS-SYSTEMS, 2017, 47 (10) :2740-2753
[6]   A Low-Latency Object Detection Algorithm for the Edge Devices of IoV Systems [J].
Dai, Cheng ;
Liu, Xingang ;
Chen, Weiting ;
Lai, Chin-Feng .
IEEE TRANSACTIONS ON VEHICULAR TECHNOLOGY, 2020, 69 (10) :11169-11178
[7]   Human action recognition using two-stream attention based LSTM networks [J].
Dai, Cheng ;
Liu, Xingang ;
Lai, Jinfeng .
APPLIED SOFT COMPUTING, 2020, 86
[8]   Human Behavior Deep Recognition Architecture for Smart City Applications in the 5G Environment [J].
Dai, Cheng ;
Liu, Xingang ;
Lai, Jinfeng ;
Li, Pan ;
Chao, Han-Chieh .
IEEE NETWORK, 2019, 33 (05) :206-211
[9]   CIRCNN: Accelerating and Compressing Deep Neural Networks Using Block-Circulant Weight Matrices [J].
Ding, Caiwen ;
Liao, Siyu ;
Wang, Yanzhi ;
Li, Zhe ;
Liu, Ning ;
Zhuo, Youwei ;
Wang, Chao ;
Qian, Xuehai ;
Bai, Yu ;
Yuan, Geng ;
Ma, Xiaolong ;
Zhang, Yipeng ;
Tang, Jian ;
Qiu, Qinru ;
Lin, Xue ;
Yuan, Bo .
50TH ANNUAL IEEE/ACM INTERNATIONAL SYMPOSIUM ON MICROARCHITECTURE (MICRO), 2017, :395-408
[10]  
Ding X., 2019, INT C MACHINE LEARNI, P1607