Abstract:Compressing Deep Neural Network (DNN) models to alleviate the storage and computation requirements is essential for practical applications, especially for resource limited devices. Although capable of reducing a reasonable amount of model parameters, previous unstructured or structured weight pruning methods can hardly truly accelerate inference, either due to the poor hardware compatibility of the unstructured sparsity or due to the low sparse rate of the structurally pruned network. Aiming at reducing both storage and computation, as well as preserving the original task performance, we propose a generalized weight unification framework at a hardware compatible micro-structured level to achieve high amount of compression and acceleration. Weight coefficients of a selected micro-structured block are unified to reduce the storage and computation of the block without changing the neuron connections, which turns to a micro-structured pruning special case when all unified coefficients are set to zero, where neuron connections (hence storage and computation) are completely removed. In addition, we developed an effective training framework based on the alternating direction method of multipliers (ADMM), which converts our complex constrained optimization into separately solvable subproblems. Through iteratively optimizing the subproblems, the desired micro-structure can be ensured with high compression ratio and low performance degradation. We extensively evaluated our method using a variety of benchmark models and datasets for different applications. Experimental results demonstrate state-of-the-art performance.

Structured Term Pruning for Computational Efficient Neural Networks Inference

Deep Neural Network Acceleration with Sparse Prediction Layers

Structured Probabilistic Pruning for Convolutional Neural Network Acceleration.

Structured Pruning for Efficient Convolutional Neural Networks Via Incremental Regularization

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

Class-Aware Pruning for Efficient Neural Networks

Pruning by Training: A Novel Deep Neural Network Compression Framework for Image Processing.

Efficient Structure Slimming for Spiking Neural Networks

Intermediate-grained kernel elements pruning with structured sparsity

Adaptive Activation-based Structured Pruning

Layer-adaptive Structured Pruning Guided by Latency

Iterative Activation-based Structured Pruning

Highly Efficient Sparse Neural Network Computing - Hardware and Software Solutions.

Connection Pruning for Deep Spiking Neural Networks with On-Chip Learning

Enhanced Sparsification via Stimulative Training

Towards Efficient Deep Spiking Neural Networks Construction with Spiking Activity based Pruning

Efficient Micro-Structured Weight Unification and Pruning for Neural Network Compression

Cloud–Edge Collaborative Inference with Network Pruning

StructADMM: A Systematic, High-Efficiency Framework of Structured Weight Pruning for DNNs

An efficient pruning and fine-tuning method for deep spiking neural network

A Dynamic Pruning Method on Multiple Sparse Structures in Deep Neural Networks