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.

CMix-NN: Mixed Low-Precision CNN Library for Memory-Constrained Edge Devices

Flexible high-resolution object detection on edge devices with tunable latency.

Memory-Driven Mixed Low Precision Quantization For Enabling Deep Network Inference On Microcontrollers

MCU-MixQ: A HW/SW Co-optimized Mixed-precision Neural Network Design Framework for MCUs

Enabling Mixed-Precision Quantized Neural Networks in Extreme-Edge Devices

Leveraging Automated Mixed-Low-Precision Quantization for tiny edge microcontrollers

A fine-grained mixed precision DNN accelerator using a two-stage big-little core RISC-V MCU.

CoopNet: Cooperative Convolutional Neural Network for Low-Power MCUs

A Mixed-Precision RISC-V Processor for Extreme-Edge DNN Inference

Joint Pruning and Channel-wise Mixed-Precision Quantization for Efficient Deep Neural Networks

CIMQ: A Hardware-Efficient Quantization Framework for Computing-In-Memory Based Neural Network Accelerators

MCUNet: Tiny Deep Learning on IoT Devices

Mixed Precision Quantized Neural Network Accelerator for Remote Sensing Images Classification

Free Bits: Latency Optimization of Mixed-Precision Quantized Neural Networks on the Edge

Deploy Large-Scale Deep Neural Networks in Resource Constrained IoT Devices with Local Quantization Region

Cheetah: Mixed Low-Precision Hardware & Software Co-Design Framework for DNNs on the Edge

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

Low-Precision Mixed-Computation Models for Inference on Edge

AddNet: Deep Neural Networks Using FPGA-Optimized Multipliers