arXiv:2606.09396cs.CLcs.LG2026-06

用预训练模型稳定重加权,提升SFT泛化能力

PriFT: Prior-Support Guided Supervised Fine-Tuning

论文配图:PriFT: Prior-Support Guided Supervised Fine-Tuning
图 1 · 摘自论文原文
  • 用冻结的预训练模型计算令牌权重,避免优化轨迹干扰
  • 在数学推理、代码生成等任务上超越现有SFT基线
  • 适合需要高质量初始化的强化学习下游任务

监督微调(SFT)是下游任务适配的高效方法,常作为强化学习(RL)的初始化阶段,但其泛化能力常弱于RL。主要瓶颈在于其离策略目标:SFT逐标记拟合固定示范,包括与模型预训练分布不匹配的目标标记,易导致过拟合。现有方法通过为与当前模型预测分布更一致的标记分配更高训练权重来缓解,但该权重由正在微调的模型计算,会与优化轨迹纠缠,引发自增强动态,使分布快速偏离预训练模型。为此,我们提出PriFT(Prior-support guided Fine-Tuning),从冻结的预训练参考模型获取权重信号,获得不受微调影响的稳定重加权。该信号衡量先验支持度:每个目标标记被预训练分布支持的程度。在多个现有重加权规则中,将在线模型替换为预训练模型作为重加权信号均能持续提升性能。我们提出两种实现:PriFT-prob使用预训练标记概率,PriFT-mass基于预训练分布的累积概率质量选择标记。在数学推理、代码生成和医学问答等多个任务上的大量实验表明,PriFT在SFT基线中达到最先进水平,并为后续强化学习训练提供更优初始化。

原文摘要 · Abstract (English)

Supervised fine-tuning (SFT) is an efficient approach for downstream task adaptation and often serves as the initialization stage for reinforcement learning (RL), but it can show weaker generalization than RL. A key limitation is its off-policy objective: SFT fits fixed demonstrations token by token, including targets poorly aligned with the model's pretrained distribution, which can lead to overfitting. A recent line of work addresses this issue by assigning larger training weights to tokens better aligned with the current model's predictive distribution, with the intuition that fitting these tokens are less distortive to the model's pretrained knowledge and representations. However, computing the token weights from the model that is currently fine-tuned entangles token weights with the optimization trajectory, inducing a self-reinforcing dynamics as the distribution rapidly departs from the pretrained model. To address this, we propose PriFT (Prior-support guided Fine-Tuning), which derives token weights from a frozen pretrained reference to obtain a stable reweighting signal unaffected by fine-tuning. This signal estimates prior support: the extent to which each target token is supported by the pretrained distribution. Across multiple existing token-reweighting rules, replacing the reweighting signal from the online model to pretrained model consistently improves performance. We introduce two instantiations: PriFT-prob uses pretrained token probability, while PriFT-mass selects tokens by cumulative probability mass under the pretrained distribution. Extensive experiments on mathematical reasoning, code generation, and medical question answering show that PriFT achieves state-of-the-art results among SFT baselines and provides a better initialization for subsequent RL training.

微调预训练强化学习

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