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.

Hard Parameter Sharing for Compressing Dense-Connection-based Image Restoration Network.

Re-training and parameter sharing with the Hash trick for compressing convolutional neural networks

Toward Compact Parameter Representations for Architecture-Agnostic Neural Network Compression

Build Receptive Pyramid for Efficient Color Image Compression Artifact Reduction

Learning Parameter Sharing with Tensor Decompositions and Sparsity

Dilated Residual Encode-Decode Networks for Image Denoising

Deep Parameterized Neural Networks for Hyperspectral Image Denoising

Frequency-Aware Re-Parameterization for Over-Fitting Based Image Compression

A Pruning Method Based on the Dissimilarity of Angle among Channels and Filters

EARN: Toward Efficient and Robust JPEG Compression Artifact Reduction

Two-stream deep sparse network for accurate and efficient image restoration

Efficient Neural Network Compression Inspired by Compressive Sensing.

SCWC: Structured channel weight sharing to compress convolutional neural networks

Lightweight Single Image Super-Resolution with Selective Channel Processing Network

LIR: A Lightweight Baseline for Image Restoration

Towards Compact CNNs via Collaborative Compression

SCGA‐Net: Skip Connections Global Attention Network for Image Restoration

Compressing complex convolutional neural network based on an improved deep compression algorithm

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

Expansion-Squeeze- Block: Linear Over-parameterization with Shortcut Connections to Train Compact Convolutional Networks

Compressing Deep Networks by Neuron Agglomerative Clustering