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

The official PyTorch code implementation of "Personalized Trajectory Prediction via Distribution Discrimination" in ICCV 2021.

Implementation of the master's thesis "Temporal copying and local hallucination for video inpainting".

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

Airborne magnetic data of the Osborne Mine and Lightning Creek sill complex, Australia

Read number plates with https://platerecognizer.com/

DumpSMBShare - A script to dump files and folders remotely from a Windows SMB share

Generalized Matrix Means for Semi-Supervised Learning with Multilayer Graphs

Blender Add-On for slicing meshes with planes

Implementation based on Paper - Learning a Probabilistic Latent Space of Object Shapes via 3D Generative-Adversarial Modeling

The code of Zero-shot learning for low-light image enhancement based on dual iteration

Code accompanying "Evolving spiking neuron cellular automata and networks to emulate in vitro neuronal activity," accepted to IEEE SSCI ICES 2021

PyTorch implementation for NED. It can be used to manipulate the facial emotions of actors in videos based on emotion labels or reference styles.

📖 Deep Attentional Guided Image Filtering

"Inductive Entity Representations from Text via Link Prediction" @ The Web Conference 2021

GMFlow: Learning Optical Flow via Global Matching

Node for thenewboston digital currency network.

Some bravo or inspiring research works on the topic of curriculum learning.

Unified file system operation experience for different backend

A new play-and-plug method of controlling an existing generative model with conditioning attributes and their compositions.

A generator of point clouds dataset for PyPipes.