Distance-aware Quantization
Not on the map. Why?
Abstract (arXiv)
We address the problem of network quantization, that is, reducing bit-widths of weights and/or activations to lighten network architectures. Quantization methods use a rounding function to map full-precision values to the nearest quantized ones, but this operation is not differentiable. There are mainly two approaches to training quantized networks with gradient-based optimizers. First, a straight-through estimator (STE) replaces the zero derivative of the rounding with that of an identity function, which causes a gradient mismatch problem. Second, soft quantizers approximate the rounding with continuous functions at training time, and exploit the rounding for quantization at test time. This alleviates the gradient mismatch, but causes a quantizer gap problem. We alleviate both problems in a unified framework. To this end, we introduce a novel quantizer, dubbed a distance-aware quantizer (DAQ), that mainly consists of a distance-aware soft rounding (DASR) and a temperature controller. To alleviate the gradient mismatch problem, DASR approximates the discrete rounding with the kernel soft argmax, which is based on our insight that the quantization can be formulated as a distance-based assignment problem between full-precision values and quantized ones. The controller adjusts the temperature parameter in DASR adaptively according to the input, addressing the quantizer gap problem. Experimental results on standard benchmarks show that DAQ outperforms the state of the art significantly for various bit-widths without bells and whistles.
Key facts. Cited by 19 papers in the map (data through 7 Oct 2026). Builds on 46 of its 59 references that are in the map. First posted on arXiv 16 Aug 2021.
Builds on
46 of 59 references.
References · 46
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2015Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networkscited by 9,617
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
- 2014Fully Convolutional Networks for Semantic Segmentationcited by 4,845
- 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,773
- 2014Going Deeper with Convolutionscited by 4,231
- 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,708
- 2016Aggregated Residual Transformations for Deep Neural Networkscited by 2,540
- 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,291
- 2013Visualizing and Understanding Convolutional Networkscited by 2,278
34 more
- 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,031
- 2016The Concrete Distribution: A Continuous Relaxation of Discrete Random Variablescited by 1,135
- 2016XNOR-Net: ImageNet Classification Using Binary Convolutional Neural Networkscited by 829
- 2019Learned Step Size Quantizationcited by 604
- 2015BinaryConnect: Training Deep Neural Networks with binary weights during propagationscited by 571
- 2016DoReFa-Net: Training Low Bitwidth Convolutional Neural Networks with Low Bitwidth Gradientscited by 557
- 2014Semantic Image Segmentation with Deep Convolutional Nets and Fully Connected CRFscited by 552
- 2018Quantizing deep convolutional networks for efficient inference: A whitepapercited by 507
- 2018PACT: Parameterized Clipping Activation for Quantized Neural Networkscited by 456
- 2016Accelerating the Super-Resolution Convolutional Neural Networkcited by 393
- 2017End-to-End Learning of Geometry and Context for Deep Stereo Regressioncited by 359
- 2018Model compression via distillation and quantizationcited by 287
- 2018LQ-Nets: Learned Quantization for Highly Accurate and Compact Deep Neural Networkscited by 272
- 2019Differentiable Soft Quantization: Bridging Full-Precision and Low-Bit Neural Networkscited by 234
- 2018Bi-Real Net: Enhancing the Performance of 1-bit CNNs With Improved Representational Capability and Advanced Training Algorithmcited by 233
- 2017Incremental Network Quantization: Towards Lossless CNNs with Low-Precision Weightscited by 211
- 2016Trained Ternary Quantizationcited by 203
- 2016Ternary Weight Networkscited by 195
- 2020LSQ+: Improving low-bit quantization through learnable offsets and better initializationcited by 180
- 2019Forward and Backward Information Retention for Accurate Binary Neural Networkscited by 169
- 2018Learning to Quantize Deep Networks by Optimizing Quantization Intervals with Task Losscited by 157
- 2019Quantization Networkscited by 151
- 2017Soft-to-Hard Vector Quantization for End-to-End Learning Compressible Representationscited by 149
- 2019Additive Powers-of-Two Quantization: An Efficient Non-uniform Discretization for Neural Networkscited by 143
- 2017Deep Learning with Low Precision by Half-wave Gaussian Quantizationcited by 115
- 2017Towards Effective Low-bitwidth Convolutional Neural Networkscited by 76
- 2019SFNet: Learning Object-aware Semantic Correspondencecited by 60
- 2016Loss-aware Binarization of Deep Networkscited by 56
- 2020PROFIT: A Novel Training Method for sub-4-bit MobileNet Modelscited by 50
- 2018Value-aware Quantization for Training and Inference of Neural Networkscited by 49
- 2015Neural Networks with Few Multiplicationscited by 46
- 2017Improving Landmark Localization with Semi-Supervised Learningcited by 39
- 2018Loss-aware Weight Quantization of Deep Networkscited by 31
- 2016Overcoming Challenges in Fixed Point Training of Deep Convolutional Networkscited by 9
Built on by
Citing papers per year: 2021: 3; 2022: 1; 2023: 5; 2024: 5; 2025: 3; 2026: 2.
19 papers.
Sort:
By area:
- 2021Nonuniform-to-Uniform Quantization: Towards Accurate Quantization via Generalized Straight-Through Estimationcited by 78
- 2021IntraQ: Learning Synthetic Images with Intra-Class Heterogeneity for Zero-Shot Network Quantizationcited by 43
- 2021Sharpness-aware Quantization for Deep Neural Networkscited by 25
- 2024Instance-Aware Group Quantization for Vision Transformerscited by 23
- 2023Quantized Feature Distillation for Network Quantizationcited by 13
- 2024LRQ: Optimizing Post-Training Quantization for Large Language Models by Learning Low-Rank Weight-Scaling Matricescited by 11
- 2023EQ-Net: Elastic Quantization Neural Networkscited by 10
- 2023MBQuant: A Novel Multi-Branch Topology Method for Arbitrary Bit-width Network Quantizationcited by 6
- 2024Custom Gradient Estimators are Straight-Through Estimators in Disguisecited by 6
- 2025AccuQuant: Simulating Multiple Denoising Steps for Quantizing Diffusion Modelscited by 3
- 2022Exploiting the Partly Scratch-off Lottery Ticket for Quantization-Aware Trainingcited by 2
- 2023On Model Explanations with Transferable Neural Pathwayscited by 1
7 more
- 2024Scheduling Weight Transitions for Quantization-Aware Trainingcited by 1
- 2024Toward INT4 Fixed-Point Training via Exploring Quantization Error for Gradientscited by 1
- 2023Understanding weight-magnitude hyperparameters in training binary networkscited by 0
- 2025Cauchy-Schwarz Regularizerscited by 0
- 2025Precision Neural Network Quantization via Learnable Adaptive Modulescited by 0
- 2026HESTIA: A Hessian-Guided Differentiable Quantization-Aware Training Framework for Extremely Low-Bit LLMscited by 0
- 2026Neural Network Quantization by Learning Low-Loss Subspacescited by 0
- 2026Neural Network Quantization by Learning Low-Loss Subspaces23 Jun 2026
- 2026HESTIA: A Hessian-Guided Differentiable Quantization-Aware Training Framework for Extremely Low-Bit LLMs28 Jan 2026
- 2025AccuQuant: Simulating Multiple Denoising Steps for Quantizing Diffusion Models23 Oct 2025
- 2025Precision Neural Network Quantization via Learnable Adaptive Modules24 Apr 2025
- 2025Cauchy-Schwarz Regularizers3 Mar 2025
- 2024Toward INT4 Fixed-Point Training via Exploring Quantization Error for Gradients17 Jul 2024
- 2024LRQ: Optimizing Post-Training Quantization for Large Language Models by Learning Low-Rank Weight-Scaling Matrices16 Jul 2024
- 2024Custom Gradient Estimators are Straight-Through Estimators in Disguise8 May 2024
- 2024Scheduling Weight Transitions for Quantization-Aware Training30 Apr 2024
- 2024Instance-Aware Group Quantization for Vision Transformers1 Apr 2024
- 2023On Model Explanations with Transferable Neural Pathways18 Sep 2023
- 2023EQ-Net: Elastic Quantization Neural Networks15 Aug 2023
7 more
- 2023Quantized Feature Distillation for Network Quantization20 Jul 2023
- 2023MBQuant: A Novel Multi-Branch Topology Method for Arbitrary Bit-width Network Quantization14 May 2023
- 2023Understanding weight-magnitude hyperparameters in training binary networks4 Mar 2023
- 2022Exploiting the Partly Scratch-off Lottery Ticket for Quantization-Aware Training12 Nov 2022
- 2021Nonuniform-to-Uniform Quantization: Towards Accurate Quantization via Generalized Straight-Through Estimation29 Nov 2021
- 2021Sharpness-aware Quantization for Deep Neural Networks24 Nov 2021
- 2021IntraQ: Learning Synthetic Images with Intra-Class Heterogeneity for Zero-Shot Network Quantization17 Nov 2021
- 2021Nonuniform-to-Uniform Quantization: Towards Accurate Quantization via Generalized Straight-Through Estimationcited by 78
- 2021IntraQ: Learning Synthetic Images with Intra-Class Heterogeneity for Zero-Shot Network Quantizationcited by 43
- 2021Sharpness-aware Quantization for Deep Neural Networkscited by 25
- 2024Instance-Aware Group Quantization for Vision Transformerscited by 23
- 2023Quantized Feature Distillation for Network Quantizationcited by 13
- 2024LRQ: Optimizing Post-Training Quantization for Large Language Models by Learning Low-Rank Weight-Scaling Matricescited by 11
- 2023EQ-Net: Elastic Quantization Neural Networkscited by 10
- 2023MBQuant: A Novel Multi-Branch Topology Method for Arbitrary Bit-width Network Quantizationcited by 6
- 2024Custom Gradient Estimators are Straight-Through Estimators in Disguisecited by 6
- 2022Exploiting the Partly Scratch-off Lottery Ticket for Quantization-Aware Trainingcited by 2
- 2023On Model Explanations with Transferable Neural Pathwayscited by 1
- 2024Scheduling Weight Transitions for Quantization-Aware Trainingcited by 1
12 most cited of 18 All 18 (JSON)
- 2026Neural Network Quantization by Learning Low-Loss Subspaces23 Jun 2026
- 2026HESTIA: A Hessian-Guided Differentiable Quantization-Aware Training Framework for Extremely Low-Bit LLMs28 Jan 2026
- 2025Precision Neural Network Quantization via Learnable Adaptive Modules24 Apr 2025
- 2025Cauchy-Schwarz Regularizers3 Mar 2025
- 2024Toward INT4 Fixed-Point Training via Exploring Quantization Error for Gradients17 Jul 2024
- 2024LRQ: Optimizing Post-Training Quantization for Large Language Models by Learning Low-Rank Weight-Scaling Matrices16 Jul 2024
- 2024Custom Gradient Estimators are Straight-Through Estimators in Disguise8 May 2024
- 2024Scheduling Weight Transitions for Quantization-Aware Training30 Apr 2024
- 2024Instance-Aware Group Quantization for Vision Transformers1 Apr 2024
- 2023On Model Explanations with Transferable Neural Pathways18 Sep 2023
- 2023EQ-Net: Elastic Quantization Neural Networks15 Aug 2023
- 2023Quantized Feature Distillation for Network Quantization20 Jul 2023
12 newest of 18 All 18 (JSON)