Abstract:Exploring the expected quantizing scheme with suitable mixed-precision policy is the key to compress deep neural networks (DNNs) in high efficiency and accuracy. This exploration implies heavy workloads for domain experts, and an automatic compression method is needed. However, the huge search space of the automatic method introduces plenty of computing budgets that make the automatic process challenging to be applied in real scenarios. In this paper, we propose an end-to-end framework named AutoQNN, for automatically quantizing different layers utilizing different schemes and bitwidths without any human labor. AutoQNN can seek desirable quantizing schemes and mixed-precision policies for mainstream DNN models efficiently by involving three techniques: quantizing scheme search (QSS), quantizing precision learning (QPL), and quantized architecture generation (QAG). QSS introduces five quantizing schemes and defines three new schemes as a candidate set for scheme search, and then uses the Differentiable Neural Architecture Search (DNAS) algorithm to seek the layer- or model-desired scheme from the set. QPL is the first method to learn mixed-precision policies by reparameterizing the bitwidths of quantizing schemes, to the best of our knowledge. QPL optimizes both classification loss and precision loss of DNNs efficiently and obtains the relatively optimal mixed-precision model within limited model size and memory footprint. QAG is designed to convert arbitrary architectures into corresponding quantized ones without manual intervention, to facilitate end-to-end neural network quantization. We have implemented AutoQNN and integrated it into Keras. Extensive experiments demonstrate that AutoQNN can consistently outperform state-of-the-art quantization. For 2-bit weight and activation of AlexNet and ResNet18, AutoQNN can achieve the accuracy results of 59.75% and 68.86%, respectively, and obtain accuracy improvements by up to 1.65% and 1.74%, respectively, compared with state-of-the-art methods. Especially, compared with the full-precision AlexNet and ResNet18, the 2-bit models only slightly incur accuracy degradation by 0.26% and 0.76%, respectively, which can fulfill practical application demands.

Data-Free Quantization with Accurate Activation Clipping and Adaptive Batch Normalization

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

Unified Data-Free Compression: Pruning and Quantization without Fine-Tuning

Data-Free Quantization Through Weight Equalization and Bias Correction

Incremental Network Quantization: Towards Lossless CNNs with Low-Precision Weights

Redistribution of Weights and Activations for AdderNet Quantization

Weight Normalization based Quantization for Deep Neural Network Compression

LQ-Nets: Learned Quantization for Highly Accurate and Compact Deep Neural Networks

Adaptive Layerwise Quantization for Deep Neural Network Compression

Residual Quantization for Low Bit-Width Neural Networks.

Efficient and Robust Quantization-aware Training via Adaptive Coreset Selection

Instance-Aware Dynamic Neural Network Quantization

AutoQNN: An End-to-End Framework for Automatically Quantizing Neural Networks

Fine-grained Data Distribution Alignment for Post-Training Quantization

Weighted-Entropy-Based Quantization for Deep Neural Networks

Data-Free Quantization via Mixed-Precision Compensation without Fine-Tuning

Balanced Quantization: An Effective and Efficient Approach to Quantized Neural Networks

AE-Qdrop: Towards Accurate and Efficient Low-Bit Post-Training Quantization for A Convolutional Neural Network

Error-aware Quantization through Noise Tempering

Deep Neural Network Compression With Single and Multiple Level Quantization

MedQ: Lossless ultra-low-bit neural network quantization for medical image segmentation