Skip to content

Author

Lei Wu

We have 3 of 10 papers

We haven’t gathered this author’s papers yet. Follow them and we’ll fetch their work.

Not the right person? Other researchers publish under this name.

#artificial intelligence Preprint Sep 2026

How Does Local Landscape Geometry Evolve in Language Model Pre-Training?

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
#machine learning Preprint Sep 2026

Momentum in large-batch training: Polyak enlarges the critical batch size, Nesterov improves data efficiency

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
Preprint Aug 2026

Effective Learning Rate Governs Loss Dynamics in Language Model Pretraining

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.