arXiv:2504.11336cs.LGcs.AI2025-04被引 5

通过重排训练数据,让模型先生成目标再补细节,提升推理与规划能力。

Looking beyond the next token

  • 用目标先行的训练序列重构数据,模仿人类写作逻辑
  • 在规划、算法推理等任务上显著提升性能,长目标生成零成本
  • 无需改架构,适合需要深度推理的应用场景

因果语言模型训练假设每个词可由前文准确预测,这与人类写作和推理时先有目标再组织内容的过程不符。尽管已有研究关注此差异,普遍认为需改变模型结构来解决。本文提出,仅通过重新排列和处理训练数据序列,即可使模型更贴近真实的数据生成过程,无需修改架构或训练基础设施。我们提出的Trelawney方法及其推导的推理算法,在多个关键基准上表现优异,涵盖规划、算法推理和故事生成任务。此外,该方法自然支持无额外开销的长周期目标生成,并探索了利用模型目标生成能力进一步提升规划与推理的效果。我们认为Trelawney可能为超越当前语言建模范式的新能力打开通道。

原文摘要 · Abstract (English)

The structure of causal language model training assumes that each token can be accurately predicted from the previous context. This contrasts with humans' natural writing and reasoning process, where goals are typically known before the exact argument or phrasings. While this mismatch has been well studied in the literature, the working assumption has been that architectural changes are needed to address this mismatch. We argue that rearranging and processing the training data sequences can allow models to more accurately imitate the true data-generating process, and does not require any other changes to the architecture or training infrastructure. We demonstrate that this technique, Trelawney, and the inference algorithms derived from it allow us to improve performance on several key benchmarks that span planning, algorithmic reasoning, and story generation tasks. Finally, our method naturally enables the generation of long-term goals at no additional cost. We investigate how using the model's goal-generation capability can further improve planning and reasoning. Additionally, we believe Trelawney could potentially open doors to new capabilities beyond the current language modeling paradigm.

语言模型推理生成数据重构

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