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

This is an official implementation of our CVPR 2021 paper "Bottom-Up Human Pose Estimation Via Disentangled Keypoint Regression" (https://arxiv.org/abs/2104.02300)

[CVPR'21] Learning to Recommend Frame for Interactive Video Object Segmentation in the Wild

Multi-Anchor Active Domain Adaptation for Semantic Segmentation (ICCV 2021 Oral)

《A-CNN: Annularly Convolutional Neural Networks on Point Clouds》(2019)

Codebase for the self-supervised goal reaching benchmark introduced in the LEXA paper

Wordle Env: A Daily Word Environment for Reinforcement Learning

PyTorch code of my WACV 2022 paper Improving Model Generalization by Agreement of Learned Representations from Data Augmentation

A library for answering questions using data you cannot see

SpineAI Bilsky Grading With Python

Official PyTorch implementation of the paper "TEMOS: Generating diverse human motions from textual descriptions"

This is the repo for the paper `SumGNN: Multi-typed Drug Interaction Prediction via Efficient Knowledge Graph Summarization'. (published in Bioinformatics'21)

Pmapper is a super-resolution and deconvolution toolkit for python 3.6+

Code for ECIR'20 paper Diagnosing BERT with Retrieval Heuristics

TeachMyAgent is a testbed platform for Automatic Curriculum Learning methods in Deep RL.

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks

RRxIO - Robust Radar Visual/Thermal Inertial Odometry: Robust and accurate state estimation even in challenging visual conditions.

[NeurIPS 2020] Code for the paper "Balanced Meta-Softmax for Long-Tailed Visual Recognition"

Project looking into use of autoencoder for semi-supervised learning and comparing data requirements compared to supervised learning.

Auto-updating data to assist in investment to NEPSE

A PyTorch implementation of the paper "Semantic Image Synthesis via Adversarial Learning" in ICCV 2017