arXiv · 2610.11730
Spectral Weight Decay: Inducing Low-Rank Structure in Neural Network Weights
Abstract
Standard weight decay treats each weight matrix as a vector and ignores its spectral structure. We introduce spectral weight decay, a post-step decoupled nuclear-norm update that applies additive rather than multiplicative spectral shrinkage. We connect the update to approximate proximal descent and show that its sensitivity to update order can exceed that of conventional $\ell_2$ weight decay near rank deficiency. Across LLaMA models with $124$M to $500$M parameters, spectral weight decay lowers effective rank and improves SVD-LLM compression at matched validation loss. At $500$M and a $4\%$ distortion budget, it reaches $1.89\times$ compression and $1.18\times$ GPU inference speedup, compared with $1.14\times$ and $1.01\times$ after standard weight decay. Under fixed-horizon training with $60\%$ label noise, it also improves final mean clean-test accuracy over matched $\ell_2$ regularization by up to $17.8$ points on MNIST and $4.6$ points across four BERT-base tasks. Code is available at https://github.com/brain-lab-research/SpectralWD.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Dmitrii Andriianov, Andrey Veprikov, Aleksandr Beznosikov. 2026-10-08. Spectral Weight Decay: Inducing Low-Rank Structure in Neural Network Weights. https://arxiv.org/abs/2610.11730
Cite the original work for its findings. Save a collection to share your selection of sources.