Back to Mvision

Dynamic network surgery 修剪+拼接恢复

CNN/Deep_Compression/pruning/DNS/readme.md

latest2.4 KB
Original Source

Dynamic network surgery 修剪+拼接恢复

NIPS’16论文链接

代码链接

这篇文章也是关于参数的修剪,但是多了一个拼接的步骤,可以大幅度恢复修剪造成的精度损失,并且能有效的提升压缩率。

Experimental Results

The authors conducted experiments on several models including LeNet-5, LeNet-300-100 and AlexNet. The experimental results can be summarized as follows:

ModelTop-1 ErrorParametersIterationsCompression
LeNet-5 reference0.91%431K10K
LeNet-5 pruned0.91%4.0K16K108$$\times$$
LeNet-100-300 reference2.28%267K10K
LeNet-100-300 pruned1.99%4.8K25K56$$\times$$
AlexNet reference43.42%/-61M450K
AlexNet pruned43.09%/19.99%3.45M700K17.7$$\times$$

More detail comparison with work of Han et. al. on AlexNet using single crop validation on ImageNet are shown as follows:

LayerParametersRemaining Parameters Rate of Han et. al.(%)Remaining Parameters Rate(%)
conv135K~84%53.8%
conv2307K~38%40.6%
conv3885K~35%29.0%
conv4664K~37%32.3%
conv5443K~37%32.5%
fc138M~9%3.7%
fc217M~9%6.6%
fc34M~25%4.6%
Total61M~11%5.7%