预训练模型的覆盖能力决定微调效果,比损失值更关键。
The Coverage Principle: How Pre-Training Enables Post-Training
- 用‘覆盖度’衡量模型对优质输出的概率分配,解释预训练为何有效。
- 覆盖度比交叉熵损失更能预测下游任务表现,且泛化更快。
- 提出选模型、归一化梯度、测试时解码等可证明有效的改进方法。
语言模型在大规模文本上预训练后微调,能展现惊人能力,但预训练如何影响最终性能仍不清晰。尽管预训练常以交叉熵损失衡量,但该指标往往无法准确预测下游表现。我们从‘覆盖’角度提供理论视角:覆盖指预训练模型对高质量响应分配的概率质量,是后训练及测试时缩放方法(如Best-of-N)成功的关键条件。研究发现,最大似然(如下一个词预测)隐式优化出高覆盖模型,形成‘覆盖原理’。核心机制在于:覆盖度比交叉熵损失泛化更快,不受序列长度等特定问题参数干扰。我们还提出三类可证明提升覆盖度的实用方法:(i) 模型/检查点选择策略,(ii) 梯度归一化方案,(iii) 测试时解码策略。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。