FxP-QNet: A Post-Training Quantizer for the Design of Mixed Low-Precision DNNs with Dynamic Fixed-Point Representation
arXiv:2203.12091 · doi:10.1109/ACCESS.2022.3157893
Abstract
Deep neural networks (DNNs) have demonstrated their effectiveness in a wide range of computer vision tasks, with the state-of-the-art results obtained through complex and deep structures that require intensive computation and memory. Now-a-days, efficient model inference is crucial for consumer applications on resource-constrained platforms. As a result, there is much interest in the research and development of dedicated deep learning (DL) hardware to improve the throughput and energy efficiency of DNNs. Low-precision representation of DNN data-structures through quantization would bring great benefits to specialized DL hardware. However, the rigorous quantization leads to a severe accuracy drop. As such, quantization opens a large hyper-parameter space at bit-precision levels, the exploration of which is a major challenge. In this paper, we propose a novel framework referred to as the Fixed-Point Quantizer of deep neural Networks (FxP-QNet) that flexibly designs a mixed low-precision DNN for integer-arithmetic-only deployment. Specifically, the FxP-QNet gradually adapts the quantization level for each data-structure of each layer based on the trade-off between the network accuracy and the low-precision requirements. Additionally, it employs post-training self-distillation and network prediction error statistics to optimize the quantization of floating-point values into fixed-point numbers. Examining FxP-QNet on state-of-the-art architectures and the benchmark ImageNet dataset, we empirically demonstrate the effectiveness of FxP-QNet in achieving the accuracy-compression trade-off without the need for training. The results show that FxP-QNet-quantized AlexNet, VGG-16, and ResNet-18 reduce the overall memory requirements of their full-precision counterparts by 7.16x, 10.36x, and 6.44x with less than 0.95%, 0.95%, and 1.99% accuracy drop, respectively.
arXiv comment30 pages, 12 figures, 5 tables. in IEEE Access, 2022
References in corpus (32)
- Very Deep Convolutional Networks for Large-Scale Image Recognition
- Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift
- Distilling the Knowledge in a Neural Network
- Caffe: Convolutional Architecture for Fast Feature Embedding
- Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Coding
- Fully Convolutional Networks for Semantic Segmentation
- Binarized Neural Networks: Training Deep Neural Networks with Weights and Activations Constrained to +1 or -1
- DoReFa-Net: Training Low Bitwidth Convolutional Neural Networks with Low Bitwidth Gradients
- Quantized Neural Networks: Training Neural Networks with Low Precision Weights and Activations
- Deep Learning with Limited Numerical Precision
- Mixed Precision Training
- Quantizing deep convolutional networks for efficient inference: A whitepaper
- Ternary Weight Networks
- Trained Ternary Quantization
- PACT: Parameterized Clipping Activation for Quantized Neural Networks
- Binary Neural Networks: A Survey
- FPGA-based Accelerators of Deep Learning Networks for Learning and Classification: A Review
- Convolutional Neural Networks using Logarithmic Data Representation
- Learned Step Size Quantization
- Mixed Precision Quantization of ConvNets via Differentiable Neural Architecture Search
- WRPN: Wide Reduced-Precision Networks
- Loss-aware Binarization of Deep Networks
- Resiliency of Deep Neural Networks under Quantization
- EIE: Efficient Inference Engine on Compressed Deep Neural Network
- Are All Layers Created Equal?
- Loss-aware Weight Quantization of Deep Networks
- A Unified Framework of DNN Weight Pruning and Weight Clustering/Quantization Using ADMM
- Training Competitive Binary Neural Networks from Scratch
- Quantization for Rapid Deployment of Deep Neural Networks
- Hardware-Centric AutoML for Mixed-Precision Quantization
- Low-Precision Batch-Normalized Activations
- Multi-Precision Policy Enforced Training (MuPPET): A precision-switching strategy for quantised fixed-point training of CNNs