Spectra: Rethinking Optimizers for LLMs Under Spectral Anisotropy
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
Gradient signals in LLM training are highly anisotropic: recurrent linguistic structure concentrates energy into a small set of dominant spectral directions, while context specific information resides in a long tail. We show that this spike tail separation persists throughout training, with the spike occupying only about 1.5% of directions yet dominating optimizer statistics. This dominance suppresses tail learning by contracting tail updates through second moment normalization and tightening the globally stable learning rate bound. Motivated by this analysis, we propose Spectra, a spike aware optimizer that suppresses the dominant low rank spike subspace without amplifying the noise sensitive spectral tail. Spectra tracks the spike subspace via cached, warm started power iteration and applies low rank spectral shaping with negligible overhead and substantially reduced optimizer state memory. On LLaMA3 8B trained on 50B tokens, Spectra reaches the same target loss 30% faster than AdamW, reduces per step end to end overhead by 0.7%, cuts optimizer state memory by 49.25%, and improves average downstream accuracy by 1.62%. Compared to Muon, Spectra is 5.1x faster in optimizer processing time, achieves a lower final loss, and improves average accuracy by 0.66%.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 16 of its 26 references that are in the map. First posted on arXiv 30 Jan 2026.
Builds on
16 of 26 references.
References · 16
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2025Qwen3 Technical Reportcited by 8,195
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,140
- 2017RACE: Large-scale ReAding Comprehension Dataset From Examinationscited by 851
- 2019How Contextual are Contextualized Word Representations? Comparing the Geometry of BERT, ELMo, and GPT-2 Embeddingscited by 770
- 2022Large Language Models Struggle to Learn Long-Tail Knowledgecited by 452
- 2024DataComp-LM: In search of the next generation of training sets for language modelscited by 381
- 2017All-but-the-Top: Simple and Effective Postprocessing for Word Representationscited by 205
- 2024SOAP: Improving and Stabilizing Shampoo using Adamcited by 179
4 more
- 2021All Bark and No Bite: Rogue Dimensions in Transformer Language Models Obscure Representational Qualitycited by 150
- 2022LAMBADA: Backward Chaining for Automated Reasoning in Natural Languagecited by 66
- 2025PolarGrad: A Class of Matrix-Gradient Optimizers from a Unifying Preconditioning Perspectivecited by 48
- 2025Metis: Training LLMs with FP4 Quantizationcited by 9
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Towards Understanding the Power and Limits of the Muon Optimizer: A River-Valley Perspectivecited by 1
- 2026MALT: Lightweight Curvature-Aware Muon via Diagonal Preconditioningcited by 1
- 2026Accelerating Zeroth-Order Spectral Optimization with Partial Orthogonalization from Power Iterationcited by 0
- 2026No Subspace to Track: Non-Identifiability and Optimizer State in Low-Rank Trainingcited by 0
- 2026MALT: Lightweight Curvature-Aware Muon via Diagonal Preconditioning5 Aug 2026
- 2026No Subspace to Track: Non-Identifiability and Optimizer State in Low-Rank Training7 Jul 2026
- 2026Towards Understanding the Power and Limits of the Muon Optimizer: A River-Valley Perspective19 Jun 2026
- 2026Accelerating Zeroth-Order Spectral Optimization with Partial Orthogonalization from Power Iteration9 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026ORCA: The Annealed Spectral Conditioning Optimizer for Faster, Better LLM Training
- 2026Rethinking Muon Beyond Pretraining: Spectral Failures and High-Pass Remedies for VLA and RLVR
- 2026Task-Aware Spectral Pruning: A Mixture-of-Masks Framework for Efficient LLM Inference
- 2026Controlled LLM Training on Spectral Sphere
- 2025SparseLoRA: Accelerating LLM Fine-Tuning with Contextual Sparsity
- 2026Compute Where it Counts: Self Optimizing Language Models
- 2026Subspace Control: Turning Constrained Model Steering into Controllable Spectral Optimization
- 2026The Curse and Blessing of Mean Bias in FP4-Quantized LLM Training