AI & ML
Spectral Condition for $\mu$P under Width-Depth Scaling
arXiv stat.ML0 views
arXiv:2603.00541v3 Announce Type: replace-cross Abstract: Generative foundation models are increasingly scaled in both width and depth, posing significant challenges for stable feature learning and reliable hyperparameter (HP) transfer across model sizes. While maximal update parameterization~($\mu$P) has provided a principled solution to both problems for width scaling, existing extensions to the joint width-depth scaling regime remain fragmented, architecture- and optimizer-specific, and often rely on technically involved theories. In this work, we develop a simple and unified spectral framework for $\mu$P under joint width-depth scaling. For deep residual networks whose residual blocks contain $k$ transformations, the framework specifies how the norms of weights and their per-step updates should scale with width and depth. It reveals a fundamental transition from $k=1$ to $k\geq 2$, {unifying previously disparate $\mu$P formulations and identifying the CompleteP-style parameterization ($k\geq 2$)} as more appropriate for practical architectures with multi-transformation branches such as Transformers. Building on this framework, we derive a general recipe for implementing $\mu$P across a broad class of optimizers by mapping spectral constraints to {concrete, optimizer-specific HP parameterizations}, recovering existing results and extending them to additional optimizers. {Finally, experiments on GPT-2 style language models show that the CompleteP-style parameterization ($k\geq 2$) achieves stable feature learning and robust HP transfer under width-depth scaling, whereas standard parameterization and the Depth-$\mu$P-style parameterization ($k=1$) often fail to do so.} These results support the practical effectiveness of the proposed spectral framework.
Source:https://arxiv.org/abs/2603.00541PreviousA Function-Space Stability Boundary for Generalization in Interpolating Learning SystemsNextToward Optimal Switching Regret for Multi-Armed Bandits with Oblivious Adversary
Related posts
Moonshot AI's Kimi K3.1 Pro Weights Are Now Open — Explained Honestly
0dev.to AI

Everyone has better models. Why do some teams fly while others don't?
0dev.to AI

"Trace Shape Reveals Model Capability: 3B vs 7B"
0dev.to AI
I've spent a lot of time thinking about how to use technology more effectively
0dev.to WebDev
HackerTube AI: A Human-Curated Video Archive for Curious Learners
0dev.to WebDev
CPCSC for build-to-print shops: does it apply?
0dev.to Security