arXiv:2608.30627cs.CL2026-08

用推理注释增强预训练数据,让模型更容易预测下一个词

REER-PT: Reverse-Engineered Reasoning for Perplexity-Guided Pre-training Data Augmentation

  • 通过困惑度优化生成简洁推理注释,补全上下文与续词间的逻辑断层
  • 在多个基准上模型性能提升最高达2.07个百分点,困惑度降低0.42至7.29
  • 仅0.05%的注释词组与原文重复,适合追求高效微调的开发者

随着语言模型算力持续增长,高质量训练数据成为关键瓶颈。传统下一个词预测只监督上下文后的输出,但忽略了中间推理过程。本文提出可扩展的REER-PT框架,将逆向推理(REER)应用于原始预训练数据。该方法识别难以预测但可由上下文推断出的续词,并插入简洁的推理注释,重建上下文与续词间的缺失关联。候选注释离线生成并优化,以困惑度为信号;长度和目标泄露约束过滤无效或平凡注释。这种稀疏转换保留源文本,兼容标准下一个词预测,无需预训练时在线推理。我们将此方法应用于源预训练语料库,生成增强语料。在增强数据、原词和选择性续词对比中,困惑度下降0.42至7.29,且仅有约0.05%的13-gram注释与源文本完全一致。随后使用相同架构和训练配置,在源语料和增强语料上分别训练两个680M参数模型。增强语料模型在多个知识与推理基准上提升最高达2.07个百分点。整体分析表明,增强后续词更易预测,而控制实验也证实该增强能提升模型性能,无需改变标准预训练目标。

原文摘要 · Abstract (English)

As language-model compute continues to scale, high-quality training data is becoming an increasingly important bottleneck. Conventional next-token prediction supervises what follows a context but leaves the intermediate reasoning behind that continuation implicit. We introduce \textbf{REER-PT}, a scalable framework that extends Reverse-Engineered Reasoning (REER) to raw pre-training data. REER-PT identifies continuations that are difficult to predict but can still be inferred from the preceding context, and inserts concise reasoning annotations that reconstruct the missing connection between context and continuation. Candidate annotations are generated and refined offline, with perplexity serving as the optimization signal. Constraints on length and target leakage filter out unhelpful or trivial annotations. This sparse transformation preserves the source text and remains compatible with standard next-token prediction, avoiding online reasoning rollouts during pre-training. We apply REER-PT to transform a source pre-training corpus into an augmented one. Across augmented-data, original-token, and selected-continuation comparisons, perplexity reductions range from 0.42 to 7.29, and only about 0.05\% of annotation 13-grams appear verbatim in the source text. We then train two 680M-parameter models with the same architecture and training configuration on the source and augmented corpora, respectively. The augmented-data model gains up to 2.07 percentage points on several knowledge and reasoning benchmarks. Together, the perplexity analysis indicates improved continuation predictability, while the controlled pre-training experiments suggest that this augmentation can improve model performance without changing the standard pre-training objective.

预训练数据增强推理注入困惑度

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。