Rethinking Language Model Scaling under Transferable Hypersphere Optimization
On the map: LLM reasoning → Test-time compute and efficient thinking
also in 1 more
Abstract (arXiv)
Scaling laws for large language models depend critically on the optimizer and parameterization. Existing hyperparameter transfer laws are mainly developed for first-order optimizers, and they do not structurally prevent training instability at scale. Recent hypersphere optimization methods constrain weight matrices to a fixed-norm hypersphere, offering a promising alternative for more stable scaling. We introduce HyperP (Hypersphere Parameterization), the first framework for transferring optimal learning rates across model width, depth, training tokens, and Mixture-of-Experts (MoE) granularity under the Frobenius-sphere constraint with the Muon optimizer. We prove that weight decay is a first-order no-op on the Frobenius sphere, show that Depth-μP remains necessary, and find that the optimal learning rate follows the same data-scaling power law with the "magic exponent" 0.32 previously observed for AdamW. A single base learning rate tuned at the smallest scale transfers across all compute budgets under HyperP, yielding 1.58× compute efficiency over a strong Muon baseline at 6×1021 FLOPs. Moreover, HyperP delivers transferable stability: all monitored instability indicators, including Z-values, output RMS, and activation outliers, remain bounded and non-increasing under training FLOPs scaling. We also propose SqrtGate, an MoE gating mechanism derived from the hypersphere constraint that preserves output RMS across MoE granularities for improved granularity scaling, and show that hypersphere optimization enables substantially larger auxiliary load-balancing weights, yielding both strong performance and good expert balance. We release our training codebase at https://github.com/microsoft/ArchScale.
Key facts. Cited by 12 papers in the map (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 34 of its 56 references that are in the map. First posted on arXiv 30 Mar 2026.
Builds on
34 of 56 references.
Closest ancestors · 3
Same direction, newest first
- 2026Mamba-3: Improved Sequence Modeling using State Space Principles16 Mar
- 2025Every Activation Boosted: Scaling General Reasoner to 1 Trillion Open Language Foundation25 Oct
- 2025gpt-oss-120b & gpt-oss-20b Model Card8 Aug
Other references · 31
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2023GPT-4 Technical Reportcited by 18,396
- 2025Qwen3 Technical Reportcited by 8,820
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2016Layer Normalizationcited by 5,669
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,326
- 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,796
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,512
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2020GLU Variants Improve Transformercited by 1,804
- 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,346
19 more
- 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,135
- 2022LLM.int8(): 8-bit Matrix Multiplication for Transformers at Scalecited by 906
- 2024DeepSeek-V3 Technical Reportcited by 811
- 2025Kimi K2: Open Agentic Intelligencecited by 567
- 2016Weight Normalization: A Simple Reparameterization to Accelerate Training of Deep Neural Networkscited by 543
- 2023Analyzing and Improving the Training Dynamics of Diffusion Modelscited by 402
- 2022ST-MoE: Designing Stable and Transferable Sparse Expert Modelscited by 399
- 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 288
- 2018An Empirical Model of Large-Batch Trainingcited by 281
- 2024Auxiliary-Loss-Free Load Balancing Strategy for Mixture-of-Expertscited by 205
- 2023Tensor Programs VI: Feature Learning in Infinite-Depth Neural Networkscited by 97
- 2024How to set AdamW's weight decay as you scale model and dataset sizecited by 43
- 2025Power Lines: Scaling Laws for Weight Decay and Batch Size in LLM Pre-trainingcited by 43
- 2025Predictable Scale: Part I, Step Law -- Optimal Hyperparameter Scaling Law in Large Language Model Pretrainingcited by 34
- 2026Controlled LLM Training on Spectral Spherecited by 31
- 2025Completed Hyperparameter Transfer across Modules, Width, Depth, Batch and Durationcited by 27
- 2026A Unified View of Attention and Residual Sinks: Outlier-Driven Rescaling is Essential for Transformer Trainingcited by 23
- 2025Why Gradients Rapidly Increase Near the End of Trainingcited by 22
- 2025Nemotron-Flash: Towards Latency-Optimal Hybrid Small Language Modelscited by 15
Built on by
Citing papers per year: 2026: 12.
12 papers.
- 2026How Much Is One Recurrence Worth? Iso-Depth Scaling Laws for Looped Language Modelscited by 24
- 2026Nora: Normalized Orthogonal Row Alignment for Scalable Matrix Optimizercited by 5
- 2026Learning Rate Transfer in Normalized Transformerscited by 3
- 2026Anytime Training with Schedule-Free Spectral Optimizationcited by 3
- 2026Improving Neural Network Training by Decoupling the Magnitude and Direction of Weight Vectorscited by 3
- 2026Normalized Architectures are Natively 4-Bitcited by 2
- 2026Scale Weight Decay and Train Bettercited by 2
- 2026Complete-muE: Optimal Hyperparameter Transfer and Scaling for MoE Modelscited by 1
- 2026PC Layer: Polynomial Weight Preconditioning for Improving LLM Pre-Trainingcited by 1
- 2026HyperTransfer: Understanding the Equivalence between Base Optimizer and Hyperballcited by 0
- 2026Quantization-Aware Pre-Training with Constrained Empirical Weight Distributioncited by 0
- 2026Optimizer-dependent training dynamics converge to the same one-third optimal data scalingcited by 0
- 2026Optimizer-dependent training dynamics converge to the same one-third optimal data scaling29 Sep 2026
- 2026Quantization-Aware Pre-Training with Constrained Empirical Weight Distribution26 Sep 2026
- 2026HyperTransfer: Understanding the Equivalence between Base Optimizer and Hyperball7 Sep 2026
- 2026Scale Weight Decay and Train Better26 Jul 2026
- 2026Improving Neural Network Training by Decoupling the Magnitude and Direction of Weight Vectors24 Jun 2026
- 2026PC Layer: Polynomial Weight Preconditioning for Improving LLM Pre-Training4 Jun 2026
- 2026Complete-muE: Optimal Hyperparameter Transfer and Scaling for MoE Models22 May 2026
- 2026Anytime Training with Schedule-Free Spectral Optimization21 May 2026
- 2026Normalized Architectures are Natively 4-Bit7 May 2026
- 2026Nora: Normalized Orthogonal Row Alignment for Scalable Matrix Optimizer5 May 2026
- 2026Learning Rate Transfer in Normalized Transformers29 Apr 2026
- 2026How Much Is One Recurrence Worth? Iso-Depth Scaling Laws for Looped Language Models22 Apr 2026
Similar, not cited
Close in wording, no citation either way
- 2025Generalizing Scaling Laws for Dense and Sparse Large Language Models
- 2026How to Scale Mixture-of-Experts: From muP to the Maximally Scale-Stable Parameterization
- 2026Scaling Down the Scaling Laws: Parameter Efficiency and Compute-Optimal Training in Resource-Constrained Large Language Models
- 2025Compute-Optimal LLMs Provably Generalize Better With Scale
- 2026MONA: Muon Optimizer with Nesterov Acceleration for Scalable Language Model Training
- 2026Advancing Model Refinement: Muon-Optimized Distillation and Quantization for LLM Deployment
- 2022Transcending Scaling Laws with 0.1% Extra Compute
- 2025MuLoCo: Muon is a practical inner optimizer for DiLoCo