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

Nandan Kumar Jha, Brandon Reagen

arXiv:2605.21803 · 2026-05-22 공개 · arXiv · PDF

transformer model-capacity muon adamw spectral-scaling optimizer-architecture eigenspectra hard-rank

Abstract

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 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.3times 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.

한국어 요약

한 줄 요약

**[Scaling Laws / Optimizer-Architecture Co-Design]** 동일 Transformer가 옵티마이저에 따라 다른 spectral scaling law 보임 — AdamW는 TAIL representation에서 hard-rank β=0.44, Muon은 β=1.02로 2.3× 큰 스케일링 지수, matched loss여도 spectral geometry는 sharply 다름.

핵심 기여도

핵심 아이디어

Validation loss가 같아도 옵티마이저는 spectral geometry를 sharply 다르게 형성하며, FFN의 utilized spectral capacity는 architecture·width 조절보다 옵티마이저 선택에 더 강하게 의존한다 — optimization을 representation scaling의 first-class 축으로 격상해 architecture와 co-design해야 한다.

기술적 접근법

주요 결과

의의 및 한계

**의의**: scaling law 연구에 옵티마이저라는 새 축 추가, matched loss ≠ matched representation의 정량 입증, optimizer-architecture co-design의 필요성 정립. **한계**: language model FFN 중심으로 multimodal·vision 일반화 추가 검증, 새 옵티마이저(Muon, Dion) 등 비교에 한정, downstream task 성능과 spectral-rank의 인과 연결은 partial.

실용적 활용