Network Pruning

203 papers with code • 5 benchmarks • 5 datasets

Network Pruning is a popular approach to reduce a heavy network to obtain a light-weight form by removing redundancy in the heavy network. In this approach, a complex over-parameterized network is first trained, then pruned based on come criterions, and finally fine-tuned to achieve comparable performance with reduced parameters.

Source: Ensemble Knowledge Distillation for Learning Improved and Efficient Networks


Use these libraries to find Network Pruning models and implementations

Most implemented papers

SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size

DeepScale/SqueezeNet 24 Feb 2016

(2) Smaller DNNs require less bandwidth to export a new model from the cloud to an autonomous car.

The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks

google-research/lottery-ticket-hypothesis ICLR 2019

Based on these results, we articulate the "lottery ticket hypothesis:" dense, randomly-initialized, feed-forward networks contain subnetworks ("winning tickets") that - when trained in isolation - reach test accuracy comparable to the original network in a similar number of iterations.

Pruning Filters for Efficient ConvNets

PaddlePaddle/PaddleOCR 31 Aug 2016

However, magnitude-based pruning of weights reduces a significant number of parameters from the fully connected layers and may not adequately reduce the computation costs in the convolutional layers due to irregular sparsity in the pruned networks.

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

NervanaSystems/distiller 1 Oct 2015

To address this limitation, we introduce "deep compression", a three stage pipeline: pruning, trained quantization and Huffman coding, that work together to reduce the storage requirement of neural networks by 35x to 49x without affecting their accuracy.

SNIP: Single-shot Network Pruning based on Connection Sensitivity

namhoonlee/snip-public ICLR 2019

To achieve this, we introduce a saliency criterion based on connection sensitivity that identifies structurally important connections in the network for the given task.

Manifold Regularized Dynamic Network Pruning

mindspore-ai/models CVPR 2021

Then, the manifold relationship between instances and the pruned sub-networks will be aligned in the training procedure.

PackNet: Adding Multiple Tasks to a Single Network by Iterative Pruning

arunmallya/packnet CVPR 2018

This paper presents a method for adding multiple tasks to a single deep neural network while avoiding catastrophic forgetting.

Network Pruning via Transformable Architecture Search

D-X-Y/NAS-Projects NeurIPS 2019

The maximum probability for the size in each distribution serves as the width and depth of the pruned network, whose parameters are learned by knowledge transfer, e. g., knowledge distillation, from the original networks.

Discovering Neural Wirings

allenai/dnw NeurIPS 2019

In this work we propose a method for discovering neural wirings.

Movement Pruning: Adaptive Sparsity by Fine-Tuning

huggingface/transformers NeurIPS 2020

Magnitude pruning is a widely used strategy for reducing model size in pure supervised learning; however, it is less effective in the transfer learning regime that has become standard for state-of-the-art natural language processing applications.