30 open-source projects similar to ekdeepslubana/flowandprune, ranked by how many features they have in common. Compare stars, activity and what each one does to find the best Flowandprune alternative.
PocketFlow is an integrated toolkit for deep learning model compression, distributed training, and mobile format optimization. It provides a system for reducing the size and complexity of neural networks to improve inference efficiency, featuring a dedicated engine for knowledge distillation and a mobile model optimizer. The framework differentiates itself through an automated hyperparameter tuning system that uses reinforcement learning and statistical models to determine optimal compression ratios and layer-wise bit allocation. It also includes a distributed training system that utilizes mu
Sample code use for NeurIPS 2021 paper: Sanity Checks for Lottery Tickets: Does Your Winning Ticket Really Win the Jackpot?
This method identifies a binary weight or binary weight and activation subnetwork within a randomly initialized network that achieves performance comparable to, and sometimes better than, a weight-optimized network. The resulting binarized and pruned networks that achieve comparable performance…
by Daiki Chijiwa\*, Shin’ya Yamaguchi, Yasutoshi Ida, Kenji Umakoshi, Tomohiro Inoue
ICLR 2023 Pruning Deep Neural Networks from a Sparsity Perspective
This is the official repo for ICLR 2023 Paper "Bit-Pruning: A Sparse Multiplication-Less Dot-Product" Yusuke Sekikawa and Shingo Yashima
This repository contains the codes for the following NeurIPS-2019 paper
This repository contains source code for the ICLR 2022 paper How many degrees of freedom do we need to train deep networks: a loss landscape perspective by Brett W. Larsen, Sanislav Fort, Nic Becker, and Surya Ganguli (arXiv version).
NeurIPS 2022 "Losses Can Be Blessings: Routing Self-Supervised Speech Representations Towards Efficient Multilingual and Multitask Speech Processing" by Yonggan Fu, Yang Zhang, Kaizhi Qian, Zhifan Ye, Zhongzhi Yu, Cheng-I Lai, Yingyan Lin
This codebase was developed by Jonathan Frankle and David Bieber at Google during the summer of 2018.
This framework implements key experiments on the sparse double descent phenomenon, as demonstrated in the following paper:
Repository with code to reproduce the results and checkpoints for compressed networks in our paper on novel pruning techniques with robust training. This repository supports all four robust training objectives: iterative adversarial training, randomized smoothing, MixTrain, and CROWN-IBP.
This code allows replicating the image experiments of AC/DC: Alternating Compressed/DeCompressed Training of Deep Neural Networks. This code allows for training compressed and decompressed WideResNet models for CIFAR-100, ResNet50 and MobileNet models for Imagenet (also for 2:4 sparsity), and…
Code for the NeurIPS 2022 paper "Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruning".
This repository contains reference implementations of all methods introduced in our ICML 2022 paper: SPDY: Accurate Pruning with Speedup Guarantees. This includes the DP algorithm for efficiently solving constrained layer-wise compression problems (see dpsolve() in spdy.py), the reparametrized…
Official implementation of the NeurIPS 2022 accepted paper "Robust Binary Models by Pruning Randomly-initialized Networks"
A PyTorch implementation of this paper.
This is the official implementation of the paper: "Layerwise Sparsity for Magnitude-based Pruning", ICLR 2021.
This repository contains the code for reproducing the results in the following paper:
Prune DNN using Alternating Direction Method of Multipliers (ADMM)
This repository is meant to provide a generic code base for neural network pruning, especially for pruning at initialization (PaI). (In preparation now, you may check our survey paper and paper collection below.)
This repository contains code for the paper SNIP: Single-shot Network Pruning based on Connection Sensitivity (ICLR 2019).
This is an PyTorch implement of the paper ``Prune Your Model Before Distill It''.
Francesco Croce\ (University of Tübingen), Maksym Andriushchenko\ (EPFL), Vikash Sehwag\ (Princeton University), Nicolas Flammarion (EPFL), Mung Chiang (Purdue University), Prateek Mittal (Princeton University), Matthias Hein (University of Tübingen)
March 15, 2019: for our most updated work on model compression and acceleration, please reference:
Hattie Zhou, Janice Lan, Rosanne Liu, Jason Yosinski
Mukund Varma T 1 , Xuxi Chen 2 , Zhenyu Zhang 2 , Tianlong Chen 2 , Subhashini Venugopalan 3 , Zhangyang Wang 2
This repo includes codes for the official implementation of the paper The Elastic Lottery Ticket Hypothesis, by Xiaohan Chen, Yu Cheng, Shuohang Wang, Zhe Gan, Jingjing Liu, Zhangyang Wang.
Sparse Training via Boosting Pruning Plasticity with Neuroregeneration Shiwei Liu,Tianlong Chen,Xiaohan Chen,Zahra Atashgahi,Lu Yin,Huanyu Kou,Li Shen,Mykola Pechenizkiy,Zhangyang Wang, Decebal Constantin Mocanu https://arxiv.org/abs/2106.10404