Laser & Optoelectronics Progress, Volume. 57, Issue 4, 041011(2020)
Method of Convolutional Neural Network Model Pruning Based on Gray Correlation Analysis
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.
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
Category: Image Processing
Received: Jun. 17, 2019
Accepted: Aug. 12, 2019
Published Online: Feb. 20, 2020
The Author Email: Bai Ruilin (hsqyc@163.com)