Dynamic Model Pruning with Feedback
Deep neural networks often have millions of parameters. This can hinder their deployment to low-end devices, not only due to high memory requirements but also because of increased latency at inference. We propose a novel model compression method that generates a sparse trained model without additional overhead: by allowing (i) dynamic allocation of the sparsity pattern and (ii) incorporating feedback signal to reactivate prematurely pruned weights we obtain a performant sparse model in one single training pass (retraining is not needed, but can further improve the performance). We evaluate the method on CIFAR-10 and ImageNet, and show that the obtained sparse models can reach the state-of-the-art performance of dense models and further that their performance surpasses all previously proposed pruning schemes (that come without feedback mechanisms).
Cite this item
Lin, T., Stich, S. U., Barba Flores, L. F., Dmitriev, D., & Jaggi, M. (2020). Dynamic Model Pruning with Feedback. ICLR - International Conference on Learning Representations. 8th International Conference on Learning Representations (ICLR), Virtual Conference, Formerly Addis Ababa, Ethiopia. https://infoscience.epfl.ch/handle/20.500.14299/168762
Click Copy to copy the citation, or select the text manually.
dynamic_model_pruning_with_feedback.pdf
Publisher's Version
Published version
openaccess
1.67 MB
Adobe PDF
1245400bc25ab6429e8f5295668907fe