Robust Training of Neural Networks at Arbitrary Precision and Sparsity
Not on the map. Why?
Abstract (arXiv)
The discontinuous operations inherent in quantization and sparsification introduce a long-standing obstacle to backpropagation, particularly in ultra-low precision and sparse regimes. While the community has long viewed quantization as unfriendly to gradient descent due to its lack of smoothness, we pinpoint-for the first time-that the key issue is the absence of a proper gradient path that allows training to learn robustness to quantization noise. The standard Straight-Through Estimator (STE) exacerbates this with its well-understood mismatch: a quantization-aware forward pass but oblivious backward pass, leading to unmanaged error and instability. We solve this by explicitly modeling quantization as additive noise, making the full forward-backward path well-defined without heuristic gradient estimation. As one natural solution, we introduce a denoising dequantization transform derived from a principled ridge regression objective, creating an explicit, corrective gradient path that makes learning robust to the noise STE bypasses. We extend this to sparsification by treating it as a special form of quantization that zeros out small values. Our unified framework trains models at arbitrary precisions and sparsity levels with off-the-shelf recipes, enabling stable A1W1 and sub-1-bit networks where others falter. It yields state-of-the-art results, mapping efficiency frontiers for modern LLMs and providing a theoretically grounded path to hyper-efficient neural networks.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 36 of its 56 references that are in the map. First posted on arXiv 14 Sep 2024.
Builds on
36 of 56 references.
References · 36
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,326
- 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,329
- 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,079
- 2017Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inferencecited by 1,593
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,417
- 2019Event-based Vision: A Surveycited by 1,016
- 2016XNOR-Net: ImageNet Classification Using Binary Convolutional Neural Networkscited by 849
- 2021A Survey of Quantization Methods for Efficient Neural Network Inferencecited by 815
- 2016End-to-end Optimized Image Compressioncited by 696
- 2023Instruction Tuning with GPT-4cited by 686
- 2024QuaRot: Outlier-Free 4-Bit Inference in Rotated LLMscited by 634
- 2019Learned Step Size Quantizationcited by 628
24 more
- 2024SpinQuant: LLM quantization with learned rotationscited by 471
- 2018HAQ: Hardware-Aware Automated Quantization with Mixed Precisioncited by 443
- 2016Quantized Neural Networks: Training Neural Networks with Low Precision Weights and Activationscited by 426
- 2024QuIP#: Even Better LLM Quantization with Hadamard Incoherence and Lattice Codebookscited by 378
- 2023SpQR: A Sparse-Quantized Representation for Near-Lossless LLM Weight Compressioncited by 367
- 2018Deep Learning in Spiking Neural Networkscited by 263
- 2018Bi-Real Net: Enhancing the Performance of 1-bit CNNs With Improved Representational Capability and Advanced Training Algorithmcited by 235
- 2019Understanding Straight-Through Estimator in Training Activation Quantized Neural Netscited by 231
- 2016Training Deep Spiking Neural Networks using Backpropagationcited by 205
- 2023BitNet: Scaling 1-bit Transformers for Large Language Modelscited by 189
- 2017Training Quantized Nets: A Deeper Understandingcited by 75
- 2023Stable and low-precision training for large-scale vision-language modelscited by 75
- 2021VS-Quant: Per-vector Scaled Quantization for Accurate Low-Precision Neural Network Inferencecited by 64
- 2023Training Transformers with 4-bit Integerscited by 60
- 2021PokeBNN: A Binary Pursuit of Lightweight Accuracycited by 49
- 2021How Do Adam and Training Strategies Help BNNs Optimization?cited by 46
- 2021Differentiable Model Compression via Pseudo Quantization Noisecited by 45
- 2019QKD: Quantization-aware Knowledge Distillationcited by 44
- 2018Discovering Low-Precision Networks Close to Full-Precision Networks for Efficient Embedded Inferencecited by 33
- 2024STBLLM: Breaking the 1-Bit Barrier with Structured Binary LLMscited by 31
- 2021Pareto-Optimal Quantized ResNet Is Mostly 4-bitcited by 26
- 2021Sharpness-aware Quantization for Deep Neural Networkscited by 25
- 2023Binarized Neural Machine Translationcited by 13
- 2024PikeLPN: Mitigating Overlooked Inefficiencies of Low-Precision Neural Networkscited by 4
Built on by
Citing papers per year: 2025: 1; 2026: 2.
3 papers.
- 2025HybridGS: High-Efficiency Gaussian Splatting Data Compression using Dual-Channel Sparse Representation and Point Cloud Encodercited by 10
- 2026QUASAR: Lowering the Loss Floor of Quantization-Aware Training with Loss-Aware Reconstructioncited by 1
- 2026SandwichQuant: Which Parameters Matter Before and After Quantization?cited by 0
- 2026SandwichQuant: Which Parameters Matter Before and After Quantization?25 Aug 2026
- 2026QUASAR: Lowering the Loss Floor of Quantization-Aware Training with Loss-Aware Reconstruction14 Aug 2026
- 2025HybridGS: High-Efficiency Gaussian Splatting Data Compression using Dual-Channel Sparse Representation and Point Cloud Encoder3 May 2025
Similar, not cited
Close in wording, no citation either way
- 2025High-Dimensional Learning Dynamics of Quantized Models with Straight-Through Estimator
- 2022Error-aware Quantization through Noise Tempering
- 2026Activation Denoising: A Robustness View on Parallel vs Sequential LLM Quantization
- 2025Beyond Discreteness: Sample Complexity Analysis of Straight-Through Estimator for 1-bit Quantization
- 2026StableQAT: Stable Quantization-Aware Training at Ultra-Low Bitwidths
- 2022Neural Networks with Quantization Constraints
- 2021Post-Training Sparsity-Aware Quantization
- 2024Effective Interplay between Sparsity and Quantization: From Theory to Practice