arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2502.02007cs.CLcs.LG
收藏

下一 Token 预测训练的推理偏差

Reasoning Bias of Next Token Prediction Training

School of Mathematical Sciences, Shanghai Jiao Tong University(上海交通大学数学科学学院);School of Artificial Intelligence, Shanghai Jiao Tong University(上海交通大学人工智能学院);Institute for Advanced Algorithms Research, Shanghai(上海高级算法研究院)

Pengxiao Lin, Zhongwang Zhang, Zhi-Qin John Xu

AI总结:

本文对比 NTP 与 CTP 训练范式,发现训练噪声带来的正则化使 NTP 模型在推理基准上更鲁棒、泛化更好,并指出二者分别适用于预训练与微调。

摘要:

自大型语言模型(LLM)诞生以来,如何高效训练它们以获得卓越的推理能力,一直是一项关键挑战。LLM 的主流训练范式基于下一 token 预测(NTP)。另一类被称为 Critical Token Prediction(CTP,关键 token 预测)的方法则只关注特定关键 token(例如问答数据集中的答案),旨在减少对无关信息和噪声的过拟合。与最初假设相反,我们的研究表明,尽管 NTP 在训练过程中会接触噪声,但其推理能力却优于 CTP。我们将这一反直觉结果归因于噪声对训练动态的正则化影响。实证分析显示,经 NTP 训练的模型在多个基准推理数据集上表现出更强的泛化性和鲁棒性,对扰动具有更强的恢复力,并达到更平坦的损失极小值。这些发现说明,NTP 有助于在预训练阶段培养推理能力,而 CTP 对微调更有效,从而加深了我们对 LLM 开发中最优训练策略的理解。

英文摘要:

Since the inception of Large Language Models (LLMs), the quest to efficiently train them for superior reasoning capabilities has been a pivotal challenge. The dominant training paradigm for LLMs is based on next token prediction (NTP). Alternative methodologies, called Critical Token Prediction (CTP), focused exclusively on specific critical tokens (such as the answer in Q\&A dataset), aiming to reduce the overfitting of extraneous information and noise. Contrary to initial assumptions, our research reveals that despite NTP's exposure to noise during training, it surpasses CTP in reasoning ability. We attribute this counterintuitive outcome to the regularizing influence of noise on the training dynamics. Our empirical analysis shows that NTP-trained models exhibit enhanced generalization and robustness across various benchmark reasoning datasets, demonstrating greater resilience to perturbations and achieving flatter loss minima. These findings illuminate that NTP is instrumental in fostering reasoning abilities during pretraining, whereas CTP is more effective for finetuning, thereby enriching our comprehension of optimal training strategies in LLM development.

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

↑