Abstract:Convolutional neural networks (CNNs) have achieved remarkable success in various computer vision tasks, which are extremely powerful to deal with massive training data by using tens of millions of parameters. However, CNNs often cost significant memory and computation consumption, which prohibits their usage in resource-limited environments such as mobile or embedded devices. To address the above issues, the existing approaches typically focus on either accelerating the convolutional layers or compressing the fully-connected layers separatedly, without pursuing a joint optimum. In this paper, we overcome such a limitation by introducing a holistic CNN compression framework, termed LRDKT, which works throughout both convolutional and fully-connected layers. First, a low-rank decomposition (LRD) scheme is proposed to remove redundancies across both convolutional kernels and fully-connected matrices, which has a novel closed-form solver to significantly improve the efficiency of the existing iterative optimization solvers. Second, a novel knowledge transfer (KT) based training scheme is introduced. To recover the accumulated accuracy loss and overcome the vanishing gradient, KT explicitly aligns outputs and intermediate responses from a teacher (original) network to its student (compressed) network. We have comprehensively analyzed and evaluated the compression and speedup ratios of the proposed model on MNIST and ILSVRC 2012 benchmarks. In both benchmarks, the proposed scheme has demonstrated superior performance gains over the state-of-the-art methods. We also demonstrate the proposed compression scheme for the task of transfer learning, including domain adaptation and object detection, which show exciting performance gains over the state-of-the-arts. Our source code and compressed models are available at https://github.com/ShaohuiLin/LRDKT.

Speeding-up and compression convolutional neural networks by low-rank decomposition without fine-tuning

Sensitivity-Oriented Layer-Wise Acceleration and Compression for Convolutional Neural Network.

Convolutional neural networks with low-rank regularization

TEC-CNN: Towards Efficient Compressing Convolutional Neural Nets with Low-rank Tensor Decomposition

CNN Acceleration by Low-rank Approximation with Quantized Factors

Holistic CNN Compression Via Low-Rank Decomposition with Knowledge Transfer.

Sparse Kronecker Canonical Polyadic Decomposition for Convolutional Neural Networks Compression

Convolutional Neural Network Compression Based on Low-Rank Decomposition

Compressing by Learning in a Low-Rank and Sparse Decomposition Form.

Speeding up Resnet Architecture with Layers Targeted Low Rank Decomposition

Stable Low-rank Tensor Decomposition for Compression of Convolutional Neural Network

Sensitivity-based Acceleration and Compression Algorithm for Convolution Neural Network.

Reduced storage direct tensor ring decomposition for convolutional neural networks compression

Convolutional neural networks compression with low rank and sparse tensor decompositions

Automatic Rank Selection for High-Speed Convolutional Neural Network

CNN Compression-Recovery Framework Via Rank Allocation Decomposition with Knowledge Transfer

An Accuracy-Preserving Neural Network Compression Via Tucker Decomposition

Speeding Up Deep Convolutional Neural Networks Based on Tucker-CP Decomposition

Compression of Deep Convolutional Neural Networks for Fast and Low Power Mobile Applications

CMD: Controllable Matrix Decomposition with Global Optimization for Deep Neural Network Compression

Low-rank Tensor Decomposition for Compression of Convolutional Neural Networks Using Funnel Regularization