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

Medical image analysis framework merging ANTsPy and deep learning

YOLOX + ROS(1, 2) object detection package

This is the pytorch implementation for the paper: Generalizable Mixed-Precision Quantization via Attribution Rank Preservation, which is accepted to ICCV2021.

基于Flask开发后端、VUE开发前端框架，在WEB端部署YOLOv5目标检测模型

Semantic Segmentation in Pytorch

Hepsiburada - Hepsiburada Urun Bilgisi Cekme

This repository provides the code for MedViLL(Medical Vision Language Learner).

FocusFace: Multi-task Contrastive Learning for Masked Face Recognition

Data manipulation and transformation for audio signal processing, powered by PyTorch

This is the official pytorch implementation of AutoDebias, an automatic debiasing method for recommendation.

SegTransVAE: Hybrid CNN - Transformer with Regularization for medical image segmentation

Robust fine-tuning of zero-shot models

Repo for WWW 2022 paper: Progressively Optimized Bi-Granular Document Representation for Scalable Embedding Based Retrieval

MaRS - a recursive filtering framework that allows for truly modular multi-sensor integration

Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networks

Official PyTorch implementation of the paper "Recycling Discriminator: Towards Opinion-Unaware Image Quality Assessment Using Wasserstein GAN", accepted to ACM MM 2021 BNI Track.

Paddle-Adversarial-Toolbox (PAT) is a Python library for Deep Learning Security based on PaddlePaddle.

Towards Representation Learning for Atmospheric Dynamics (AtmoDist)

Controlling the MicriSpotAI robot from scratch

The software associated with a paper accepted at EMNLP 2021 titled "Open Knowledge Graphs Canonicalization using Variational Autoencoders".