arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.00523cs.CV

Automatic Pruning for Quantized Neural Networks

Monash University(莫纳什大学);The University of Adelaide(阿德莱德大学)

Luis Guerra, Bohan Zhuang, Ian Reid, Tom Drummond

英文摘要:

Neural network quantization and pruning are two techniques commonly used to reduce the computational complexity and memory footprint of these models for deployment. However, most existing pruning strategies operate on full-precision and cannot be directly applied to discrete parameter distributions after quantization. In contrast, we study a combination of these two techniques to achieve further network compression. In particular, we propose an effective pruning strategy for selecting redundant low-precision filters. Furthermore, we leverage Bayesian optimization to efficiently determine the pruning ratio for each layer. We conduct extensive experiments on CIFAR-10 and ImageNet with various architectures and precisions. In particular, for ResNet-18 on ImageNet, we prune 26.12% of the model size with Binarized Neural Network quantization, achieving a top-1 classification accuracy of 47.32% in a model of 2.47 MB and 59.30% with a 2-bit DoReFa-Net in 4.36 MB.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑