Title :
Fixed point optimization of deep convolutional neural networks for object recognition
Author :
Anwar, Sajid ; Kyuyeon Hwang ; Wonyong Sung
Author_Institution :
Dept. of Electr. & Comput. Eng., Seoul Nat. Univ., Seoul, South Korea
Abstract :
Deep convolutional neural networks have shown promising results in image and speech recognition applications. The learning capability of the network improves with increasing depth and size of each layer. However this capability comes at the cost of increased computational complexity. Thus reduction in hardware complexity and faster classification are highly desired. This work proposes an optimization method for fixed point deep convolutional neural networks. The parameters of a pre-trained high precision network are first directly quantized using L2 error minimization. We quantize each layer one by one, while other layers keep computation with high precision, to know the layer-wise sensitivity on word-length reduction. Then the network is retrained with quantized weights. Two examples on object recognition, MNIST and CIFAR-10, are presented. Our results indicate that quantization induces sparsity in the network which reduces the effective number of network parameters and improves generalization. This work reduces the required memory storage by a factor of 1/10 and achieves better classification results than the high precision networks.
Keywords :
neural nets; object recognition; optimisation; speech recognition; L2 error minimization; computational complexity; deep convolutional neural networks; faster classification; fixed point deep convolutional neural networks; fixed point optimization; hardware complexity; image recognition applications; network parameters; object recognition; optimization method; speech recognition applications; Biological neural networks; Convolution; Minimization; Quantization (signal); Sensitivity analysis; Training; convolutional neural network; quantization; sparsity; word length optimization;
Conference_Titel :
Acoustics, Speech and Signal Processing (ICASSP), 2015 IEEE International Conference on
Conference_Location :
South Brisbane, QLD
DOI :
10.1109/ICASSP.2015.7178146