arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2602.10657cs.CL
收藏

基准不是那么分布外:词重叠预测性能

The Lexical Imprint of Pre-training Data

Korea Advanced Institute of Science and Technology (KAIST)(韩国科学技术院)

Jeonghoon Kim, Woojin Chung, Woomin Song, Cheonbok Park, Nako Sung, Jinwoo Shin

AI总结:

本研究发现词重叠程度与基准性能呈反比关系,表明基准性能受预训练数据与测试数据之间的词重叠影响,词频统计在其中起重要作用。

摘要:

理解构成高质量预训练数据的因素仍然是语言模型训练中的核心问题。在本工作中,我们研究了基准性能是否主要由预训练语料库与评估数据集之间的统计模式重叠程度决定。我们通过词级单字交叉熵和词频统计来测量这种重叠,并在10个零样本基准、4个覆盖85亿至600亿词的预训练数据集以及模型大小从4亿到300亿参数的范围内进行受控实验。我们的结果表明词级单字交叉熵与基准性能之间存在稳健的反比关系,这表明广泛使用的基准强烈受到训练和评估数据之间词重叠的影响。因此,具有相似词级单字交叉熵的更大预训练子集会产生更好的下游结果,表明词频统计在塑造基准分数中起着额外的作用。综合来看,这些结果表明许多标准基准相对于预训练语料库只略微分布外,因此简单的词重叠统计可以预测基准性能。

英文摘要:

How much of zero-shot benchmark accuracy is captured by word-level alignment between a pre-training corpus and the benchmark target? We study this question through word-level unigram cross-entropy (UCE), a tokenizer-agnostic, model-free measure of lexical alignment between a reference corpus and a target benchmark. Likelihood training shapes the model's predictive distribution from the pre-training corpus, and zero-shot scores under likelihood-based evaluation depend in part on the lexical alignment between that corpus and the benchmark target. Cleanly tracing how the lexical imprint of pre-training data scales with corpus--benchmark alignment requires controlled from-scratch pre-training that varies corpus identity while holding the training recipe fixed. Across this controlled sweep, spanning $10$ zero-shot benchmarks, $4$ pre-training corpora, and $3$ model scales, we find that corpus--benchmark lexical alignment, UCE, consistently tracks zero-shot accuracy on every benchmark, with corpora more aligned with the benchmark (lower UCE) achieving higher accuracy. The same signal extends from measurement to control at finer scopes. At the document level, strengthening the imprint by selecting training documents to maximize lexical alignment with the benchmark produces the accuracy gains predicted by the mechanism. At the prompt level, matching the imprint per prompt by routing each prompt to the drafter whose domain corpus is most lexically aligned with it produces measurable throughput gains over a generalist-drafter baseline. Taken together, pre-training corpora leave a measurable lexical imprint on base-model zero-shot behavior, and UCE provides a model-free probe of lexical alignment between reference and target data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑