paper-with-me

홈 › Papers

Same Architecture, Different Capacity: Optimizer-Induced Spectral Scaling Laws

2026-05-20 · Nandan Kumar Jha, Brandon Reagen arxiv

Scaling laws have made language-model performance predictable from model size, data, and compute, but they typically treat the optimizer as a fixed training detail. We show that this assumption misses a fundamental axis of representation scaling: how effectively the optimizer converts added FFN width into utilized spectral capacity. Using eigenspectra of feed-forward network representations, measured through soft and hard spectral-ranks, we find that \emph{the same Transformer architecture realizes markedly different spectral scaling laws when trained with different optimizers}. Holding architecture and width schedule fixed, AdamW exhibits weak hard-rank scaling ($β$=0.44) on rare-token (TAIL) representations where learning is known to be hardest, whereas Muon achieves linear scaling ($β$=1.02) in the same regimes, a $2.3\times$ increase in the scaling exponent. This difference is not reducible to validation loss: AdamW configurations can match low-rank Dion variants in perplexity, under extended training, while exhibiting sharply different spectral geometry, demonstrating that matched loss does not imply matched representation structure. Hard--soft rank asymmetry further reveals that optimizers differ not only in how much capacity is realized, but also in how that capacity is structured across eigenmodes. To disentangle optimizer effects from architectural ones, we compare against architectural interventions (e.g., attention rank and positional encoding), and find that optimizer-induced spectral shifts often exceed the architectural effects. These results suggest optimization as a first-class axis of representation scaling, motivating optimizer--architecture co-design.

📄 PDF Abstract BibTeX arXiv:2605.21803

Code (0)

등록된 구현이 없습니다.

Similar Papers 제목 키워드 기반

Optimizer-Induced Mode Connectivity: From AdamW to Muon

2026-05-11 · Fangzhao Zhang, Sungyoon Kim, Erica Zhang, Yiqi Jiang 외 arxiv

Mode connectivity has been widely studied, yet the role of the optimizer remains underexplored. We revisit it through optimizer-induced implicit regularization, asking how connectivity behaves when restricted to solution…

Noise-Adaptive Layerwise Learning Rates: Accelerating Geometry-Aware Optimization for Deep Neural Network Training

2025-10-15 · Jie Hao, Xiaochuan Gong, Jie Xu, Zhengdao Wang 외 arxiv

Geometry-aware optimization algorithms, such as Muon, have achieved remarkable success in training deep neural networks (DNNs). These methods leverage the underlying geometry of DNNs by selecting appropriate norms for di…

Pre-Training LLMs on a budget: A comparison of three optimizers

2025-07-11 · Joel Schlotthauer, Christian Kroos, Chris Hinze, Viktor Hangya 외 arxiv

Optimizers play a decisive role in reducing pre-training times for LLMs and achieving better-performing models. In this study, we compare three major variants: the de-facto standard AdamW, the simpler Lion, developed thr…

The Role of Linear Layers in Nonlinear Interpolating Networks

2022-02-02 · Greg Ongie, Rebecca Willett

This paper explores the implicit bias of overparameterized neural networks of depth greater than two layers. Our framework considers a family of networks of varying depth that all have the same capacity but different imp…

Large Language Model Assisted Adversarial Robustness Neural Architecture Search

2024-06-08 · Rui Zhong, Yang Cao, Jun Yu, Masaharu Munetomo

Motivated by the potential of large language models (LLMs) as optimizers for solving combinatorial optimization problems, this paper proposes a novel LLM-assisted optimizer (LLMO) to address adversarial robustness neural…

Adversarial RobustnessCombinatorial OptimizationLanguage ModelingLanguage Modelling+2