Nonuniform-to-Uniform Quantization: Towards Accurate Quantization via Generalized Straight-Through Estimation
Not on the map. Why?
Abstract (arXiv)
The nonuniform quantization strategy for compressing neural networks usually achieves better performance than its counterpart, i.e., uniform strategy, due to its superior representational capacity. However, many nonuniform quantization methods overlook the complicated projection process in implementing the nonuniformly quantized weights/activations, which incurs non-negligible time and space overhead in hardware deployment. In this study, we propose Nonuniform-to-Uniform Quantization (N2UQ), a method that can maintain the strong representation ability of nonuniform methods while being hardware-friendly and efficient as the uniform quantization for model inference. We achieve this through learning the flexible in-equidistant input thresholds to better fit the underlying distribution while quantizing these real-valued inputs into equidistant output levels. To train the quantized network with learnable input thresholds, we introduce a generalized straight-through estimator (G-STE) for intractable backward derivative calculation w.r.t. threshold parameters. Additionally, we consider entropy preserving regularization to further reduce information loss in weight quantization. Even under this adverse constraint of imposing uniformly quantized weights and activations, our N2UQ outperforms state-of-the-art nonuniform quantization methods by 0.5~1.7 on ImageNet, demonstrating the contribution of N2UQ design. Code and models are available at: https://github.com/liuzechun/Nonuniform-to-Uniform-Quantization.
Key facts. Cited by 78 papers in the map (data through 7 Oct 2026). Builds on 49 of its 58 references that are in the map. First posted on arXiv 29 Nov 2021.
Builds on
49 of 58 references.
References · 49
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,773
- 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,827
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,590
- 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,291
- 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,031
- 2017Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inferencecited by 1,548
- 2017ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devicescited by 1,194
37 more
- 2018ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Designcited by 1,132
- 2016XNOR-Net: ImageNet Classification Using Binary Convolutional Neural Networkscited by 829
- 2021A Survey of Quantization Methods for Efficient Neural Network Inferencecited by 798
- 2019Once-for-All: Train One Network and Specialize it for Efficient Deploymentcited by 778
- 2017Learning Efficient Convolutional Networks through Network Slimmingcited by 673
- 2018ProxylessNAS: Direct Neural Architecture Search on Target Task and Hardwarecited by 666
- 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
- 2016Weight Normalization: A Simple Reparameterization to Accelerate Training of Deep Neural Networkscited by 541
- 2018FBNet: Hardware-Aware Efficient ConvNet Design via Differentiable Neural Architecture Searchcited by 480
- 2018PACT: Parameterized Clipping Activation for Quantized Neural Networkscited by 456
- 2016Quantized Neural Networks: Training Neural Networks with Low Precision Weights and Activationscited by 413
- 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
- 2019MetaPruning: Meta Learning for Automatic Neural Network Channel Pruningcited by 226
- 2020ReActNet: Towards Precise Binary Neural Network with Generalized Activation Functionscited by 183
- 2018Learning to Quantize Deep Networks by Optimizing Quantization Intervals with Task Losscited by 157
- 2019Additive Powers-of-Two Quantization: An Efficient Non-uniform Discretization for Neural Networkscited by 143
- 2018Mixed Precision Quantization of ConvNets via Differentiable Neural Architecture Searchcited by 122
- 2016Binarized Neural Networks: Training Deep Neural Networks with Weights and Activations Constrained to +1 or -1cited by 102
- 2016Convolutional Neural Networks using Logarithmic Data Representationcited by 97
- 2019XNOR-Net++: Improved Binary Neural Networkscited by 94
- 2017Towards Effective Low-bitwidth Convolutional Neural Networkscited by 76
- 2019Global Sparse Momentum SGD for Pruning Very Deep Neural Networkscited by 70
- 2018MEAL: Multi-Model Ensemble via Adversarial Learningcited by 52
- 2020PROFIT: A Novel Training Method for sub-4-bit MobileNet Modelscited by 50
- 2018Bi-Real Net: Binarizing Deep Network Towards Real-Network Performancecited by 42
- 2021Learnable Companding Quantization for Accurate Low-bit Neural Networkscited by 40
- 2018Binary Ensemble Neural Network: More Bits per Network or More Networks per Bit?cited by 38
- 2019Training Quantized Neural Networks with a Full-precision Auxiliary Modulecited by 35
- 2020MeliusNet: Can Binary Neural Networks Achieve MobileNet-level Accuracy?cited by 32
- 2021Distance-aware Quantizationcited by 19
- 2020BiQGEMM: Matrix Multiplication with Lookup Table For Binary-Coding-based Quantized DNNscited by 17
- 2021Design and Operation of Shared Machine Learning Clusters on Campuscited by 12
- 2019An efficient floating point multiplier design for high speed applications using Karatsuba algorithm and Urdhva-Tiryagbhyam algorithmcited by 1
Built on by
Citing papers per year: 2022: 4; 2023: 20; 2024: 27; 2025: 21; 2026: 6.
78 papers.
Sort:
By area:
- 2023OmniQuant: Omnidirectionally Calibrated Quantization for Large Language Modelscited by 403
- 2023LLM-QAT: Data-Free Quantization Aware Training for Large Language Modelscited by 320
- 2022A Comprehensive Survey on Model Quantization for Deep Neural Networks in Image Classificationcited by 64
- 2023QLLM: Accurate and Efficient Low-Bitwidth Quantization for Large Language Modelscited by 63
- 2023Oscillation-free Quantization for Low-bit Vision Transformerscited by 55
- 2022BiT: Robustly Binarized Multi-distilled Transformercited by 51
- 2024MoH: Multi-Head Attention as Mixture-of-Head Attentioncited by 41
- 2024COAT: Compressing Optimizer states and Activation for Memory-Efficient FP8 Trainingcited by 32
- 2024SpikeLLM: Scaling up Spiking Neural Network to Large Language Models via Saliency-based Spikingcited by 29
- 2024ApiQ: Finetuning of 2-Bit Quantized Large Language Modelcited by 27
- 2023Q-DETR: An Efficient Low-Bit Quantized Detection Transformercited by 23
- 2023A Survey on Deep Learning Hardware Accelerators for Heterogeneous HPC Platformscited by 21
38 more
- 2024FIARSE: Model-Heterogeneous Federated Learning via Importance-Aware Submodel Extractioncited by 21
- 2024LeanQuant: Accurate and Scalable Large Language Model Quantization with Loss-error-aware Gridcited by 20
- 2024ELMGS: Enhancing memory and computation scaLability through coMpression for 3D Gaussian Splattingcited by 19
- 2023IRGen: Generative Modeling for Image Retrievalcited by 18
- 2024How Uniform Random Weights Induce Non-uniform Bias: Typical Interpolating Neural Networks Generalize with Narrow Teacherscited by 18
- 2025PMQ-VE: Progressive Multi-Frame Quantization for Video Enhancementcited by 17
- 2024A Comprehensive Survey of Compression Algorithms for Language Modelscited by 15
- 2024L4Q: Parameter Efficient Quantization-Aware Fine-Tuning on Large Language Modelscited by 15
- 2024Advancing Multimodal Large Language Models with Quantization-Aware Scale Learning for Efficient Adaptationcited by 15
- 2023Quantized Feature Distillation for Network Quantizationcited by 13
- 2025Low-bit Model Quantization for Deep Neural Networks: A Surveycited by 11
- 2023Stochastic Approximation Beyond Gradient for Signal Processing and Machine Learningcited by 10
- 2024PassionSR: Post-Training Quantization with Adaptive Scale in One-Step Diffusion based Image Super-Resolutioncited by 8
- 2023MetaMix: Meta-state Precision Searcher for Mixed-precision Activation Quantizationcited by 7
- 2023SEAM: Searching Transferable Mixed-Precision Quantization Policy through Large Margin Regularizationcited by 6
- 2023Reducing the Side-Effects of Oscillations in Training of Quantized YOLO Networkscited by 6
- 2024Custom Gradient Estimators are Straight-Through Estimators in Disguisecited by 6
- 2024SKIM: Any-bit Quantization Pushing The Limits of Post-Training Quantizationcited by 6
- 2024Improving Quantization-aware Training of Low-Precision Network via Block Replacement on Full-Precision Counterpartcited by 6
- 2025LowRA: Accurate and Efficient LoRA Fine-Tuning of LLMs under 2 Bitscited by 6
- 2025Neural Network Quantization for Microcontrollers: A Comprehensive Survey of Methods, Platforms, and Applicationscited by 6
- 2023GSB: Group Superposition Binarization for Vision Transformer with Limited Training Samplescited by 4
- 2023Low-Precision Mixed-Computation Models for Inference on Edgecited by 4
- 2024decoupleQ: Towards 2-bit Post-Training Uniform Quantization via decoupling Parameters into Integer and Floating Pointscited by 4
- 2025Q-MambaIR: Accurate Quantized Mamba for Efficient Image Restorationcited by 4
- 2026Switch Attention: Towards Dynamic and Fine-grained Hybrid Transformerscited by 4
- 2023Efficient and Effective Methods for Mixed Precision Neural Network Quantization for Faster, Energy-efficient Inferencecited by 3
- 2024CATP: Cross-Attention Token Pruning for Accuracy Preserved Multimodal Model Inferencecited by 3
- 2025Rotate, Clip, and Partition: Towards W2A4KV4 Quantization by Integrating Rotation and Learnable Non-uniform Quantizercited by 3
- 2025TruncQuant: Truncation-Ready Quantization for DNNs with Flexible Weight Bit Precisioncited by 3
- 2025MoQE: Improve Quantization Model performance via Mixture of Quantization Expertscited by 3
- 2022Exploiting the Partly Scratch-off Lottery Ticket for Quantization-Aware Trainingcited by 2
- 2023Efficient and Robust Quantization-aware Training via Adaptive Coreset Selectioncited by 2
- 2024HEQuant: Marrying Homomorphic Encryption and Quantization for Communication-Efficient Private Inferencecited by 2
- 2024SQUAT: Stateful Quantization-Aware Training in Recurrent Spiking Neural Networkscited by 2
- 2024Compensate Quantization Errors: Make Weights Hierarchical to Compensate Each Othercited by 2
- 2024Towards Accurate and Efficient Sub-8-Bit Integer Trainingcited by 2
- 2025Extending Straight-Through Estimation for Robust Neural Networks on Analog CIM Hardwarecited by 2
50 most cited of 78 All 78 (JSON)
- 2026Dynamic Thermal Gaussians: Multimodal 4D Gaussian Splatting21 Sep 2026
- 2026Decoupling Semantics and Fingerprints: A Universal Representation for AI-Generated Image Detection8 May 2026
- 2026Switch Attention: Towards Dynamic and Fine-grained Hybrid Transformers27 Mar 2026
- 2026Prune-then-Quantize or Quantize-then-Prune? Understanding the Impact of Compression Order in Joint Model Compression19 Mar 2026
- 2026Boosting Entropy with Bell Box Quantization2 Mar 2026
- 2026CQ-CiM: Hardware-Aware Embedding Shaping for Robust CiM-Based Retrieval23 Feb 2026
- 2025Large Speech Model Enabled Semantic Communication4 Dec 2025
- 2025Q2: Quantization-Aware Gradient Balancing and Attention Alignment for Low-Bit Quantization8 Nov 2025
- 2025ELUTQ: Optimizing Quantization Accuracy under LUT-Based Computation for Edge LLMs22 Oct 2025
- 2025MEC-Quant: Maximum Entropy Coding for Extremely Low Bit Quantization-Aware Training19 Sep 2025
- 2025Exploiting Information Redundancy in Attention Maps for Extreme Quantization of Vision Transformers22 Aug 2025
- 2025Neural Network Quantization for Microcontrollers: A Comprehensive Survey of Methods, Platforms, and Applications20 Aug 2025
38 more
- 2025Extending Straight-Through Estimation for Robust Neural Networks on Analog CIM Hardware16 Aug 2025
- 2025MoQE: Improve Quantization Model performance via Mixture of Quantization Experts9 Aug 2025
- 2025IML-Spikeformer: Input-aware Multi-Level Spiking Transformer for Speech Processing10 Jul 2025
- 2025TruncQuant: Truncation-Ready Quantization for DNNs with Flexible Weight Bit Precision13 Jun 2025
- 2025Optimizing Learned Image Compression on Scalar and Entropy-Constraint Quantization10 Jun 2025
- 2025QuantFace: Efficient Quantization for Face Restoration1 Jun 2025
- 2025It Takes a Good Model to Train a Good Model: Generalized Gaussian Priors for Optimized LLMs31 May 2025
- 2025PMQ-VE: Progressive Multi-Frame Quantization for Video Enhancement18 May 2025
- 2025Low-bit Model Quantization for Deep Neural Networks: A Survey8 May 2025
- 2025BackSlash: Rate Constrained Optimized Training of Large Language Models23 Apr 2025
- 2025AccLLM: Accelerating Long-Context LLM Inference Via Algorithm-Hardware Co-Design7 Apr 2025
- 2025Q-MambaIR: Accurate Quantized Mamba for Efficient Image Restoration27 Mar 2025
- 2025QArtSR: Quantization via Reverse-Module and Timestep-Retraining in One-Step Diffusion based Image Super-Resolution7 Mar 2025
- 2025Rotate, Clip, and Partition: Towards W2A4KV4 Quantization by Integrating Rotation and Learnable Non-uniform Quantizer17 Feb 2025
- 2025LowRA: Accurate and Efficient LoRA Fine-Tuning of LLMs under 2 Bits12 Feb 2025
- 2024Improving Quantization-aware Training of Low-Precision Network via Block Replacement on Full-Precision Counterpart20 Dec 2024
- 2024Efficient Quantization-Aware Training on Segment Anything Model in Medical Images and Its Deployment15 Dec 2024
- 2024A Data-Free Analytical Quantization Scheme for Deep Learning Models10 Dec 2024
- 2024SKIM: Any-bit Quantization Pushing The Limits of Post-Training Quantization5 Dec 2024
- 2024PassionSR: Post-Training Quantization with Adaptive Scale in One-Step Diffusion based Image Super-Resolution26 Nov 2024
- 2024Towards Accurate and Efficient Sub-8-Bit Integer Training17 Nov 2024
- 2024ELMGS: Enhancing memory and computation scaLability through coMpression for 3D Gaussian Splatting30 Oct 2024
- 2024COAT: Compressing Optimizer states and Activation for Memory-Efficient FP8 Training25 Oct 2024
- 2024MoH: Multi-Head Attention as Mixture-of-Head Attention15 Oct 2024
- 2024Advancing Multimodal Large Language Models with Quantization-Aware Scale Learning for Efficient Adaptation7 Aug 2024
- 2024FIARSE: Model-Heterogeneous Federated Learning via Importance-Aware Submodel Extraction28 Jul 2024
- 2024Compensate Quantization Errors+: Quantized Models Are Inquisitive Learners22 Jul 2024
- 2024LeanQuant: Accurate and Scalable Large Language Model Quantization with Loss-error-aware Grid14 Jul 2024
- 2024SpikeLLM: Scaling up Spiking Neural Network to Large Language Models via Saliency-based Spiking5 Jul 2024
- 2024Compensate Quantization Errors: Make Weights Hierarchical to Compensate Each Other24 Jun 2024
- 2024LCQ: Low-Rank Codebook based Quantization for Large Language Models31 May 2024
- 2024Accurate Quantization for Gait Representation Learning22 May 2024
- 2024Custom Gradient Estimators are Straight-Through Estimators in Disguise8 May 2024
- 2024Torch2Chip: An End-to-end Customizable Deep Neural Network Compression and Deployment Toolkit for Prototype Hardware Accelerator Design2 May 2024
- 2024decoupleQ: Towards 2-bit Post-Training Uniform Quantization via decoupling Parameters into Integer and Floating Points19 Apr 2024
- 2024SQUAT: Stateful Quantization-Aware Training in Recurrent Spiking Neural Networks15 Apr 2024
- 2024CATP: Cross-Attention Token Pruning for Accuracy Preserved Multimodal Model Inference2 Apr 2024
- 2024How Uniform Random Weights Induce Non-uniform Bias: Typical Interpolating Neural Networks Generalize with Narrow Teachers9 Feb 2024
50 newest of 78 All 78 (JSON)
- 2024PassionSR: Post-Training Quantization with Adaptive Scale in One-Step Diffusion based Image Super-Resolutioncited by 8
- 2025QArtSR: Quantization via Reverse-Module and Timestep-Retraining in One-Step Diffusion based Image Super-Resolutioncited by 0
- 2025QuantFace: Efficient Quantization for Face Restorationcited by 0
- 2026Decoupling Semantics and Fingerprints: A Universal Representation for AI-Generated Image Detectioncited by 0
- 2026Decoupling Semantics and Fingerprints: A Universal Representation for AI-Generated Image Detection8 May 2026
- 2025QuantFace: Efficient Quantization for Face Restoration1 Jun 2025
- 2025QArtSR: Quantization via Reverse-Module and Timestep-Retraining in One-Step Diffusion based Image Super-Resolution7 Mar 2025
- 2024PassionSR: Post-Training Quantization with Adaptive Scale in One-Step Diffusion based Image Super-Resolution26 Nov 2024
- 2023OmniQuant: Omnidirectionally Calibrated Quantization for Large Language Modelscited by 403
- 2023LLM-QAT: Data-Free Quantization Aware Training for Large Language Modelscited by 320
- 2022A Comprehensive Survey on Model Quantization for Deep Neural Networks in Image Classificationcited by 64
- 2023QLLM: Accurate and Efficient Low-Bitwidth Quantization for Large Language Modelscited by 63
- 2023Oscillation-free Quantization for Low-bit Vision Transformerscited by 55
- 2022BiT: Robustly Binarized Multi-distilled Transformercited by 51
- 2024COAT: Compressing Optimizer states and Activation for Memory-Efficient FP8 Trainingcited by 32
- 2024SpikeLLM: Scaling up Spiking Neural Network to Large Language Models via Saliency-based Spikingcited by 29
- 2024ApiQ: Finetuning of 2-Bit Quantized Large Language Modelcited by 27
- 2023Q-DETR: An Efficient Low-Bit Quantized Detection Transformercited by 23
- 2023A Survey on Deep Learning Hardware Accelerators for Heterogeneous HPC Platformscited by 21
- 2024FIARSE: Model-Heterogeneous Federated Learning via Importance-Aware Submodel Extractioncited by 21
12 most cited of 66 All 66 (JSON)
- 2026Dynamic Thermal Gaussians: Multimodal 4D Gaussian Splatting21 Sep 2026
- 2026Switch Attention: Towards Dynamic and Fine-grained Hybrid Transformers27 Mar 2026
- 2026Prune-then-Quantize or Quantize-then-Prune? Understanding the Impact of Compression Order in Joint Model Compression19 Mar 2026
- 2026Boosting Entropy with Bell Box Quantization2 Mar 2026
- 2025Large Speech Model Enabled Semantic Communication4 Dec 2025
- 2025Q2: Quantization-Aware Gradient Balancing and Attention Alignment for Low-Bit Quantization8 Nov 2025
- 2025ELUTQ: Optimizing Quantization Accuracy under LUT-Based Computation for Edge LLMs22 Oct 2025
- 2025Exploiting Information Redundancy in Attention Maps for Extreme Quantization of Vision Transformers22 Aug 2025
- 2025Neural Network Quantization for Microcontrollers: A Comprehensive Survey of Methods, Platforms, and Applications20 Aug 2025
- 2025Extending Straight-Through Estimation for Robust Neural Networks on Analog CIM Hardware16 Aug 2025
- 2025IML-Spikeformer: Input-aware Multi-Level Spiking Transformer for Speech Processing10 Jul 2025
- 2025TruncQuant: Truncation-Ready Quantization for DNNs with Flexible Weight Bit Precision13 Jun 2025
12 newest of 66 All 66 (JSON)