The scale and expense of pre-training language models make efficient hyperparameter tuning essential, yet a principled guidance is still missing. In this work, we analyze language model pre-training dynamics from a local landscape geometry perspective. Our study reveals two distinct phases. In Phase I, sharpness of the...
Zhan-Peng Zhou, Yu-Han Sun, Bing-Rui Li et al.· 1 citation
A three-regime batch-size phase diagram that reveals how the role of momentum changes with batch size is revealed, with Polyak enlarging the critical batch size, the largest batch size preserving the best small-batch data-scaling exponent, thereby enabling greater parallelism without sacrificing data efficiency.
Jia-Nan Wang, Zi-Xun Huang, Kai-Rui Li et al.· 0 citations
Results establish ELR as a common coordinate linking LR scheduling, norm control, and loss dynamics, and Controlled interventions further show that weight decay and Hyperball shape loss dynamics primarily through the ELR schedules they induce.
Zi-Han Liu, Rui-Heng Zheng, Shaobo Zhang et al.· 4 citations
We use cookies to run the site and, with your consent, for analytics and to show ads.
See our Cookie Policy.