mlsanityData to 14 Oct 2026

Rethinking Language Model Scaling under Transferable Hypersphere Optimization

arXiv 2603.28743 ↗First posted 30 Mar 2026cs.LGcited by 12

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

  1. 2026Mamba-3: Improved Sequence Modeling using State Space Principles16 Mar
  2. 2025Every Activation Boosted: Scaling General Reasoner to 1 Trillion Open Language Foundation25 Oct
  3. 2025gpt-oss-120b & gpt-oss-20b Model Card8 Aug

Other references · 31

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2020Language Models are Few-Shot Learnerscited by 32,419
  3. 2023GPT-4 Technical Reportcited by 18,396
  4. 2025Qwen3 Technical Reportcited by 8,820
  5. 2020Scaling Laws for Neural Language Modelscited by 6,586
  6. 2016Layer Normalizationcited by 5,669
  7. 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,326
  8. 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,796
  9. 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,512
  10. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  11. 2020GLU Variants Improve Transformercited by 1,804
  12. 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,346
19 more
  1. 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,135
  2. 2022LLM.int8(): 8-bit Matrix Multiplication for Transformers at Scalecited by 906
  3. 2024DeepSeek-V3 Technical Reportcited by 811
  4. 2025Kimi K2: Open Agentic Intelligencecited by 567
  5. 2016Weight Normalization: A Simple Reparameterization to Accelerate Training of Deep Neural Networkscited by 543
  6. 2023Analyzing and Improving the Training Dynamics of Diffusion Modelscited by 402
  7. 2022ST-MoE: Designing Stable and Transferable Sparse Expert Modelscited by 399
  8. 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 288
  9. 2018An Empirical Model of Large-Batch Trainingcited by 281
  10. 2024Auxiliary-Loss-Free Load Balancing Strategy for Mixture-of-Expertscited by 205
  11. 2023Tensor Programs VI: Feature Learning in Infinite-Depth Neural Networkscited by 97
  12. 2024How to set AdamW's weight decay as you scale model and dataset sizecited by 43
  13. 2025Power Lines: Scaling Laws for Weight Decay and Batch Size in LLM Pre-trainingcited by 43
  14. 2025Predictable Scale: Part I, Step Law -- Optimal Hyperparameter Scaling Law in Large Language Model Pretrainingcited by 34
  15. 2026Controlled LLM Training on Spectral Spherecited by 31
  16. 2025Completed Hyperparameter Transfer across Modules, Width, Depth, Batch and Durationcited by 27
  17. 2026A Unified View of Attention and Residual Sinks: Outlier-Driven Rescaling is Essential for Transformer Trainingcited by 23
  18. 2025Why Gradients Rapidly Increase Near the End of Trainingcited by 22
  19. 2025Nemotron-Flash: Towards Latency-Optimal Hybrid Small Language Modelscited by 15

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 12.

12 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Generalizing Scaling Laws for Dense and Sparse Large Language Models
  2. 2026How to Scale Mixture-of-Experts: From muP to the Maximally Scale-Stable Parameterization
  3. 2026Scaling Down the Scaling Laws: Parameter Efficiency and Compute-Optimal Training in Resource-Constrained Large Language Models
  4. 2025Compute-Optimal LLMs Provably Generalize Better With Scale
  5. 2026MONA: Muon Optimizer with Nesterov Acceleration for Scalable Language Model Training
  6. 2026Advancing Model Refinement: Muon-Optimized Distillation and Quantization for LLM Deployment
  7. 2022Transcending Scaling Laws with 0.1% Extra Compute
  8. 2025MuLoCo: Muon is a practical inner optimizer for DiLoCo