Adaptive Weight Compression for Memory-Efficient Neural Networks
Jong Hwan Koa, Duckhwan Kimb, Taesik Nac, Jaeha Kungd and Saibal Mukhopadhyaye
School of ECE, Georgia Institute of Technology, Atlanta, Georgia, USA.
ajonghwan.ko@gatech.edu
bkimduckhwan@gatech.edu
ctaesik.na@gatech.edu
djhkung@gatech.edu
esaibal@ece.gatech.edu
ABSTRACT
Neural networks generally require significant memory capacity/bandwidth to store/access a large number of synaptic weights. This paper presents an application of JPEG image encoding to compress the weights by exploiting the spatial locality and smoothness of the weight matrix. To minimize the loss of accuracy due to JPEG encoding, we propose to adaptively control the quantization factor of the JPEG algorithm depending on the error-sensitivity (gradient) of each weight. With the adaptive compression technique, the weight blocks with higher sensitivity are compressed less for higher accuracy. The adaptive compression reduces memory requirement, which in turn results in higher performance and lower energy of neural network hardware. The simulation for inference hardware for multilayer perceptron with the MNIST dataset shows up to 42X compression with less than 1% loss of recognition accuracy, resulting in 3X higher effective memory bandwidth and ~ 19X lower system energy.
Keywords: Neural network, Weight, Compression, Memoryefficient, JPEG, MLP.
