mlsanityData to 14 Oct 2026

GradientStabilizer:Fix the Norm, Not the Gradient

arXiv 2502.17055 ↗First posted 24 Feb 2025cs.LG cs.AIcited by 6

Not on the map. Why?

Abstract (arXiv)

Training instability in modern deep learning systems is frequently triggered by rare but extreme gradient-norm spikes, which can induce oversized parameter updates, corrupt optimizer state, and lead to slow recovery or divergence. Widely used safeguards such as gradient clipping mitigate these failures but require threshold tuning and indiscriminately truncate large updates. We propose GradientStabilizer, a lightweight, drop-in gradient transform that preserves the instantaneous gradient direction while replacing the update magnitude with a statistically stabilized estimate derived from running gradient-norm statistics. We prove that the resulting stabilized magnitude is uniformly bounded on spike steps, independent of the spike size, and show how this boundedness controls optimizer state evolution in adaptive methods. Across LLM pre-training (FP16), quantization-aware pre-training (FP4), ImageNet classification, reinforcement learning, and time-series forecasting, GradientStabilizer consistently improves training stability, widens stable learning-rate regions, and reduces divergence relative to clipping-based baselines, even substantially reducing Adam's sensitivity to weight-decay strength. Code will be released soon.

Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 26 of its 44 references that are in the map. First posted on arXiv 24 Feb 2025.

Builds on

26 of 44 references.

References · 26

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2017Proximal Policy Optimization Algorithmscited by 15,215
  4. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
  5. 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,326
  6. 2022A ConvNet for the 2020scited by 3,579
  7. 2022A Time Series is Worth 64 Words: Long-term Forecasting with Transformerscited by 2,039
  8. 2020On Layer Normalization in the Transformer Architecturecited by 922
  9. 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 905
  10. 2022GLM-130B: An Open Bilingual Pre-trained Modelcited by 843
  11. 2019On the Convergence of Adam and Beyondcited by 841
  12. 2021CogView: Mastering Text-to-Image Generation via Transformerscited by 694
14 more
  1. 2023Symbolic Discovery of Optimization Algorithmscited by 434
  2. 20218-bit Optimizers via Block-wise Quantizationcited by 403
  3. 2021High-Performance Large-Scale Image Recognition Without Normalizationcited by 360
  4. 2019Transformers without Tears: Improving the Normalization of Self-Attentioncited by 153
  5. 2024Adam-mini: Use Fewer Learning Rates To Gain Morecited by 137
  6. 2023Stable and low-precision training for large-scale vision-language modelscited by 75
  7. 2023Spike No More: Stabilizing the Pre-training of Large Language Modelscited by 52
  8. 2025SPAM: Spike-Aware Adam with Momentum Reset for Stable LLM Trainingcited by 28
  9. 2025QuEST: Stable Training of LLMs with 1-Bit Weights and Activationscited by 22
  10. 2023Improving Training Stability for Multitask Ranking Models in Recommender Systemscited by 13
  11. 2019Ctrl-Z: Recovering from Instability in Reinforcement Learningcited by 6
  12. 2024Towards Understanding the Impact of Data Bugs on Deep Learning Models in Software Engineeringcited by 6
  13. 2024Navigating Data Corruption in Machine Learning: Balancing Quality, Quantity, and Imputation Strategiescited by 4
  14. 2024Outlier-Robust Training of Machine Learning Modelscited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 4; 2026: 2.

6 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Greedy Alignment Principle for Optimizer Selection
  2. 2026Singularity-aware Optimization via Randomized Geometric Probing: Towards Stable Non-smooth Optimization
  3. 2024Non-asymptotic Analysis of Biased Adaptive Stochastic Approximation
  4. 2026Gradient Under Microscope: Benchmarking Resource Utilization of Memory-Efficient Gradient Computation Methods
  5. 2025HVAdam: A Full-Dimension Adaptive Optimizer
  6. 2025Stable Gradients for Stable Learning at Scale in Deep Reinforcement Learning
  7. 2026Gefen: Optimized Stochastic Optimizer
  8. 2025CAGE: Curvature-Aware Gradient Estimation For Accurate Quantization-Aware Training