Gradient Clipping Beyond Vector Norms: A Spectral Approach for Matrix-Valued Parameters
Not on the map. Why?
Abstract (arXiv)
Gradient clipping is a standard safeguard for training neural networks under noisy, heavy-tailed stochastic gradients; yet, most clipping rules treat all parameters as vectors and ignore the matrix structure of modern architectures. We show empirically that data outliers often amplify only a small number of leading singular values in layer-wise gradient matrices, while the rest of the spectrum remains largely unchanged. Motivated by this phenomenon, we propose spectral clipping, which stabilizes training by clamping singular values that exceed a threshold while preserving the singular directions. This framework generalizes classical gradient norm clipping and can be easily integrated into existing optimizers. We provide a convergence analysis for non-convex optimization with spectrally clipped SGD, yielding the optimal Oleft(K^frac2 - 2α3α - 2right) rate for heavy-tailed noise. To minimize hyperparameter tuning, we introduce layer-wise adaptive thresholds based on moving averages or sliding-window quantiles of the top singular values. Finally, we develop efficient implementations that clip only the top r singular values via randomized truncated SVD, avoiding full decompositions for large layers. We demonstrate competitive performance across synthetic heavy-tailed settings and neural network training tasks.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 41 of its 68 references that are in the map. First posted on arXiv 12 May 2026.
Builds on
41 of 68 references.
References · 41
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2023GPT-4 Technical Reportcited by 18,132
- 2014Sequence to Sequence Learning with Neural Networkscited by 2,974
- 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,104
- 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 716
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
- 2021High-Performance Large-Scale Image Recognition Without Normalizationcited by 359
- 2017SARAH: A Novel Method for Machine Learning Problems Using Stochastic Recursive Gradientcited by 218
- 2019A Tail-Index Analysis of Stochastic Gradient Noise in Deep Neural Networkscited by 213
29 more
- 2024SOAP: Improving and Stabilizing Shampoo using Adamcited by 179
- 2025Training Deep Learning Models with Norm-Constrained LMOscited by 177
- 2023Prodigy: An Expeditiously Adaptive Parameter-Free Learnercited by 129
- 2020An Improved Analysis of Stochastic Gradient Descent with Momentumcited by 117
- 2020Stochastic Optimization with Heavy-Tailed Noise via Accelerated Gradient Clippingcited by 104
- 2025The Polar Express: Optimal Matrix Sign Methods and Their Application to the Muon Algorithmcited by 104
- 2023Linear attention is (maybe) all you need (to understand transformer optimization)cited by 97
- 2020Improved Analysis of Clipping Algorithms for Non-convex Optimizationcited by 88
- 2019A Unified Theory of SGD: Variance Reduction, Sampling, Quantization and Coordinate Descentcited by 84
- 2023Revisiting Gradient Clipping: Stochastic bias and tight convergence guaranteescited by 83
- 2025Practical Efficiency of Muon for Pretrainingcited by 74
- 2025ASGO: Adaptive Structured Gradient Optimizationcited by 55
- 2024From Gradient Clipping to Normalization for Heavy Tailed SGDcited by 48
- 2023Revisiting the Noise Model of Stochastic Gradient Descentcited by 35
- 2016A Multi-Batch L-BFGS Method for Machine Learningcited by 30
- 2021On Proximal Policy Optimization's Heavy-tailed Gradientscited by 30
- 2025On the Convergence of Muon and Beyondcited by 28
- 2023MoMo: Momentum Models for Adaptive Learning Ratescited by 24
- 2024Clipping Improves Adam-Norm and AdaGrad-Norm when the Noise Is Heavy-Tailedcited by 18
- 2025Generalized Gradient Norm Clipping & Non-Euclidean (L0,L1)-Smoothnesscited by 17
- 2021AI-SARAH: Adaptive and Implicit Stochastic Recursive Gradient Methodscited by 15
- 2022SP2: A Second Order Stochastic Polyak Methodcited by 13
- 2025GaLore 2: Large-Scale LLM Pre-Training by Gradient Low-Rank Projectioncited by 13
- 2013Mini-Batch Primal and Dual Methods for SVMscited by 12
- 2020Understanding Decoupled and Early Weight Decaycited by 12
- 2015The Singular Value Decomposition, Applications and Beyondcited by 10
- 2024Remove that Square Root: A New Efficient Scale-Invariant Version of AdaGradcited by 8
- 2026Muon with Nesterov Momentum: Heavy-Tailed Noise and (Randomized) Inexact Polar Decompositioncited by 8
- 2020Partial Trace Regression and Low-Rank Kraus Decompositioncited by 5
Built on by
Citing papers per year: 2026: 2.
2 papers.
- 2026Musec: MomentUm SpEctral Clipping for Stable Muon-type Trainingcited by 1
- 2026Spectrally Targeted Muoncited by 0
- 2026Spectrally Targeted Muon7 Oct 2026
- 2026Musec: MomentUm SpEctral Clipping for Stable Muon-type Training10 Sep 2026