通过反思式纠错提升大模型跨领域推理能力
ROSD: Reflective On-Policy Self-Distillation for Language Model Reasoning across Domains

- 用自我反思定位错误位置,只对出错部分进行精炼
- 在多个领域测试中,推理准确率显著优于传统方法
- 特别适合需要跨领域泛化的复杂逻辑任务
基于策略的自蒸馏(OPSD)通过提供密集的词级别监督来提升大语言模型的推理性能。然而,现有方法在域内推理中提升有限,且跨域泛化能力差。我们发现两个关键原因:以验证解为条件会诱导模仿训练域参考轨迹,而非针对性修正错误;对完整回复进行蒸馏可能覆盖有效推理前缀并加剧过拟合。为此提出反射式基于策略自蒸馏(ROSD),将参考解模仿转化为基于反思的定向纠错。每个推理回溯中,自反思器提取修正思路并定位首个错误段落。修正思路引导教师模型进行精准监督,而局部错误段落限制蒸馏范围,确保仅在需要处修正,同时保留有效推理前缀。在多个域内与跨域推理基准测试中,ROSD整体表现优于标准OPSD,尤其在跨域泛化上显著领先。代码已开源。
原文摘要 · Abstract (English)
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 https://github.com/ZiqiZhao1/ROSD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。