arXiv:2605.12227cs.CL2026-05

融合强化学习与教师引导,提升大模型长文本推理能力。

A Recipe for Long-Context Reasoning in Large Language Models via On-Policy Optimization and Distillation

论文配图:A Recipe for Long-Context Reasoning in Large Language Models via On-Policy Optimization and Distillation
图 1 · 摘自论文原文
  • 用自生成轨迹加教师指导,实现密集词级监督。
  • 在多跳推理任务上比基线提升12.7%准确率。
  • 适合需要长程逻辑推理的智能系统开发。

现有长上下文任务后训练方法存在互补性局限:(i) 监督微调(SFT)虽稳定但有暴露偏差;(ii) 如组相对策略优化(GRPO)等强化学习方法依赖模型生成轨迹,但在长时序任务中面临信用分配困难和奖励稀疏问题;(iii) 在策略蒸馏(OPD)中虽提供密集的逐标记指导,但未直接优化任务奖励。本文研究这些策略的互补性,提出结合GRPO与类OPD教师指导的方案:学生基于自身采样轨迹使用结果级奖励学习,而更强的教师提供密集的词级正则化,替代标准参考策略。该方法在难以获取过程级监督时尤为有效。为支持研究,我们构建了LongBlocks——一个涵盖多跳推理、上下文定位与长文本生成的合成多语言数据集。通过控制消融实验,我们验证了冷启动初始化、教师锚定与数据混合的作用,结果表明该方案在长上下文推理上比GRPO或OPD更稳定高效,同时保持短上下文能力。

原文摘要 · Abstract (English)

Existing approaches to post-train models for long-context tasks face complementary limitations: (i) supervised fine-tuning (SFT) provides stable supervision but suffers from exposure bias; (ii) reinforcement learning methods such as Group Relative Policy Optimization (GRPO) train on model-generated trajectories but struggle with long-horizon credit assignment and sparse rewards; and (iii) on-policy distillation (OPD) provides dense token-level guidance but does not directly optimize task rewards. We study these complementary strategies for long-context alignment and derive a recipe that combines GRPO with OPD-style teacher guidance: the student learns from its own rollouts using outcome-level rewards, while a stronger teacher provides dense token-level regularization in place of the standard reference policy. This is especially useful when process-level supervision is difficult to obtain. To support this study, we introduce LongBlocks, a synthetic multilingual dataset spanning multi-hop reasoning, contextual grounding, and long-form generation. Through controlled ablations, we isolate the roles of cold-start initialization, teacher anchoring, and data mixing, showing that our recipe yields a more stable and effective path to long-context reasoning than GRPO or OPD while preserving short-context capabilities.

长文本推理强化学习知识蒸馏

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