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
来源
PROCEEDINGS OF THE 2018 INTERNATIONAL WORKSHOP ON EMBEDDED AND MOBILE DEEP LEARNING (EMDL '18) | 2018年
关键词
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 条
  • [21] Thinning of convolutional neural network with mixed pruning
    Yang, Wenzhu
    Jin, Lilei
    Wang, Sile
    Cu, Zhenchao
    Chen, Xiangyang
    Chen, Liping
    IET IMAGE PROCESSING, 2019, 13 (05) : 779 - 784
  • [22] Method of plant leaf recognition based on improved deep convolutional neural network
    Zhu, Xiaolong
    Zhu, Meng
    Ren, Honge
    COGNITIVE SYSTEMS RESEARCH, 2018, 52 : 223 - 233
  • [23] Structured Pruning for Deep Convolutional Neural Networks: A Survey
    He, Yang
    Xiao, Lingao
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2024, 46 (05) : 2900 - 2919
  • [24] Pruning convolutional neural networks via filter similarity analysis
    Lili Geng
    Baoning Niu
    Machine Learning, 2022, 111 : 3161 - 3180
  • [25] A Sketch Recognition Method Based on Deep Convolutional-Recurrent Neural Network
    Zhao P.
    Liu Y.
    Liu H.
    Yao S.
    2018, Institute of Computing Technology (30): : 217 - 224
  • [26] Pruning convolutional neural networks via filter similarity analysis
    Geng, Lili
    Niu, Baoning
    MACHINE LEARNING, 2022, 111 (09) : 3161 - 3180
  • [27] Pruning by explaining: A novel criterion for deep neural network pruning
    Yeom, Seul-Ki
    Seegerer, Philipp
    Lapuschkin, Sebastian
    Binder, Alexander
    Wiedemann, Simon
    Mueller, Klaus-Robert
    Samek, Wojciech
    PATTERN RECOGNITION, 2021, 115
  • [28] CAPTCHA recognition based on deep convolutional neural network
    Wang, Jing
    Qin, Jiaohua
    Xiang, Xuyu
    Tan, Yun
    Pan, Nan
    MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2019, 16 (05) : 5851 - 5861
  • [29] Acupoint Detection Based on Deep Convolutional Neural Network
    Sun, Lingyao
    Sun, Shiying
    Fu, Yuanbo
    Zhao, Xiaoguang
    PROCEEDINGS OF THE 39TH CHINESE CONTROL CONFERENCE, 2020, : 7418 - 7422
  • [30] Link Prediction Based on Deep Convolutional Neural Network
    Wang, Wentao
    Wu, Lintao
    Huang, Ye
    Wang, Hao
    Zhu, Rongbo
    INFORMATION, 2019, 10 (05)