On Compressing Deep Models by Low Rank and Sparse Decomposition

Citations Over TimeTop 1% of 2017 papers

Abstract

Deep compression refers to removing the redundancy of parameters and feature maps for deep learning models. Low-rank approximation and pruning for sparse structures play a vital role in many compression works. However, weight filters tend to be both low-rank and sparse. Neglecting either part of these structure information in previous methods results in iteratively retraining, compromising accuracy, and low compression rates. Here we propose a unified framework integrating the low-rank and sparse decomposition of weight matrices with the feature map reconstructions. Our model includes methods like pruning connections as special cases, and is optimized by a fast SVD-free algorithm. It has been theoretically proven that, with a small sample, due to its generalizability, our model can well reconstruct the feature maps on both training and test data, which results in less compromising accuracy prior to the subsequent retraining. With such a warm start to retrain, the compression method always possesses several merits: (a) higher compression rates, (b) little loss of accuracy, and (c) fewer rounds to compress deep models. The experimental results on several popular models such as AlexNet, VGG-16, and GoogLeNet show that our model can significantly reduce the parameters for both convolutional and fully-connected layers. As a result, our model reduces the size of VGG-16 by 15×, better than other recent compression methods that use a single strategy.

Related Papers

• → Hyperspectral Images Classification With Gabor Filtering and Convolutional Neural Network(2017)264 cited
• → A deep feature-based real-time system for Alzheimer disease stage detection(2020)155 cited
• → Comparison of deep convolutional neural network models with OCT images for dental caries classification(2022)3 cited
• → Deep and Lightweight Neural Network for Histopathological Image Classification(2022)3 cited
• → Challenges of Deep Learning Methods for COVID-19 Detection Using Public Datasets(2020)10 cited