用模型自身历史轨迹动态调整训练标签,解决指令微调中的遗忘问题。
TMS: Trajectory-Mixed Supervision for Reward-Free, On-Policy SFT
- 基于模型历史版本生成动态训练标签,缓解标签与策略不匹配问题。
- 在MATH和GSM8K上显著提升准确率,接近强化学习的模型保留能力。
- 无需奖励模型或验证器,适合追求高效且稳定微调的研究者。
强化学习(RL)和监督微调(SFT)是提升大语言模型下游任务性能的两大主流范式。尽管RL通常比SFT更好地保持模型泛化能力(保留性),但其代价高昂:需复杂奖励设计、训练不稳定且采样成本高。相比之下,SFT虽高效却脆弱,常因“监督不匹配”导致灾难性遗忘——即模型策略演化与静态标签间的偏差。本文提出无奖励、在线策略的轨迹混合监督(TMS)框架,通过利用模型自身的历史检查点构建动态课程,近似实现RL的在线策略优势。TMS最小化策略-标签偏差(PLD),防止标准SFT中的模式崩溃。在推理(MATH、GSM8K)和指令遵循基准上的实验表明,TMS有效推动了准确率-保留性的帕累托前沿。尽管RL仍是保留性的金标准,TMS显著优于标准及迭代SFT,缩小了与RL的差距,且无需奖励模型或验证器。机制分析证实,PLD漂移可准确预测遗忘,而TMS能有效抑制该漂移。
原文摘要 · Abstract (English)
Reinforcement Learning (RL) and Supervised Fine-Tuning (SFT) are the two dominant paradigms for enhancing Large Language Model (LLM) performance on downstream tasks. While RL generally preserves broader model capabilities (retention) better than SFT, it comes with significant costs: complex reward engineering, instability, and expensive on-policy sampling. In contrast, SFT is efficient but brittle, often suffering from catastrophic forgetting due to $\textbf{Supervision Mismatch}$: the divergence between the model's evolving policy and static training labels. We address this trade-off with $\textbf{Trajectory-Mixed Supervision (TMS)}$, a reward-free framework that approximates the on-policy benefits of RL by creating a dynamic curriculum from the model's own historical checkpoints. TMS minimizes $\textit{Policy-Label Divergence (PLD)}$, preventing the mode collapse that drives forgetting in standard SFT. Experiments across reasoning (MATH, GSM8K) and instruction-following benchmarks demonstrate that TMS effectively shifts the accuracy--retention Pareto frontier. While RL remains the gold standard for retention, TMS significantly outperforms standard and iterative SFT, bridging the gap to RL without requiring reward models or verifiers. Mechanistic analysis confirms that PLD drift accurately predicts forgetting and that TMS successfully mitigates this drift.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。