SFT其实能通用,只要训练时加点提示多样性。
Debunk the Myth of SFT Generalization
- 用多样化提示训练SFT,打破固定模板依赖。
- 在新指令和更难任务上表现接近甚至超过RL基线。
- 适合想用简单稳定方法实现强泛化的研究者。
主流观点认为监督微调(SFT)会记忆训练数据而无法泛化,而强化学习(RL)更具鲁棒性。我们通过在两个决策基准(Sokoban和General Points)上的系统评估重新审视这一观点。发现SFT泛化不佳很大程度源于冻结提示的缺陷:当使用固定指令模板训练时,SFT模型会固守训练语义而非适应新指令。在训练中引入提示多样性可打破此捷径,使模型在未见指令变体上实现强泛化,同时保持分布内性能。此外,链式思维(CoT)监督提供算法框架,显著提升对更复杂任务的迁移能力,如更大规模的Sokoban网格、包含外分布数值的算术题以及五张牌组合带来的组合复杂度增加。将提示多样性与CoT结合,实现指令变体与难度变体双重泛化最优,其表现匹配或超越RL基线,同时保留SFT的简洁性和稳定性。这些结果挑战了SFT不如RL的叙事,支持以数据为中心的视角:经过恰当设计的演示数据,基础SFT也能达到与RL相当的泛化能力。代码见:https://github.com/XiaofengLin7/debunking-sft-generalization。
原文摘要 · Abstract (English)
A prevailing view holds that supervised fine-tuning (SFT) memorizes training data and fails to generalize, whereas reinforcement learning (RL) attains broader robustness. We revisit this claim through a systematic evaluation on two decision-making benchmarks, Sokoban and General Points, and arrive at a different conclusion. We show that much of SFT's perceived failure stems from frozen-prompt artifacts: when trained on fixed instruction templates, SFT models cling to training semantics rather than adapting to new ones. Introducing prompt diversity during training breaks this shortcut and yields strong generalization to unseen instruction variants without harming in-distribution performance. Beyond instruction shifts, we ask whether SFT can generalize to strictly harder tasks. Here, chain-of-thought (CoT) supervision provides an algorithmic scaffold that markedly improves transfer to more difficult regimes, such as larger Sokoban grids with additional boxes and arithmetic with out-of-distribution values or five-card compositions that increase combinatorial complexity. Finally, combining prompt diversity with CoT achieves the best of both worlds: robust generalization across both instruction-variant and difficulty-variant settings, matching or surpassing RL baselines on our benchmarks while retaining SFT's simplicity and stability. These findings challenge the narrative that SFT is inherently inferior to RL and support a data-centric perspective: with appropriately curated demonstrations, vanilla SFT can generalize as strongly as RL. Code reproducing the results in the paper can be found at: https://github.com/XiaofengLin7/debunking-sft-generalization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。