Laser & Optoelectronics Progress, Volume. 57, Issue 4, 041011(2020)

Method of Convolutional Neural Network Model Pruning Based on Gray Correlation Analysis

Shiqing Huang1, Ruilin Bai1、*, and Gaoe Qin2
Author Affiliations
  • 1Key Laboratory of Advanced Process Control for Light Industry (Ministry of Education), Jiangnan University, Wuxi, Jiangsu 214122, China
  • 2Xinje Electronic Co., Ltd, Wuxi, Jiangsu 214122, China
  • show less

    A model pruning method based on gray correlation analysis is proposed to solve the problem that the convolutional neural network cannot be deployed on embedded devices due to the huge computation and memory space. For the weight model file after data training, the importance of each convolution kernel is quantized by using the pruning method based on gray correlation analysis. In each pruning, the convolution kernel with the minimum quantization result is deleted from the model so as to reduce the computation and accelerate the inferential speed. Iteration training is used to compensate for the performance loss of the new model. The experimental results show that compared with APoZ method and L1 method, the accuracy of the proposed method increases by 5.3% and 10.4% at the same inferential speed, the acceleration effect of VGG-16 model is 2.7 times that of the original model, and the memory space is reduced to 1/13.5.

    Tools

    Get Citation

    Copy Citation Text

    Shiqing Huang, Ruilin Bai, Gaoe Qin. Method of Convolutional Neural Network Model Pruning Based on Gray Correlation Analysis[J]. Laser & Optoelectronics Progress, 2020, 57(4): 041011

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Jun. 17, 2019

    Accepted: Aug. 12, 2019

    Published Online: Feb. 20, 2020

    The Author Email: Bai Ruilin (hsqyc@163.com)

    DOI:10.3788/LOP57.041011

    Topics