通过动态优化教师模型,让AI从语言反馈中持续学习复杂推理任务。
Learning from Language Feedback via Variational Policy Distillation
- 提出变分策略蒸馏框架,让教师和学生策略共同进化。
- 在科学推理与代码生成任务上,性能优于传统强化学习与自蒸馏方法。
- 适合需要语言反馈的复杂任务,如数学推理和冷启动场景。
基于可验证奖励的强化学习(RLVR)在复杂推理任务中面临稀疏结果信号的问题,导致探索困难。现有基于策略的自蒸馏方法虽利用语言反馈生成密集的逐标记监督,但依赖固定且被动的教师模型进行反馈解析。随着学生策略提升,教师的零样本评估能力趋于饱和,阻碍进一步学习。为此,本文提出变分策略蒸馏(VPD),将语言反馈学习形式化为变分期望-最大化(EM)问题。在E步中,教师通过自适应信任区域更新主动优化轨迹结果,将文本反馈转化为动态改进的目标标记分布;在M步中,学生在自身策略回放中内化该分布引导。通过持续提升教师提取可操作信号的能力,克服了静态蒸馏的局限。在科学推理与代码生成任务的多种诊断性反馈源上测试,VPD均显著优于标准RLVR及现有自蒸馏基线。最后,在严格数学推理与冷启动条件下验证框架边界,揭示反馈驱动自蒸馏相较于纯环境驱动强化学习的根本限制。
原文摘要 · Abstract (English)
Reinforcement learning from verifiable rewards (RLVR) suffers from sparse outcome signals, creating severe exploration bottlenecks on complex reasoning tasks. Recent on-policy self-distillation methods attempt to address this by utilizing language feedback to generate dense, token-level supervision. However, these approaches rely on a fixed, passive teacher to interpret the feedback. As the student policy improves, the teacher's zero-shot assessment capabilities plateau, ultimately halting further learning. To overcome this, we propose Variational Policy Distillation (VPD), a framework that formalizes learning from language feedback as a Variational Expectation-Maximization (EM) problem. VPD co-evolves both policies: in the E-step, the teacher is actively refined on trajectory outcomes via an adaptive trust-region update, translating textual feedback into a dynamically improved target token distribution. In the M-step, the student internalizes this dense distributional guidance on its own on-policy rollouts. By continuously improving the teacher's ability to extract actionable signals from textual critique, VPD overcomes the limitations of passive distillation. Evaluated across diverse sources of diagnostic feedback on scientific reasoning and code generation tasks, VPD consistently outperforms both standard RLVR and existing self-distillation baselines. Finally, by stress-testing our framework on rigid mathematical reasoning and cold-start regimes, we illuminate the fundamental bounds of feedback-driven self-distillation compared to pure environment-driven RL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。