.png){kind=icon}
Return
Quantization-Aware Pruning Criterion for Industrial Applications
.png){kind=icon}
PRE
AI
DOI：10.1109/TIE.2021.3070517.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
Y In this study, we propose a simple and effective quantization-aware neural network pruning method to prune and quantize deep neural networks and then run them on lightweight and cost-effective embedded industrial devices. The proposed pruning method provides a suitable pruning strategy for a given bit precision before performing the subsequent quantization. In particular, the dropout rate is set to be bounded below by the maximum number of quantization levels, which enables the regulation of the tradeoff between sparsity and performance loss. The Kullback-Leibler divergence term of the evidence lower bound is also weighted for adjusting the performance loss. From learned dropout rates, efficient bit allocation is provided to maintain randomness even through the quantization step. In supervised and reinforcement learning applications, the proposed quantization-aware pruning method was proven to be successful in obtaining sparse and lightweight neural networks while maintaining their performance even after quantization. In particular, a real inverted pendulum system was used to determine whether the proposed method works well for actual physical control systems.
Keywords:
Network compression
network pruning
sparse variational dropout
uniform scalar quantization
AI Summary
Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.
Journal
IF:
7.2
Papers:
1.8W
Citations:
9.8W
Organization
No organization information available
Cited Papers
.png){kind=icon}
ShareSaveShareSaveShareSaveShareSaveShareSave
View more