arXiv:2602.13407cs.AI2026-02被引 5

用简单监督微调让大模型推理更短更准,还省资源。

On-Policy Supervised Fine-Tuning for Efficient Reasoning

  • 直接用自生成数据做监督微调,去掉复杂奖励和正则化。
  • 推理长度最多缩短80%,准确率不变,五项基准全超越旧方法。
  • 训练快70%、显存减半,适合追求效率的部署场景。

大型推理模型(LRMs)通常通过强化学习(RL)训练以探索长链思维推理,但计算成本高昂。近期方法引入多奖励目标以同时优化正确性和简洁性,但复杂设计常导致训练不稳定且权衡效果不佳。本文重新审视该目标,指出根本性错配:当正确性和长度可直接验证时,KL正则化失去作用;多重奖励信号下分组归一化变得模糊。通过移除这两项并简化奖励为截断式长度惩罚,优化问题退化为对自生成数据的监督微调,过滤出正确且简洁的样本。我们称此策略为“在策略监督微调”(on-policy SFT)。尽管结构简单,其始终定义了准确率-效率的帕累托前沿,在五项基准上将思维链长度减少高达80%且保持原准确率,优于更复杂的基于RL的方法。此外,训练效率显著提升,显存使用减少50%,收敛速度加快70%。代码已开源。

原文摘要 · Abstract (English)

Large reasoning models (LRMs) are commonly trained with reinforcement learning (RL) to explore long chain-of-thought reasoning, achieving strong performance at high computational cost. Recent methods add multi-reward objectives to jointly optimize correctness and brevity, but these complex extensions often destabilize training and yield suboptimal trade-offs. We revisit this objective and challenge the necessity of such complexity. Through principled analysis, we identify fundamental misalignments in this paradigm: KL regularization loses its intended role when correctness and length are directly verifiable, and group-wise normalization becomes ambiguous under multiple reward signals. By removing these two items and simplifying the reward to a truncation-based length penalty, we show that the optimization problem reduces to supervised fine-tuning on self-generated data filtered for both correctness and conciseness. We term this simplified training strategy on-policy SFT. Despite its simplicity, on-policy SFT consistently defines the accuracy-efficiency Pareto frontier. It reduces CoT length by up to 80 while maintaining original accuracy, surpassing more complex RL-based methods across five benchmarks. Furthermore, it significantly enhances training efficiency, reducing GPU memory usage by 50% and accelerating convergence by 70%. Our code is available at https://github.com/EIT-NLP/On-Policy-SFT.

推理优化监督微调效率提升

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