Efficient Neural Network Compression via Transfer Learning for Industrial Optical Inspection

In this paper, we investigate learning the deep neural networks for automated optical 1 inspection in industrial manufacturing. Our preliminary result has shown the stun2 ning performance improvement by transfer learning from the completely dissimilar 3 source domain: ImageNet. Further study for demystifying this improvement shows 4 that the transfer learning produces a highly compressible network, which was not 5 the case for the network learned from scratch. The experimental result shows that 6 there is a negligible accuracy drop in the network learned by transfer learning until 7 it is compressed to 1/128 reduction of the number of convolution filters. This result 8 is contrary to the compression without transfer learning which loses more than 5% 9 accuracy at the same compression rate. 10

[1]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[2]  Dacheng Tao,et al.  On Compressing Deep Models by Low Rank and Sparse Decomposition , 2017, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[3]  Luc Van Gool,et al.  The Pascal Visual Object Classes (VOC) Challenge , 2010, International Journal of Computer Vision.

[4]  Michael S. Bernstein,et al.  ImageNet Large Scale Visual Recognition Challenge , 2014, International Journal of Computer Vision.

[5]  Geoffrey E. Hinton,et al.  Distilling the Knowledge in a Neural Network , 2015, ArXiv.

[6]  Rich Caruana,et al.  Do Deep Nets Really Need to be Deep? , 2013, NIPS.

[7]  Frank C. Park,et al.  Transfer learning for automated optical inspection , 2017, 2017 International Joint Conference on Neural Networks (IJCNN).

[8]  Song Han,et al.  Deep Compression: Compressing Deep Neural Network with Pruning, Trained Quantization and Huffman Coding , 2015, ICLR.