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
🚀 Ready to explore the world of AWS Cloud and build something amazing together? ☁️
0dev.to Programming
The Second Scrape Should Be Almost Free
0dev.to Programming
Updates to Authtics Advisories
0dev.to Programming
My betting model looked great for 3,671 predictions. It never beat the bookmaker.
0dev.to TypeScript

Build efficient agents with knowledge distillation on Amazon SageMaker AI
0AWS Machine Learning Blog

Build World Models for Interactive Simulation with NVIDIA Cosmos
0NVIDIA Developer Blog