The Coverage Principle: How Pre-Training Enables Post-Training

LLM PTSAD ML NNGTB SSLRO LFDDR FT
语言模型在大规模文本语料库上进行预训练,并针对特定任务进行微调后,展现出卓越的能力。然而,预训练如何以及为何能够决定最终模型的成功,目前仍缺乏充分理解。值得注意的是,尽管预训练的效果通常以交叉熵损失来衡量,但交叉熵对下游任务性能的预测能力往往较差。为此,我们通过“覆盖率”这一概念提供了一种理论视角:覆盖率衡量了预训练模型为高质量回复分配的概率质量,而这一指标对于诸如Best-of-N等后训练及测试时扩展方法的成功而言,是必要且充分的。我们的主要成果在于阐明了“覆盖率原则”这一现象,即下一个词预测(更广义地说,最大似然估计)会隐式地优化模型,使其具备良好的覆盖率。特别地,我们揭示了一个机制,解释了覆盖率为何能更有效地预测下游性能:与交叉熵相比,**覆盖率的泛化速度更快**,避免了对问题相关参数(如序列长度)产生虚假依赖。此外,我们还研究了几种具有可证明优势、可用于提升覆盖率的实用算法干预措施,包括:(i)模型/检查点选择方法,(ii)梯度归一化方案,以及(iii)测试时解码策略。
Language models demonstrate remarkable abilities when pre-trained on large text corpora and fine-tuned for specific tasks, but how and why pre-training shapes the success of the final model remains poorly understood. Notably, although pre-training success is often quantified by cross-entropy loss, cross-entropy can be a poor predictor of downstream performance. Instead, we provide a theoretical perspective on this relationship through the lens of \emph{coverage}, which quantifies the probability mass the pre-trained model places on high-quality responses and which is necessary and sufficient for post-training and test-time scaling methods such as Best-of-N to succeed. Our main results develop an understanding of \emph{the coverage principle}, a phenomenon whereby next-token prediction (more generally, maximum likelihood) implicitly optimizes toward a model with good coverage. In particular, we uncover a mechanism that explains the power of coverage in predicting downstream performance: \emph{coverage generalizes faster than cross-entropy}, avoiding spurious dependence on problem-dependent parameters such as the sequence length. We also study practical algorithmic interventions with provable benefits for improving coverage, including (i) model/checkpoint selection procedures, (ii) gradient normalization schemes, and (iii) test-time decoding strategies.
许愿