arXiv:2604.12002cs.CL2026-04被引 44

用自修正机制把稀疏奖励转成密集监督,提升模型推理能力

Self-Distillation Zero: Self-Revision Turns Binary Rewards into Dense Supervision

论文配图:Self-Distillation Zero: Self-Revision Turns Binary Rewards into Dense Supervision
图 1 · 摘自论文原文
  • 让模型自扮演生成者和修订者,用自身输出反馈迭代优化
  • 在数学代码任务上比基础模型提升超10%,且仅需少量训练样本
  • 无需外部教师或高质量数据,适合资源有限的场景

当前可验证场景下的后训练方法分为两类:强化学习(RLVR)依赖二值奖励,虽通用但监督稀疏;知识蒸馏提供密集的逐标记监督,但通常需外部教师或高质量示范,成本高或不可得。本文提出Self-Distillation Zero(SD-Zero),仅用单个模型即可实现高效训练,无需外部教师或优质示范。该方法让模型同时扮演生成器(生成初始回答)与修订者(基于生成结果和二值奖励生成改进回答)。通过在线自蒸馏,将修订者的词分布作为监督信号,指导生成器优化。本质上,实现了将二值奖励转化为密集的自监督信号。在使用Qwen3-4B-Instruct和Olmo-3-7B-Instruct的数学与代码推理基准测试中,SD-Zero相较基线模型性能提升至少10%,并优于强基线如RFT、GRPO和SDFT,且在相同问题集与训练样本预算下表现更优。大量消融实验揭示两个新特性:(a) 逐标记自定位,即修订者可根据奖励识别生成结果中需修改的关键标记;(b) 迭代自演化,修订能力可定期回传至生成性能,实现持续进化。

原文摘要 · Abstract (English)

Current post-training methods in verifiable settings fall into two categories. Reinforcement learning (RLVR) relies on binary rewards, which are broadly applicable and powerful, but provide only sparse supervision during training. Distillation provides dense token-level supervision, typically obtained from an external teacher or using high-quality demonstrations. Collecting such supervision can be costly or unavailable. We propose Self-Distillation Zero (SD-Zero), a method that is substantially more training sample-efficient than RL and does not require an external teacher or high-quality demonstrations. SD-Zero trains a single model to play two roles: a Generator, which produces an initial response, and a Reviser, which conditions on that response and its binary reward to produce an improved response. We then perform on-policy self-distillation to distill the reviser into the generator, using the reviser's token distributions conditioned on the generator's response and its reward as supervision. In effect, SD-Zero trains the model to transform binary rewards into dense token-level self-supervision. On math and code reasoning benchmarks with Qwen3-4B-Instruct and Olmo-3-7B-Instruct, SD-Zero improves performance by at least 10% over the base models and outperforms strong baselines, including Rejection Fine-Tuning (RFT), GRPO, and Self-Distillation Fine-Tuning (SDFT), under the same question set and training sample budget. Extensive ablation studies show two novel characteristics of our proposed algorithm: (a) token-level self-localization, where the reviser can identify the key tokens that need to be revised in the generator's response based on reward, and (b) iterative self-evolution, where the improving ability to revise answers can be distilled back into generation performance with regular teacher synchronization. Code: https://github.com/princeton-pli/Self-Distillation-Zero.

自蒸馏强化学习推理增强

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