← 开源
Eric-mingjie

rethinking-network-pruning

Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)

InfrastructureModel optimizationPython
在 GitHub 打开
增长势头
+024 小时新增 Star0.0%
1.51k
Star
286
Fork
+1
本周
4
贡献者
创建于 2018-10-11 · 更新于 2026-10-10 · 今日第 10122 名
主要开发者

Eric-mingjie/rethinking-network-pruning 排名第 10,122,有 1,513 个星标。

同类别比较

Eric-mingjie/rethinking-network-pruning 排名第 10,122,此处有 8 个同类条目;o19s/elasticsearch-learning-to-rank 排名第 10,106。

同类别比较
名称排名指标
BennyKok/comfyui-deploy10,070
语言:
TypeScript
星标数:
1,533
分叉数:
222
简介:
An open source `vercel` like deployment platform for Comfy UI
e2b-dev/desktop10,076
语言:
Python
星标数:
1,532
分叉数:
186
简介:
E2B Desktop Sandbox for LLMs. E2B Sandbox with desktop graphical environment that you can connect to any LLM for secure computer use.
mxsm/rocketmq-rust10,084
语言:
Rust
星标数:
1,529
分叉数:
289
o19s/elasticsearch-learning-to-rank10,106
语言:
Java
星标数:
1,520
分叉数:
373
简介:
Plugin to integrate Learning to Rank (aka machine learning for better relevance) with Elasticsearch
limecloud/lime10,186
语言:
TypeScript
星标数:
1,486
分叉数:
208
简介:
Full-stack AI agent for coding, files, terminals, tools, research, content, multimodal work, and multi-agent workflows.
NVIDIA/deepops10,213
语言:
Shell
星标数:
1,477
分叉数:
361
简介:
Tools for building GPU clusters
SeldonIO/seldon-server10,214
语言:
Java
星标数:
1,477
分叉数:
297
简介:
Machine Learning Platform and Recommendation Engine built on Kubernetes
chrysb/alphaclaw10,225
语言:
JavaScript
星标数:
1,474
分叉数:
233
简介:
The ultimate setup harness for OpenClaw. Deploy in minutes. Stay running for months. No CLI required.
关键参数与规格
关键参数与规格
参数值
语言Python
类别Infrastructure
子类别Model optimization
星标数1,513
分叉数286
贡献者4
创建日期2018-10-11
更新日期2026-10-10
1 天星标数0
7 天星标数1
排名10,122
上次排名10,135
排名变化13
7 天星标排名9,784
上次 7 天星标排名9,845
7 天星标排名变化61
总星标排名8,040
上次总星标排名8,028
总星标排名变化-12
国家USA
README 摘录

Rethinking the Value of Network Pruning This repository contains the code for reproducing the results, and trained ImageNet models, in the following paper: Rethinking the Value of Network Pruning. [[arXiv]]( [[OpenReview]]( Zhuang Liu\*, Mingjie Sun\*, Tinghui Zhou, Gao Huang, Trevor Darrell (\* equal contribution). Also Best Paper Award at NIPS 2018 Workshop on Compact Deep Neural Networks. Several pruning methods' implementations contained in this repo can also be readily used for other research purposes. Paper Summary Fig 1: A typical three

-stage network pruning pipeline. Our paper shows that for **structured** pruning, **training the pruned model from scratch can almost always achieve comparable or higher level of accuracy than the model obtained from the typical "training, pruning and fine-tuning" (Fig. We conclude that for those pruning methods: 1. Training a large, over-parameterized model is often not necessary to obtain an efficient final model. Learned “important” weights of the large model are typically not useful for the small pruned model

. The pruned architecture itself, rather than a set of inherited “important” weights, is more crucial to the efficiency in the final model, which suggests that in some cases pruning can be useful as an architecture search paradigm. Our results suggest the need for more careful baseline evaluations in future research on structured pruning methods. Fig 2: Difference between predefined and automatically discovered target architectures, in channel pruning. The pruning ratio x is user-specified, while a, b, c, d are

determined by the pruning algorithm. Unstructured sparse pruning can also be viewed as automatic. Our finding has different implications for predefined and automatic methods: for a predefined method, it is possible to skip the traditional "training, pruning and fine-tuning" pipeline and directly train the pruned model; for automatic methods, the pruning can be seen as a form of architecture learning. We also compare with the "Lottery Ticket Hypothesis" (Frankle & Carbin 2019), and find that with optimal learning rate, the

"winning ticket" initialization as used in Frankle & Carbin (2019) does not bring improvement over random initialization. For more details please refer to our paper. Implementation We evaluated the following seven pruning methods. Regression based feature reconstruction 4. Unstructured weight-level pruning The first six is structured while the last one is unstructured (or sparse). For CIFAR, our code is based on pytorch-classification and network-slimming. For ImageNet, we use the official Pytorch ImageNet training code. The instructions and models

are in each subfolder. For experiments on The Lottery Ticket Hypothesis, please refer to the folder cifar/lottery-ticket. Contact Feel free to discuss papers/code with us through issues/emails! com Citation If you use our code in your research, please cite:

常见问题

记录了多少星标和分叉?

Eric-mingjie/rethinking-network-pruning 有 1,513 个星标和 286 个分叉。

仓库排名是多少?

Eric-mingjie/rethinking-network-pruning 排名第 10,122。

记录的语言和类别是什么?

Eric-mingjie/rethinking-network-pruning 有 1,513 个星标,语言为 Python,类别为 Infrastructure。

哪个仓库排名最接近?

Eric-mingjie/rethinking-network-pruning 与 o19s/elasticsearch-learning-to-rank 比较,后者排名第 10,106,有 1,520 个星标。

README

Rethinking the Value of Network Pruning

This repository contains the code for reproducing the results, and trained ImageNet models, in the following paper:

Rethinking the Value of Network Pruning. [arXiv] [OpenReview]

Zhuang Liu*, Mingjie Sun*, Tinghui Zhou, Gao Huang, Trevor Darrell (* equal contribution).

ICLR 2019. Also Best Paper Award at NIPS 2018 Workshop on Compact Deep Neural Networks.

Several pruning methods' implementations contained in this repo can also be readily used for other research purposes.

Paper Summary

Fig 1: A typical three-stage network pruning pipeline.

Our paper shows that for structured pruning, training the pruned model from scratch can almost always achieve comparable or higher level of accuracy than the model obtained from the typical "training, pruning and fine-tuning" (Fig. 1) procedure. We conclude that for those pruning methods:

  1. Training a large, over-parameterized model is often not necessary to obtain an efficient final model.
  2. Learned “important” weights of the large model are typically not useful for the small pruned model.
  3. The pruned architecture itself, rather than a set of inherited “important” weights, is more crucial to the efficiency in the final model, which suggests that in some cases pruning can be useful as an architecture search paradigm.

Our results suggest the need for more careful baseline evaluations in future research on structured pruning methods.

Fig 2: Difference between predefined and automatically discovered target architectures, in channel pruning. The pruning ratio x is user-specified, while a, b, c, d are determined by the pruning algorithm. Unstructured sparse pruning can also be viewed as automatic. Our finding has different implications for predefined and automatic methods: for a predefined method, it is possible to skip the traditional "training, pruning and fine-tuning" pipeline and directly train the pruned model; for automatic methods, the pruning can be seen as a form of architecture learning.

We also compare with the "Lottery Ticket Hypothesis" (Frankle & Carbin 2019), and find that with optimal learning rate, the "winning ticket" initialization as used in Frankle & Carbin (2019) does not bring improvement over random initialization. For more details please refer to our paper.

Implementation

We evaluated the following seven pruning methods.

  1. L1-norm based channel pruning
  2. ThiNet
  3. Regression based feature reconstruction
  4. Network Slimming
  5. Sparse Structure Selection
  6. Soft filter pruning
  7. Unstructured weight-level pruning

The first six is structured while the last one is unstructured (or sparse). For CIFAR, our code is based on pytorch-classification and network-slimming. For ImageNet, we use the official Pytorch ImageNet training code. The instructions and models are in each subfolder.

For experiments on The Lottery Ticket Hypothesis, please refer to the folder cifar/lottery-ticket.

Our experiment environment is Python 3.6 & PyTorch 0.3.1.

Contact

Feel free to discuss papers/code with us through issues/emails!

sunmj15 at gmail.com
liuzhuangthu at gmail.com

Citation

If you use our code in your research, please cite:

@inproceedings{liu2018rethinking,
  title={Rethinking the Value of Network Pruning},
  author={Liu, Zhuang and Sun, Mingjie and Zhou, Tinghui and Huang, Gao and Darrell, Trevor},
  booktitle={ICLR},
  year={2019}
}
排名相邻
同类项目