arXiv:2606.09887cs.LGcs.AI2026-06被引 1

让大模型在错误时获得老师指点,且帮了忙也得扣分,更靠谱地学推理。

SocraticPO: Policy Optimization via Interactive Guidance

  • 用对话式指导代替简单对错反馈,引导模型修正错误思路
  • 在科学推理任务上超越强基线方法,正确率提升显著
  • 适合需要严谨推理的AI教育、自动解题系统应用

大规模语言模型的强化学习通常依赖二值正确性等标量奖励信号进行推理监督。这类奖励仅提供优化方向,却很少说明如何修正错误推理,容易导致模型走捷径、策略脆弱。我们提出SocraticPO(苏格拉底式策略优化),在强化学习采样过程中引入苏格拉底式的自然语言指导:学生先独立作答,若错误,教师诊断并给出简洁纠正指引,学生在此基础上继续推理。关键在于,经教师干预后正确的答案仅获衰减奖励,防止模型将教师帮助视为免费得分路径。由于SocraticPO仅改写采样过程而保持原有期望奖励目标不变,可无缝接入Reinforce++等标准策略梯度框架。同时,因教师仅输出文本级指导,无需访问概率分布或对齐,可使用更强的黑箱教师模型。在SciKnowEval的本科级科学推理基准测试中,SocraticPO优于强基线的强化学习与自蒸馏方法。消融实验表明,针对性指导与奖励衰减均必不可少,后者有效抑制对辅助纠错的依赖。

原文摘要 · Abstract (English)

Reinforcement learning (RL) for large language models usually supervises reasoning with scalar outcome rewards, such as binary correctness. Such rewards provide an optimization direction but rarely explain how a model should revise its mistaken reasoning, which can encourage shortcut learning and brittle policies. We propose \textbf{SocraticPO} (Socratic Policy Optimization), a policy-optimization framework that augments RL rollouts with Socratic-style natural-language guidance. During rollout, the student first answers independently; if the answer is incorrect, a teacher diagnoses the attempt and provides concise corrective guidance, after which the student continues under the expanded context. Crucially, this guidance is paired with reward decay: correct answers obtained after teacher intervention only receive decayed rewards, preventing the policy from treating teacher help as a free path to reward. Since SocraticPO only modifies the rollout process while leaving the standard expected-reward objective intact, it can be plugged into existing policy-gradient backends such as Reinforce++. Moreover, because the teacher provides only text-level guidance, SocraticPO can leverage stronger black-box teacher models without requiring access to logits or distribution matching. On undergraduate-level scientific reasoning benchmarks from SciKnowEval, SocraticPO improves over strong RL and self-distillation baselines. Ablations show that both targeted guidance and reward decay are necessary, with reward decay mitigating reliance on assisted correction.

强化学习推理优化教学交互大模型

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