Enhanced Bayesian Compression Via Deep Reinforcement Learning

Xin Yuan,Liangliang Ren,Jiwen Lu,Jie Zhou
DOI: https://doi.org/10.1109/cvpr.2019.00711
2019-01-01
Abstract:In this paper, we propose an Enhanced Bayesian Compression method to flexibly compress the deep networks via reinforcement learning. Unlike the existing Bayesian compression method which cannot explicitly enforce quantization weights during training, our method learns flexible codebooks in each layer for an optimal network quantization. To dynamically adjust the state of codebooks, we employ an Actor-Critic network to collaborate with the original deep network. Different from most existing network quantization methods, our EBC does not require re-training procedures after the quantization. Experimental results show that our method obtains low-bit precision with acceptable accuracy drop on MNIST, CIFAR and ImageNet.
What problem does this paper attempt to address?