Abstract:Deep learning models have evolved into powerful tools that can be used for many artificial intelligence tasks. However, deploying deep neural networks into real-world applications is still challenging due to their high computational complexity and storage overhead. Fortunately, a densely connected neural network can be converted into a sparsely connected network with low resource demand by the neural network compression. Since deep neural networks are complicated, compression mechanism should find a tradeoff between compression ratio and model accuracy. In this article, by analyzing the statistics of channel connection, we propose an interactive neural network compression mechanism including out-in-channel pruning and neural network quantization. Many channel pruning works apply structured sparsity regularization on each layer separately. We consider correlations between successive layers to retain predictive power of the compact network. A global greedy pruning algorithm is designed to remove redundant out-in-channels in an iterative way. Moreover, in order to solve the shortcomings of the one-shot quantization, we propose the incremental quantization algorithm in the dimension of the output channel, which can smooth network fluctuations and recover accuracy better during retraining. Our mechanism is comprehensively evaluated with various Convolutional Neural Networks (CNN) architectures on popular datasets. Notably, on ImageNet-1K, the out-in-channel pruning reduce 54.0 percent FLOPS on AlexNet and 50.0 percent FLOPs on ResNet-50 with only 0.15 and 0.37 percent top-1 accuracy drop respectively. On classification and style transfer tasks, the superiority of incremental quantization increases with the decrease of the number of quantization bits.

Quantisation and Pruning for Neural Network Compression and Regularisation

Single-shot Pruning and Quantization for Hardware-Friendly Neural Network Acceleration

Structured Deep Neural Network Pruning by Varying Regularization Parameters.

Pruning and quantization for deep neural network acceleration: A survey

Class-Aware Pruning for Efficient Neural Networks

Loss Constrains Added Squeeze and Excitation Blocks for Pruning Deep Neural Networks

Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Coding

CLIP-Q: Deep Network Compression Learning by In-parallel Pruning-Quantization

OPQ: Compressing Deep Neural Networks with One-shot Pruning-Quantization

Pruning at a Glance: Global Neural Pruning for Model Compression

Learning Low Resource Consumption CNN through Pruning and Quantization

Automatic Pruning for Quantized Neural Networks

On the compression of neural networks using ℓ0-norm regularization and weight pruning

On the Compression of Neural Networks Using $\ell_0$-Norm Regularization and Weight Pruning

Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruning

Differentiable Joint Pruning and Quantization for Hardware Efficiency

Towards Hardware-Specific Automatic Compression of Neural Networks

Pruning Ternary Quantization

Neural Network Compression using Binarization and Few Full-Precision Weights

End-to-End Neural Network Compression via $\frac{\ell_1}{\ell_2}$ Regularized Latency Surrogates