D-Pruner: Filter-Based Pruning Method for Deep Convolutional Neural Network

被引:4
|
作者
Huynh, Loc N. [1 ]
Lee, Youngki [1 ]
Balan, Rajesh Krishna [1 ]
机构
[1] Singapore Management Univ, Singapore, Singapore
关键词
Continuous Vision; Deep Learning; Compression;
D O I
10.1145/3212725.3212730
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The emergence of augmented reality devices such as Google Glass and Microsoft Hololens has opened up a new class of vision sensing applications. Those applications often require the ability to continuously capture and analyze contextual information from video streams. They often adopt various deep learning algorithms such as convolutional neural networks (CNN) to achieve high recognition accuracy while facing severe challenges to run computationally intensive deep learning algorithms on resource-constrained mobile devices. In this paper, we propose and explore a new class of compression technique called D-Pruner to efficiently prune redundant parameters within a CNN model to run the model efficiently on mobile devices. D-Pruner removes redundancy by embedding a small additional network. This network evaluates the importance of filters and removes them during the fine-tuning phase to efficiently reduce the size of the model while maintaining the accuracy of the original model. We evaluated D-Pruner on various datasets such as CIFAR-10 and CIFAR-100 and showed that D-Pruner could reduce a significant amount of parameters up to 4.4 times on many existing models while maintaining accuracy drop less than 1%.
引用
收藏
页码:7 / 12
页数:6
相关论文
共 50 条
  • [41] Intrusion detection method based on a deep convolutional neural network
    Zhang S.
    Xie X.
    Xu Y.
    Qinghua Daxue Xuebao/Journal of Tsinghua University, 2019, 59 (01): : 44 - 52
  • [42] Improved method of deblocking filter based on convolutional neural network in VVC
    Yang, Jing
    Du, Biao
    Tang, Tong
    2020 IEEE/CIC INTERNATIONAL CONFERENCE ON COMMUNICATIONS IN CHINA (ICCC), 2020, : 764 - 769
  • [43] Filter Pruning via Probabilistic Model-based Optimization for Accelerating Deep Convolutional Neural Networks
    Li, Qinghua
    Li, Cuiping
    Chen, Hong
    WSDM '21: PROCEEDINGS OF THE 14TH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING, 2021, : 653 - 661
  • [44] Variational Convolutional Neural Network Pruning
    Zhao, Chenglong
    Ni, Bingbing
    Zhang, Jian
    Zhao, Qiwei
    Zhang, Wenjun
    Tian, Qi
    2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 2775 - 2784
  • [45] Convolutional Neural Network Pruning: A Survey
    Xu, Sheng
    Huang, Anran
    Chen, Lei
    Zhang, Baochang
    PROCEEDINGS OF THE 39TH CHINESE CONTROL CONFERENCE, 2020, : 7458 - 7463
  • [46] THREE-DIMENSIONAL CONVOLUTIONAL NEURAL NETWORK PRUNING WITH REGULARIZATION-BASED METHOD
    Zhang, Yuxin
    Wang, Huan
    Luo, Yang
    Yu, Lu
    Hu, Haoji
    Shan, Hangguan
    Quek, Tony Q. S.
    2019 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2019, : 4270 - 4274
  • [47] A Novel Clustering-Based Filter Pruning Method for Efficient Deep Neural Networks
    Wei, Xiaohui
    Shen, Xiaoxian
    Zhou, Changbao
    Yue, Hengshan
    ALGORITHMS AND ARCHITECTURES FOR PARALLEL PROCESSING, ICA3PP 2020, PT II, 2020, 12453 : 245 - 258
  • [48] Deep neural network pruning method based on sensitive layers and reinforcement learning
    Yang, Wenchuan
    Yu, Haoran
    Cui, Baojiang
    Sui, Runqi
    Gu, Tianyu
    ARTIFICIAL INTELLIGENCE REVIEW, 2023, 56 (SUPPL 2) : 1897 - 1917
  • [49] Pruning algorithm of convolutional neural network based on optimal threshold
    Wang, Jianjun
    Liu, Leshan
    Pan, Ximeng
    2020 5TH INTERNATIONAL CONFERENCE ON MATHEMATICS AND ARTIFICIAL INTELLIGENCE (ICMAI 2020), 2020, : 50 - 54
  • [50] Deep neural network pruning method based on sensitive layers and reinforcement learning
    Wenchuan Yang
    Haoran Yu
    Baojiang Cui
    Runqi Sui
    Tianyu Gu
    Artificial Intelligence Review, 2023, 56 : 1897 - 1917