文章基本信息

标题：An efficient pruning scheme of deep neural networks for Internet of Things applications
本地全文：下载
作者：Chen Qi ; Shibo Shen ; Rongpeng Li 等
期刊名称：EURASIP Journal on Advances in Signal Processing
印刷版ISSN：1687-6172
电子版ISSN：1687-6180
出版年度：2021
卷号：2021
期号：1
页码：1
DOI：10.1186/s13634-021-00744-4
出版社：Hindawi Publishing Corporation
摘要：Nowadays, deep neural networks (DNNs) have been rapidly deployed to realize a number of functionalities like sensing, imaging, classification, recognition, etc. However, the computational-intensive requirement of DNNs makes it difficult to be applicable for resource-limited Internet of Things (IoT) devices. In this paper, we propose a novel pruning-based paradigm that aims to reduce the computational cost of DNNs, by uncovering a more compact structure and learning the effective weights therein, on the basis of not compromising the expressive capability of DNNs. In particular, our algorithm can achieve efficient end-to-end training that transfers a redundant neural network to a compact one with a specifically targeted compression rate directly. We comprehensively evaluate our approach on various representative benchmark datasets and compared with typical advanced convolutional neural network (CNN) architectures. The experimental results verify the superior performance and robust effectiveness of our scheme. For example, when pruning VGG on CIFAR-10, our proposed scheme is able to significantly reduce its FLOPs (floating-point operations) and number of parameters with a proportion of 76.2% and 94.1%, respectively, while still maintaining a satisfactory accuracy. To sum up, our scheme could facilitate the integration of DNNs into the common machine-learning-based IoT framework and establish distributed training of neural networks in both cloud and edge.
关键词：Deep neural networks ; Deep learning ; Internet of Things ; Resource-limited edge computing ; Pruning ; Efficiency