Abstract:Low-bit quantization emerges as one of the most promising compression approaches for deploying deep neural networks on edge devices. Mixed-precision quantization leverages a mixture of bit-widths to unleash the accuracy and efficiency potential of quantized models. However, existing mixed-precision quantization methods rely on simulations in high-performance devices to achieve accuracy and efficiency trade-offs in immense search spaces. This leads to a non-negligible gap between the estimated efficiency metrics and the actual hardware that makes quantized models far away from the optimal accuracy and efficiency, and also causes the quantization process to rely on additional high-performance devices. In this paper, we propose an On-Chip Hardware-Aware Quantization (OHQ) framework, performing hardware-aware mixed-precision quantization on deployed edge devices to achieve accurate and efficient computing. Specifically, for efficiency metrics, we built an On-Chip Quantization Aware pipeline, which allows the quantization process to perceive the actual hardware efficiency of the quantization operator and avoid optimization errors caused by inaccurate simulation. For accuracy metrics, we propose Mask-Guided Quantization Estimation technology to effectively estimate the accuracy impact of operators in the on-chip scenario, getting rid of the dependence of the quantization process on high computing power. By synthesizing insights from quantized models and hardware through linear optimization, we can obtain optimized bit-width configurations to achieve outstanding performance on accuracy and efficiency. We evaluate inference accuracy and acceleration with quantization for various architectures and compression ratios on hardware. OHQ achieves 70% and 73% accuracy for ResNet-18 and MobileNetV3, respectively, and can reduce latency by 15~30% compared to INT8 on real deployment.

CSMPQ: Class Separability Based Mixed-Precision Quantization.

CSMPQ:Class Separability Based Mixed-Precision Quantization

Pse: Mixed Quantization Framework of Neural Networks for Efficient Deployment

Hessian-based Mixed-Precision Quantization with Transition Aware Training for Neural Networks

Mixed-Precision Neural Network Quantization Via Learned Layer-Wise Importance

Mixed-Precision Quantization with Cross-Layer Dependencies

CSQ: Growing Mixed-Precision Quantization Scheme with Bi-level Continuous Sparsification

Bit-shrinking: Limiting Instantaneous Sharpness for Improving Post-training Quantization

Quantizing separable convolution of MobileNets with mixed precision

Mixed-Precision Quantization for CNN-Based Remote Sensing Scene Classification

Solving Oscillation Problem in Post-Training Quantization Through a Theoretical Perspective

BMPQ: Bit-Gradient Sensitivity Driven Mixed-Precision Quantization of DNNs from Scratch

LSMQ: A Layer-Wise Sensitivity-Based MixedPrecision Quantization Method for Bit-Flexible CNN Accelerator

MPQ-Diff: Mixed Precision Quantization for Diffusion Models

Quantization without Tears

GroupQ: Group-Wise Quantization With Multi-Objective Optimization for CNN Accelerators

SQuantizer: Simultaneous Learning for Both Sparse and Low-precision Neural Networks

DPQ: dynamic pseudo-mean mixed-precision quantization for pruned neural network

MixQuant: Mixed Precision Quantization with a Bit-width Optimization Search

HAWQV3: Dyadic Neural Network Quantization

On-Chip Hardware-Aware Quantization for Mixed Precision Neural Networks