How to Set the Learning Rate for Large-Scale Pre-training?

Yunhua Zhou, Shuhao Xing, Junhao Huang, Xipeng Qiu, Qipeng Guo


Abstract
Optimal configuration of the learning rate (LR) is a fundamental yet formidable challenge in large-scale pre-training. Given the stringent trade-off between training costs and model performance, the pivotal question is whether the optimal LR can be accurately extrapolated from low-cost experiments. In this paper, we formalize this investigation into two distinct research paradigms: Fitting and Transfer. Within the Fitting Paradigm, we innovatively introduce a Scaling Law for search factor, effectively reducing the search complexity from 𝒪(n3) to 𝒪(n ⋅ CD ⋅ C𝜂) via predictive modeling. Within the Transfer Paradigm, we extend the principles of 𝜇Transfer to the Mixture of Experts (MoE) architecture, broadening its applicability to encompass model depth, weight decay, and token horizons.By pushing the boundaries of existing hyperparameter research in terms of scale, we conduct a comprehensive comparison between these two paradigms. Our empirical results challenge the scalability of the widely adopted 𝜇Transfer in large-scale pre-training scenarios. Furthermore, we provide a rigorous analysis through the dual lenses of training stability and feature learning to elucidate the underlying reasons why module-wise parameter tuning underperforms in large-scale settings. This work offers systematic practical guidelines and a fresh theoretical perspective for optimizing industrial-level pre-training.
Anthology ID:
2026.findings-acl.1861
Volume:
Findings of the Association for Computational Linguistics: ACL 2026
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
37344–37361
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.1861/
DOI:
Bibkey:
Cite (ACL):
Yunhua Zhou, Shuhao Xing, Junhao Huang, Xipeng Qiu, and Qipeng Guo. 2026. How to Set the Learning Rate for Large-Scale Pre-training?. In Findings of the Association for Computational Linguistics: ACL 2026, pages 37344–37361, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
How to Set the Learning Rate for Large-Scale Pre-training? (Zhou et al., Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.1861.pdf
Checklist:
 2026.findings-acl.1861.checklist.pdf