Quantization-Aware Pruning Criterion for Industrial Applications
SCIE
SCOPUS
- Title
- Quantization-Aware Pruning Criterion for Industrial Applications
- Authors
- GIL, YOONHEE; PARK, JONG HYEOK; BAEK, JONGCHAN; HAN, SOOHEE
- Date Issued
- 2022-03
- Publisher
- Institute of Electrical and Electronics Engineers
- Abstract
- 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.
- URI
- https://oasis.postech.ac.kr/handle/2014.oak/110555
- DOI
- 10.1109/TIE.2021.3070517
- ISSN
- 0278-0046
- Article Type
- Article
- Citation
- IEEE Transactions on Industrial Electronics, vol. 69, no. 3, page. 3203 - 3213, 2022-03
- Files in This Item:
- There are no files associated with this item.
Items in DSpace are protected by copyright, with all rights reserved, unless otherwise indicated.