journal article·2024Open access

A Novel Quantization and Model Compression Approach for Hardware Accelerators in Edge Computing

Fangzhou He, Ke Ding, Dingjiang Yan, Jie Li, Jiajun Wang, Mingzhe Chen

Computers, Materials & Continua· Volume 80· Issue 2

Abstract

Massive computational complexity and memory requirement of artificial intelligence models impede their deployability on edge computing devices of the Internet of Things (IoT). While Power-of-Two (PoT) quantization is proposed to improve the efficiency for edge inference of Deep Neural Networks (DNNs), existing PoT schemes require a huge amount of bit-wise manipulation and have large memory overhead, and their efficiency is bounded by the bottleneck of computation latency and memory footprint. To tackle this challenge, we present an efficient inference approach on the basis of PoT quantization and model compression. An integer-only scalar PoT quantization (IOS-PoT) is designed jointly with a distribution loss regularizer, wherein the regularizer minimizes quantization errors and training disturbances. Additionally, two-stage model compression is developed to effectively reduce memory requirement, and alleviate bandwidth usage in communications of networked heterogenous learning systems. The product look-up table (P-LUT) inference scheme is leveraged to replace bit-shifting with only indexing and addition operations for achieving low-latency computation and implementing efficient edge accelerators. Finally, comprehensive experiments on Residual Networks (ResNets) and efficient architectures with Canadian Institute for Advanced Research (CIFAR), ImageNet, and Real-world Affective Faces Database (RAF-DB) datasets, indicate that our approach achieves 2×∼10× improvement in the reduction of both weight size and computation cost in comparison to state-of-the-art methods. A P-LUT accelerator prototype is implemented on the Xilinx KV260 Field Programmable Gate Array (FPGA) platform for accelerating convolution operations, with performance results showing that P-LUT reduces memory footprint by 1.45×, achieves more than 3× power efficiency and 2× resource efficiency, compared to the conventional bit-shifting scheme.
Source: openalex_inverted_index · Quality: baseline

Research context

OpenAlex topics: Distributed and Parallel Computing Systems

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

References (41)

Show source-provided references
1. Quantization and training of neural networks for efficient integer-arithmetic-only inference, Quantization and training of neural networks for efficient integer-arithmetic-only inference
2. Q. Jin et al., “F8Net: Fixed-point 8-bit only multiplication for network quantization,” arXiv preprint arXiv:2202.05239, 2022.
3. LSQ+: Improving low-bit quantization through learnable offsets and better initialization
4. A block minifloat representation for training deep neural networks
5. Profit: A novel training method for sub-4-bit mobilenet models
6. A. Zhou, A. Yao, Y. Guo, L. Xu, and Y. Chen, “Incremental network quantization: Towards lossless cnns with low-precision weights,” arXiv preprint arXiv:1702.03044, 2017.
7. J. Choi, Z. Wang, S. Venkataramani, P. I. -J. Chuang, V. Srinivasan and K. Gopalakrishnan, “PACT: Parameterized clipping activation for quantized neural networks,” arXiv preprint arXiv:1805.06085, 2018.
8. Y. Li, X. Dong, and W. Wang, “Additive powers-of-two quantization: An efficient non-uniform discretization for neural networks,” arXiv preprint arXiv:1909.13144, 2019.
9. Deepshift: Towards multiplication-less neural networks
10. Iteratively training look-up tables for network quantization10.1109/JSTSP.2020.3005030 ↗
11. S. Han, H. Mao, and W. J. Dally, “Deep compression: Compressing deep neural networks with pruning, trained quantization and huffman coding,” arXiv preprint arXiv:1510.00149, 2015.
12. Y. Gong, L. Liu, M. Yang, and L. Bourdev, “Compressing deep convolutional networks using vector quantization,” arXiv preprint arXiv:1412.6115, 2014.
13. Universal deep neural network compression10.1109/JSTSP.2020.2975903 ↗
14. Lane compression: A lightweight lossless compression method for machine learning on embedded systems10.1145/3431815 ↗
15. Memory- and communication-aware model compression for distributed deep learning inference on IoT10.1145/3358205 ↗
16. EncoDeep: Realizing bit-flexible encoding for deep neural networks10.1145/3391901 ↗
17. Model loss and distribution analysis of regression problems in machine learning
18. Early stopping for iterative regularization with general loss functions
19. Rethinking feature distribution for loss functions in image classification
20. Adaptive weighted losses with distribution approximation for efficient consistency-based semi-supervised learning10.1109/TCSVT.2022.3186041 ↗
21. Learning to binarize convolutional neural networks with adaptive neural encoder
22. Compensated-DNN: Energy efficient low-precision deep neural networks by compensating quantization errors
23. VecQ: Minimal loss DNN model compression with vectorized weight quantization10.1109/TC.2020.2995593 ↗
24. General plane-based clustering with distribution loss10.1109/TNNLS.2020.3016078 ↗
25. LUTNet: Rethinking inference in fpga soft logic
26. Y. Bengio, N. Léonard, and A. Courville, “Estimating or propagating gradients through stochastic neurons for conditional computation,” arXiv preprint arXiv:1308.3432, 2013.
27. A. Lempel and J. Ziv, “Lempel-ziv–markov chain algorithm,” Accessed: Dec. 7, 2023. [Online]. Available: https://en.wikipedia.org/wiki/Lempel-Ziv-Markov_chain_algorithm
28. A. Krizhevsky and G. Hinton, “Learning multiple layers of features from tiny images,” Accessed: Dec. 7, 2023. 2009. [Online]. Available: http://www.cs.utoronto.ca/~kriz/learning-features-2009-TR.pdf
29. ImageNet: A large-scale hierarchical image database
30. Reliable crowdsourcing and deep locality-preserving learning for expression recognition in the wild
31. Bi-real Net: Enhancing the performance of 1-bit CNNs with improved representational capability and advanced training algorithm
32. V. Sovrasov, “ptflops: A flops counting tool for neural networks in pytorch framework,” Accessed: Dec. 7, 2023. 2020. [Online]. Available: https://github.com/sovrasov/flops-counter.pytorch
33. Weight normalization: A simple reparameterization to accelerate training of deep neural networks
34. Learning to quantize deep networks by optimizing quantization intervals with task loss
35. D. Miyashita, E. H. Lee, and B. Murmann, “Convolutional neural networks using logarithmic data representation,” arXiv preprint arXiv:1603.01025, 2016.
36. Deep residual learning for image recognition
37. Y. Idelbayev, “Proper ResNet implementation for CIFAR10/CIFAR100 in PyTorch,” Accessed: Dec. 7, 2023. 2020. [Online]. Available: https://github.com/akamaster/pytorch_resnet_cifar10
38. LQ-Nets: Learned quantization for highly accurate and compact deep neural networks
39. F. N. Iandola, S. Han, M. W. Moskewicz, K. Ashraf, W. J. Dally and K. Keutzer, “SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5 mb model size,” arXiv preprint arXiv:1602.07360, 2016.
40. ShuffleNet v2: Practical guidelines for efficient CNN architecture design
41. Suppressing uncertainties for large-scale facial expression recognition