反思式策略内自蒸馏:面向跨领域语言模型推理 / ROSD: Reflective On-Policy Self-Distillation for Language Model Reasoning across Domains
1️⃣ 一句话总结
本文提出了一种名为ROSD的新方法,通过让语言模型在训练时反思自己的错误并只在错误位置进行针对性修正,从而显著提升了模型在熟悉和不熟悉问题上的推理能力。
On-policy self-distillation (OPSD) improves the reasoning performance of large language models (LLMs) by providing dense token-level supervision for on-policy rollouts. However, existing OPSD methods often yield limited gains on in-domain reasoning and generalize poorly to out-of-domain problems. We identify two key causes: conditioning the self-teacher on a verified solution encourages imitation of training-domain reference trajectories rather than error-specific correction, and applying distillation to the full response can overwrite valid reasoning prefixes and reinforce overfitting. We propose Reflective On-policy Self-Distillation (ROSD), a framework that turns reference-solution imitation into targeted reasoning correction through reflection-guided, error-localized distillation. For each rollout, ROSD uses a self-reflector to extract a corrective idea and locate the first erroneous span. The corrective idea guides the self-teacher toward targeted supervision, while the localized error span restricts distillation to where correction is needed. This design corrects flawed reasoning while preserving valid prefixes. Experiments on multiple in-domain and out-of-domain reasoning benchmarks show that ROSD yields stronger in-domain reasoning performance overall and substantially better out-of-domain generalization than standard OPSD. Code is available at this https URL.
反思式策略内自蒸馏:面向跨领域语言模型推理 / ROSD: Reflective On-Policy Self-Distillation for Language Model Reasoning across Domains
本文提出了一种名为ROSD的新方法,通过让语言模型在训练时反思自己的错误并只在错误位置进行针对性修正,从而显著提升了模型在熟悉和不熟悉问题上的推理能力。
源自 arXiv: 2605.28014