Optimizer Convergence Dynamics & Momentum Selection Rules establishes the algorithmic decision matrix choosing between first-order momentum and adaptive learning rate optimizers based on model topology (CNNs, Transformers, Sparse Embeddings); the 3 dominant families are: 1) SGD with Momentum / Nesterov Accelerated Gradient (NAG), dominating Computer Vision due to flat minima generalization; 2) AdamW (Decoupled Weight Decay Adam: updating θt+1=θt−η(vt+ϵmt+λθt) by isolating weight decay from adaptive gradient scaling), serving as the universal standard for Transformers and LLMs; 3) AdaGrad / FTRL, optimizing sparse categorical embeddings in ad ranking.
💡Use Cases
LLM pre-training/fine-tuning, computer vision network training, and debugging training divergence or loss oscillations.
⚡Key Problems Solved
Original Adam coupled L2 regularization into gradient updates, causing adaptive learning rate scaling to suppress weight decay on frequent tokens and degrading generalization; AdamW decouples weight decay, achieving fast convergence and superior out-of-distribution generalization.
🎯5 High-Frequency Exam Points
1
Derive the mathematical difference between original Adam and AdamW in weight decay updates, explaining why L2 regularization fails in Adam?
2
Derive the initial step bias correction formulas m^t=1−β1tmt and v^t=1−β2tvt in Adam/AdamW?
3
Why does SGD with Momentum often generalize better to flat minima in Computer Vision compared to adaptive optimizers?
4
Explain memory savings of Adafactor (low-rank factorization) and the sign-gradient operation of the Lion optimizer?
5
How to tune β1,β2, and ϵ to stabilize training against loss spikes and gradient oscillations in mixed precision?
🔗Foundational Prerequisite Cards (Click to Review)