arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.03073cs.LG

TMS:轨迹混合监督用于无奖励、在线策略微调

TMS: Trajectory-Mixed Supervision for Reward-Free, On-Policy SFT

Rana Muhammad Shahroz Khan, Zijie Liu, Zhen Tan, Charles Fleming, Tianlong Chen

AI总结:

TMS通过轨迹混合监督在无奖励的情况下提升大语言模型的保留能力,有效弥补了传统SFT与RL之间的差距。

摘要:

强化学习(RL)和监督微调(SFT)是增强大语言模型(LLM)在下游任务上性能的两种主导范式。虽然RL通常比SFT更好地保留更广泛模型能力(保留),但它伴随着显著的成本:复杂的奖励工程、不稳定性以及昂贵的在线策略采样。相比之下,SFT效率高但脆弱,常常因监督不匹配(Supervision Mismatch)而遭受灾难性遗忘:模型演进策略与静态训练标签之间的分歧。我们通过轨迹混合监督(TMS),一种无奖励框架,来解决这一权衡问题。TMS通过从模型自身的历史检查点创建动态课程来近似RL的在线策略收益。TMS最小化策略-标签分歧(PLD),防止标准SFT中导致遗忘的模式崩溃。在推理(MATH、GSM8K)和指令遵循基准测试中,实验表明TMS有效地将准确性-保留帕累托前沿向前推进。虽然RL仍是保留的黄金标准,但TMS在标准和迭代SFT上显著优于,无需奖励模型或验证器即可缩小与RL的差距。机理分析确认PLD漂移准确预测遗忘,并且TMS成功缓解了这种漂移。

英文摘要:

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.

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

↑