Compressing Deep Neural Networks With Sparse Matrix Factorization

Kailun Wu,Yiwen Guo,Changshui Zhang
DOI: https://doi.org/10.1109/TNNLS.2019.2946636
IF: 14.255
2020-01-01
IEEE Transactions on Neural Networks and Learning Systems
Abstract:Modern deep neural networks (DNNs) are usually overparameterized and composed of a large number of learnable parameters. One of a few effective solutions attempts to compress DNN models via learning sparse weights and connections. In this article, we follow this line of research and present an alternative framework of learning sparse DNNs, with the assistance of matrix factorization. We provide an underlying principle for substituting the original parameter matrices with the multiplications of highly sparse ones, which constitutes the theoretical basis of our method. Experimental results demonstrate that our method substantially outperforms previous states of the arts for compressing various DNNs, giving rich empirical evidence in support of its effectiveness. It is also worth mentioning that, unlike many other works that focus on feedforward networks like multi-layer perceptrons and convolutional neural networks only, we also evaluate our method on a series of recurrent networks in practice.
What problem does this paper attempt to address?