让大模型在微调中保持探索能力,提升推理效果
Entropy-Preserving Supervised Fine-Tuning via Adaptive Self-Distillation for Large Reasoning Models
- 用自生成教师指导模型,动态调节温度增强探索
- 在数学推理任务上,SFT阶段提升2.5~2.9分,RL阶段平均增5.0分
- 适合需要强推理与多样性的大模型优化场景
大型推理模型的标准后训练流程(SFT-then-RL)可能限制强化学习阶段的收益:监督微调(SFT)虽模仿专家示范,但常导致模型过自信、降低生成多样性,使强化学习面临狭窄的探索空间。在SFT中加入熵正则化并非万能,往往使词分布趋于均匀,增加熵却不提升有意义的探索能力。本文提出CurioSFT,一种熵保持型监督微调方法,通过内在好奇心增强探索能力。其包含:(a) 自探索蒸馏,将模型向自生成的温度调整教师蒸馏,以激发能力范围内的探索;(b) 基于熵的温度自适应选择,在推理标记处增强探索,同时稳定事实性标记,缓解知识遗忘。在数学推理任务上的大量实验表明,相比原始SFT,CurioSFT在分布内任务上提升2.5分,分布外任务上提升2.9分;更重要的是,保留的探索能力成功转化为强化学习阶段的收益,平均提升5.0分。
原文摘要 · Abstract (English)
The standard post-training recipe for large reasoning models, supervised fine-tuning followed by reinforcement learning (SFT-then-RL), may limit the benefits of the RL stage: while SFT imitates expert demonstrations, it often causes overconfidence and reduces generation diversity, leaving RL with a narrowed solution space to explore. Adding entropy regularization during SFT is not a cure-all; it tends to flatten token distributions toward uniformity, increasing entropy without improving meaningful exploration capability. In this paper, we propose CurioSFT, an entropy-preserving SFT method designed to enhance exploration capabilities through intrinsic curiosity. It consists of (a) Self-Exploratory Distillation, which distills the model toward a self-generated, temperature-scaled teacher to encourage exploration within its capability; and (b) Entropy-Guided Temperature Selection, which adaptively adjusts distillation strength to mitigate knowledge forgetting by amplifying exploration at reasoning tokens while stabilizing factual tokens. Extensive experiments on mathematical reasoning tasks demonstrate that, in SFT stage, CurioSFT outperforms the vanilla SFT by 2.5 points on in-distribution tasks and 2.9 points on out-of-distribution tasks. We also verify that exploration capabilities preserved during SFT successfully translate into concrete gains in RL stage, yielding an average improvement of 5.0 points.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。