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

A flexible submap-based framework towards spatio-temporally consistent volumetric mapping and scene understanding.

Official pytorch implementation of Active Learning for deep object detection via probabilistic modeling (ICCV 2021)

Code for Multiple Instance Active Learning for Object Detection, CVPR 2021

Neural networks applied in recognizing guitar chords using python, AutoML.NET with C# and .NET Core

Time Delayed NN implemented in pytorch

Efficient Training of Audio Transformers with Patchout

Tensorflow implementation of "BEGAN: Boundary Equilibrium Generative Adversarial Networks"

Amazing-Python-Scripts - 🚀 Curated collection of Amazing Python scripts from Basics to Advance with automation task scripts.

Time Dependent DFT in Tamm-Dancoff Approximation

PiCIE: Unsupervised Semantic Segmentation using Invariance and Equivariance in clustering (CVPR2021)

OcclusionFusion: realtime dynamic 3D reconstruction based on single-view RGB-D

SHIFT15M: multiobjective large-scale fashion dataset with distributional shifts

This is the repo for our work "Towards Persona-Based Empathetic Conversational Models" (EMNLP 2020)

“Robust Lightweight Facial Expression Recognition Network with Label Distribution Training”, AAAI 2021.

Unsupervised Real-World Super-Resolution: A Domain Adaptation Perspective

code from "Tensor decomposition of higher-order correlations by nonlinear Hebbian plasticity"

Seq2seq - Sequence to Sequence Learning with Keras

Official PyTorch implementation of MX-Font (Multiple Heads are Better than One: Few-shot Font Generation with Multiple Localized Experts)

Sub-tomogram-Detection - Deep learning based model for Cyro ET Sub-tomogram-Detection

RCDNet: A Model-driven Deep Neural Network for Single Image Rain Removal (CVPR2020)