arXiv:2607.05339cs.LGcs.AI2026-07

用教师引导探索难例,让模型学会解难题。

TREK: Distill to Explore, Reinforce to Refine

论文配图:TREK: Distill to Explore, Reinforce to Refine
图 1 · 摘自论文原文
  • 先找学生解不出来的难题,再用教师生成候选解
  • 选高概率候选解,用前向KL拉近学生分布
  • 无需教师内部信息,适合各类模型和任务

当当前策略已能采样有效推理轨迹时,组相对策略优化(GRPO)表现良好,但在正确解模式不在学生策略支持范围内的难题上会停滞。本文提出TREK(基于前向KL的教师引导探索),一种分阶段过程:不用于模仿,而是通过蒸馏扩展探索空间。其关键优势在于通用性——仅需验证过的输出轨迹,即可使用外部黑盒教师、白盒教师或同一模型在额外推理上下文下的输出,且可在教师不可见时高效识别最值得巩固的难例。TREK首先识别学生通过率极低的提示,调用候选源生成已验证的解,保留按学生似然排名的前r个解,通过短时前向KL将这些验证解拉入学生支持集,再回归标准的在线策略优化。在数学推理任务中,使用DeepSeek-V4生成的解,TREK使Qwen3系列模型在AIME 2024和AIME 2025上均取得提升;对Qwen3-8B,AIME 2025从36.9提升至40.3,AIME 2024从47.9提升至51.1(avg@16);自上下文版本无外部教师也能达到38.5和49.6。在代理任务中,TREK将ALFWorld成功率从75.8%提升至82.8%,ScienceWorld从12.5%提升至26.7%;尤其在最难任务类型上,TREK早期即达高成功率,而原始GRPO需更多训练步才能接近。

原文摘要 · Abstract (English)

Group Relative Policy Optimization (GRPO) is effective when the current policy already samples useful reasoning trajectories, but it stalls on hard prompts whose correct solution modes lie outside the student's on-policy support. We propose TREK (Teacher-Routed Exploration via Forward KL), a simple staged procedure that uses distillation not for imitation but for exploration support expansion. A key advantage of TREK is its generality: because it only consumes verified output trajectories, it can use an external black-box teacher, a white-box teacher, or the same model given additional inference-time context, and it can efficiently identify which hard-prompt samples are most worth consolidating even when teacher internals are unavailable. TREK first identifies prompts where the unaided student has very low pass rate, queries a proposal source to produce verified candidate solutions, keeps the top-$r$ proposals ranked by current student likelihood, applies a short forward-KL phase to pull those verified modes into the student's support, and then returns to standard on-policy GRPO refinement. On mathematical reasoning, TREK with DeepSeek-V4 proposals improves Qwen3 models across all tested scales on AIME 2024 and AIME 2025; for Qwen3-8B, it improves AIME 2025 from 36.9 to 40.3 and AIME 2024 from 47.9 to 51.1 (avg@16), while the self-context variant reaches 38.5 and 49.6 without an external teacher. On agentic tasks, TREK raises ALFWorld success rate from 75.8 to 82.8 and ScienceWorld success rate from 12.5 to 26.7; notably, on the hardest task types, TREK achieves high success rates early in training while unaided GRPO requires substantially more optimization steps to reach comparable levels.

强化学习模型蒸馏推理增强多模态

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