i-SpaSP: Structured Neural Pruning via Sparse Signal Recovery

This is a public code repository for the publication:

i-SpaSP: Structured Neural Pruning via Sparse Signal Recovery
Cameron R Wolfe, Anastasios Kyrillidis

Environment/Dependencies

Requires anaconda to be installed (python3) Anaconda can be installed at https://www.anaconda.com/products/individual

conda create -n ispasp python=3.6 anaconda
conda activate ispasp
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
pip install -r requirements.txt

File Arrangement

Here we summarize all files present in this repo and their purpose.

+-- prune_resnet34_ispasp.py : i-SpaSP implementation for ResNet34
+-- prune_mobilenetv2_ispasp.py : i-SpaSP implementation for MobileNetV2
+-- lib/ : utility/helper functions
|   +-- data.py: helper functions for handling data
|   +-- utils.py: helper functions for computing performance metrics
+-- scripts/ : contains python scripts for running pruning experiments
|   +-- prune_rn34_ispasp.py: run an i-SpaSP pruning experiment for ResNet34  
|   +-- prune_mbnv2_ispasp.py: run an i-SpaSP pruning experiment for MobileNetV2
+-- requirements.txt : dependencies for pruning experiments

i-SpaSP: Structured Neural Pruning via Sparse Signal Recovery

Related tags

Overview

i-SpaSP: Structured Neural Pruning via Sparse Signal Recovery

Environment/Dependencies

File Arrangement

Owner

Cameron Ronald Wolfe

Code of Puregaze: Purifying gaze feature for generalizable gaze estimation, AAAI 2022.

Adversarial vulnerability of powerful near out-of-distribution detection

Neural network for recognizing the gender of people in photos

Code for KDD'20 "An Efficient Neighborhood-based Interaction Model for Recommendation on Heterogeneous Graph"

QKeras: a quantization deep learning library for Tensorflow Keras

DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

Attention-driven Robot Manipulation (ARM) which includes Q-attention

The implementation of the CVPR2021 paper "Structure-Aware Face Clustering on a Large-Scale Graph with 10^7 Nodes"

scikit-learn inspired API for CRFsuite

【CVPR 2021, Variational Inference Framework, PyTorch】 From Rain Generation to Rain Removal

TorchX: A PyTorch Extension Library for More Efficient Deep Learning

Code for ICCV2021 paper PARE: Part Attention Regressor for 3D Human Body Estimation

Scripts and a shader to get you started on setting up an exported Koikatsu character in Blender.

Colour detection is necessary to recognize objects, it is also used as a tool in various image editing and drawing apps.

Adaptive, interpretable wavelets across domains (NeurIPS 2021)

A PyTorch implementation of the paper Mixup: Beyond Empirical Risk Minimization in PyTorch

Oriented Response Networks, in CVPR 2017

Rendering Point Clouds with Compute Shaders

Adaptive Prototype Learning and Allocation for Few-Shot Segmentation (CVPR 2021)

Detecting Human-Object Interactions with Object-Guided Cross-Modal Calibrated Semantics