Estimator Meets Equilibrium Perspective: A Rectified Straight Through Estimator for Binary Neural Networks Training
Not on the map. Why?
Abstract (arXiv)
Binarization of neural networks is a dominant paradigm in neural networks compression. The pioneering work BinaryConnect uses Straight Through Estimator (STE) to mimic the gradients of the sign function, but it also causes the crucial inconsistency problem. Most of the previous methods design different estimators instead of STE to mitigate it. However, they ignore the fact that when reducing the estimating error, the gradient stability will decrease concomitantly. These highly divergent gradients will harm the model training and increase the risk of gradient vanishing and gradient exploding. To fully take the gradient stability into consideration, we present a new perspective to the BNNs training, regarding it as the equilibrium between the estimating error and the gradient stability. In this view, we firstly design two indicators to quantitatively demonstrate the equilibrium phenomenon. In addition, in order to balance the estimating error and the gradient stability well, we revise the original straight through estimator and propose a power function based estimator, Rectified Straight Through Estimator (ReSTE for short). Comparing to other estimators, ReSTE is rational and capable of flexibly balancing the estimating error with the gradient stability. Extensive experiments on CIFAR-10 and ImageNet datasets show that ReSTE has excellent performance and surpasses the state-of-the-art methods without any auxiliary modules or losses.
Key facts. Cited by 19 papers in the map (data through 14 Oct 2026). Builds on 50 of its 60 references that are in the map. First posted on arXiv 13 Aug 2023.
Builds on
50 of 60 references.
References · 50
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
- 2014Fully Convolutional Networks for Semantic Segmentationcited by 4,859
- 2015You Only Look Once: Unified, Real-Time Object Detectioncited by 4,852
- 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,845
- 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,329
- 2013Rich feature hierarchies for accurate object detection and semantic segmentationcited by 2,162
- 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,079
- 2017ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devicescited by 1,200
38 more
- 2016SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model sizecited by 1,060
- 2019Relational Knowledge Distillationcited by 889
- 2016XNOR-Net: ImageNet Classification Using Binary Convolutional Neural Networkscited by 849
- 2019Contrastive Representation Distillationcited by 748
- 2015BinaryConnect: Training Deep Neural Networks with binary weights during propagationscited by 592
- 2016DoReFa-Net: Training Low Bitwidth Convolutional Neural Networks with Low Bitwidth Gradientscited by 569
- 2017ThiNet: A Filter Level Pruning Method for Deep Neural Network Compressioncited by 442
- 2016Quantized Neural Networks: Training Neural Networks with Low Precision Weights and Activationscited by 426
- 2022Decoupled Knowledge Distillationcited by 388
- 2018LQ-Nets: Learned Quantization for Highly Accurate and Compact Deep Neural Networkscited by 274
- 2019Differentiable Soft Quantization: Bridging Full-Precision and Low-Bit Neural Networkscited by 238
- 2018Bi-Real Net: Enhancing the Performance of 1-bit CNNs With Improved Representational Capability and Advanced Training Algorithmcited by 235
- 2020ReActNet: Towards Precise Binary Neural Network with Generalized Activation Functionscited by 184
- 2018Discrimination-aware Channel Pruning for Deep Neural Networkscited by 171
- 2019Forward and Backward Information Retention for Accurate Binary Neural Networkscited by 169
- 2017Towards Accurate Binary Convolutional Neural Networkcited by 149
- 2017Deep Learning with Low Precision by Half-wave Gaussian Quantizationcited by 117
- 2019XNOR-Net++: Improved Binary Neural Networkscited by 97
- 2020ResRep: Lossless CNN Pruning via Decoupling Remembering and Forgettingcited by 85
- 2021Vision Transformer Pruningcited by 79
- 2020Rotated Binary Neural Networkcited by 77
- 2017Extremely Low Bit Neural Network: Squeeze the Last Bit Out with ADMMcited by 61
- 2019Regularizing Activation Distribution for Training Binarized Deep Networkscited by 61
- 2019Latent Weights Do Not Exist: Rethinking Binarized Neural Network Optimizationcited by 60
- 2018ProxQuant: Quantized Neural Networks via Proximal Operatorscited by 59
- 2016Loss-aware Binarization of Deep Networkscited by 56
- 2017Introduction to Tensor Decompositions and their Applications in Machine Learningcited by 46
- 2021Learning Frequency Domain Approximation for Binary Neural Networkscited by 38
- 2022AdaBin: Improving Binary Neural Networks with Adaptive Binary Setscited by 38
- 2016Ternary Neural Networks for Resource-Efficient AI Applicationscited by 33
- 2018Projection Convolutional Neural Networks for 1-bit CNNs via Discrete Back Propagationcited by 29
- 2018From Hashing to CNNs: Training BinaryWeight Networks via Hashingcited by 23
- 2017Learning Accurate Low-Bit Deep Neural Networks with Stochastic Quantizationcited by 21
- 2019Mirror Descent View for Neural Network Quantizationcited by 16
- 2019Bayesian Optimized 1-Bit CNNscited by 14
- 2022Lipschitz Continuity Retained Binary Neural Networkcited by 14
- 2022Recurrent Bilinear Optimization for Binary Neural Networkscited by 12
- 2015Training Binary Multilayer Neural Networks for Image Classification using Expectation Backpropagationcited by 6
Built on by
Citing papers per year: 2023: 1; 2024: 7; 2025: 10; 2026: 1.
19 papers.
Sort:
By area:
- 2025Decoupled Distillation to Erase: A General Unlearning Method for Any Class-centric Taskscited by 28
- 2024Binarized Low-light Raw Video Enhancementcited by 17
- 2024Solving Token Gradient Conflict in Mixture-of-Experts for Large Vision-Language Modelcited by 11
- 2024BiDM: Pushing the Limit of Quantization for Diffusion Modelscited by 11
- 2025Panorama Generation From NFoV Image Done Rightcited by 11
- 2024BiPer: Binary Neural Networks using a Periodic Functioncited by 9
- 2023On the Opportunities of Green Computing: A Surveycited by 7
- 2025BiMaCoSR: Binary One-Step Diffusion Model Leveraging Flexible Matrix Compression for Real Super-Resolutioncited by 5
- 2025CAGE: Curvature-Aware Gradient Estimation For Accurate Quantization-Aware Trainingcited by 4
- 20241-Bit FQT: Pushing the Limit of Fully Quantized Training to 1-bitcited by 3
- 2024OvSW: Overcoming Silent Weights for Accurate Binary Neural Networkscited by 2
- 2025GDNSQ: Gradual Differentiable Noise Scale Quantization for Low-bit Neural Networkscited by 2
7 more
- 2024Fast and Slow Gradient Approximation for Binary Neural Network Optimizationcited by 0
- 2025Hyperbolic Binary Neural Networkcited by 0
- 2025Expanding-and-Shrinking Binary Neural Networkscited by 0
- 2025NuWa: Deriving Lightweight Class-Specific Vision Transformers for Edge Devicescited by 0
- 2025End-to-end fully-binarized network design: from Generic Learned Thermometer to Block Pruningcited by 0
- 2025AMAQ: Adaptive Mixed-bit Activation Quantization for Collaborative Parameter Efficient Fine-tuningcited by 0
- 2026Training single-electron and single-photon stochastic physical neural networkscited by 0
- 2026Training single-electron and single-photon stochastic physical neural networks12 Apr 2026
- 2025CAGE: Curvature-Aware Gradient Estimation For Accurate Quantization-Aware Training21 Oct 2025
- 2025AMAQ: Adaptive Mixed-bit Activation Quantization for Collaborative Parameter Efficient Fine-tuning7 Oct 2025
- 2025GDNSQ: Gradual Differentiable Noise Scale Quantization for Low-bit Neural Networks19 Aug 2025
- 2025End-to-end fully-binarized network design: from Generic Learned Thermometer to Block Pruning5 May 2025
- 2025NuWa: Deriving Lightweight Class-Specific Vision Transformers for Edge Devices4 Apr 2025
- 2025Decoupled Distillation to Erase: A General Unlearning Method for Any Class-centric Tasks31 Mar 2025
- 2025Expanding-and-Shrinking Binary Neural Networks31 Mar 2025
- 2025Panorama Generation From NFoV Image Done Right24 Mar 2025
- 2025BiMaCoSR: Binary One-Step Diffusion Model Leveraging Flexible Matrix Compression for Real Super-Resolution1 Feb 2025
- 2025Hyperbolic Binary Neural Network7 Jan 2025
- 2024Fast and Slow Gradient Approximation for Binary Neural Network Optimization16 Dec 2024
7 more
- 2024BiDM: Pushing the Limit of Quantization for Diffusion Models8 Dec 2024
- 20241-Bit FQT: Pushing the Limit of Fully Quantized Training to 1-bit26 Aug 2024
- 2024OvSW: Overcoming Silent Weights for Accurate Binary Neural Networks7 Jul 2024
- 2024Solving Token Gradient Conflict in Mixture-of-Experts for Large Vision-Language Model28 Jun 2024
- 2024BiPer: Binary Neural Networks using a Periodic Function1 Apr 2024
- 2024Binarized Low-light Raw Video Enhancement29 Mar 2024
- 2023On the Opportunities of Green Computing: A Survey1 Nov 2023
- 2024Binarized Low-light Raw Video Enhancementcited by 17
- 2025Panorama Generation From NFoV Image Done Rightcited by 11
- 2024BiPer: Binary Neural Networks using a Periodic Functioncited by 9
- 2023On the Opportunities of Green Computing: A Surveycited by 7
- 2025CAGE: Curvature-Aware Gradient Estimation For Accurate Quantization-Aware Trainingcited by 4
- 20241-Bit FQT: Pushing the Limit of Fully Quantized Training to 1-bitcited by 3
- 2024OvSW: Overcoming Silent Weights for Accurate Binary Neural Networkscited by 2
- 2025GDNSQ: Gradual Differentiable Noise Scale Quantization for Low-bit Neural Networkscited by 2
- 2024Fast and Slow Gradient Approximation for Binary Neural Network Optimizationcited by 0
- 2025Hyperbolic Binary Neural Networkcited by 0
- 2025Expanding-and-Shrinking Binary Neural Networkscited by 0
- 2025NuWa: Deriving Lightweight Class-Specific Vision Transformers for Edge Devicescited by 0
12 most cited of 15 All 15 (JSON)
- 2026Training single-electron and single-photon stochastic physical neural networks12 Apr 2026
- 2025CAGE: Curvature-Aware Gradient Estimation For Accurate Quantization-Aware Training21 Oct 2025
- 2025AMAQ: Adaptive Mixed-bit Activation Quantization for Collaborative Parameter Efficient Fine-tuning7 Oct 2025
- 2025GDNSQ: Gradual Differentiable Noise Scale Quantization for Low-bit Neural Networks19 Aug 2025
- 2025End-to-end fully-binarized network design: from Generic Learned Thermometer to Block Pruning5 May 2025
- 2025NuWa: Deriving Lightweight Class-Specific Vision Transformers for Edge Devices4 Apr 2025
- 2025Expanding-and-Shrinking Binary Neural Networks31 Mar 2025
- 2025Panorama Generation From NFoV Image Done Right24 Mar 2025
- 2025Hyperbolic Binary Neural Network7 Jan 2025
- 2024Fast and Slow Gradient Approximation for Binary Neural Network Optimization16 Dec 2024
- 20241-Bit FQT: Pushing the Limit of Fully Quantized Training to 1-bit26 Aug 2024
- 2024OvSW: Overcoming Silent Weights for Accurate Binary Neural Networks7 Jul 2024
12 newest of 15 All 15 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2021AdaSTE: An Adaptive Straight-Through Estimator to Train Binary Neural Networks
- 2026Stability and Generalization of Straight-Through Estimators for Training Two-Layer Quantized Neural Networks
- 2021A Bop and Beyond: A Second Order Optimizer for Binarized Neural Networks
- 2026SURGE: Surrogate Gradient Adaptation in Binary Neural Networks
- 2024Understanding Neural Network Binarization with Forward and Backward Proximal Quantizers
- 2023Injecting Logical Constraints into Neural Networks via Straight-Through Estimators
- 2022Binarizing by Classification: Is soft function really necessary?
- 2025Beyond Discreteness: Sample Complexity Analysis of Straight-Through Estimator for 1-bit Quantization