ByteDance Seed: Effective Learning Rate Trajectories, Not Normalization, Drive WN Training Gains

ByteDance-Seed · hf · 2026-10-30

A paper from ByteDance Seed and Peking University shows that the recent pretraining loss gains attributed to weight-normalized (WN) training are driven primarily by the induced effective learning rate (ELR) trajectory rather than normalization alone — matching ELR trajectories largely collapses the training loss dynamics of WN and weight decay.

Schedule design should therefore target the ELR rather than the nominal learning rate. Building on this principle, the authors propose AIR (Affine Inverse-square-Root decay), which improves both large-scale pretraining loss and robustness.

Original post →

More from Infra

Infra channel →