强化学习解决长上下文推理问题:通义推出QwenLong-L1-32B

  新万博官网     |      2025-06-10 12:24

  上下文长度达 13 万 token,适用于多段文档综合分析、金融、法律、科研等复杂领域任务。

  近期的推理大模型(LRMs)通过强化学习(RL)展现出强大的推理能力,但这些改进主要体现在短上下文推理任务中。相比之下,如何通过强化学习扩展 LRMs 以有效处理和推理长上下文输入,仍然是一个尚未解决的关键挑战。

  来自阿里巴巴通义实验室的团队首先形式化定义长上下文推理强化学习范式,并识别出其中的两个核心挑战:次优的训练效率与不稳定的优化过程

  区别于短上下文推理强化学习促进模型利用内部知识推理,长上下文推理强化学习需要模型首先定位外部关键信息然后整合内部推理

  长上下文推理强化学习训练效率低,具体表现在(a)奖励收敛较慢,(b)模型输出熵的显著降低,限制了优化过程中的探索行为。同时,长上下文推理强化学习训练不稳定,具体表现为(c)KL 散度突刺较多,这是由于(d)较长的输出长度和不均匀的输入长度导致方差变大,导致策略更新不稳定。

  基于渐进式上下文扩展技术混合奖励机制,QwenLong-L1 通过强化学习实现了从短文本到长文本的稳定上下文适应。

  基于传统的短上下文推理强化学习框架,QwenLong-L1 主要提出如下改进:渐进式上下文扩展技术和混合奖励机制。

  训练长上下文推理大模型存在不稳定的优化动态特性。为解决这些问题,我们提出了一种渐进式上下文扩展框架,该框架包含:课程引导的分阶段强化学习策略以稳定从短到长上下文的优化过程;难度感知的回顾采样机制,优先探索复杂实例;以及稳定的监督微调预热阶段,在强化学习训练前提供稳健的初始化基础。

  稳健的监督微调预热:使用蒸馏的长上下文推理数据在强化学习前监督微调模型,获取稳定的初始策略,降低训练过程中的不稳定。

  课程引导的分阶段强化学习:将强化学习训练分为两阶段,阶段 I 输入长度 20K,阶段 II 扩展至 60K,逐步适应长上下文。每阶段仅训练当前长度区间的样本,避免混合长度导致的优化冲突。

  难度感知的回顾采样:根据样本平均奖励动态计算难度,低奖励样本(高难度)被优先保留至后续阶段。阶段 II 训练时,包含阶段 I 的高难度样本,强制模型持续探索复杂案例。

  在数学、编程和逻辑推理等短上下文推理任务中,先前的研究工作通常采用基于规则的奖励函数。然而,开放域问答等长上下文推理任务因其固有的答案多样性带来了独特挑战。在这种情境下,限制性过强的基于规则的奖励机制可能会制约有效答案的多样性,从而可能影响整体性能。针对这些局限性,我们提出一种融合规则验证模型评判的混合奖励机制,通过互补性评估实现精确率与召回率的平衡。

  规则奖励:通过正则表达式从模型输出中提取答案,与标准答案严格匹配。确保答案格式正确性,防止 Reward Hacking。

  模型评判:训练过程采用 Qwen2.5-1.5B-Instruct 作为轻量级评判模型,评估预测答案和标准答案之间语义等价性。

  SFT 和 RL 发挥着互补作用,SFT 较低代价到可接受性能,而 RL 对达到最佳结果至关重要;要实现最优性能,必须优先考虑 RL 而不是 SFT,因为过度关注 SFT 可能使模型陷入局部最优,从而限制 RL 提升;

  所有模型都表现出明显的各类推理模式,且长上下文相关的 Grounding 出现频率最高RL 自然地使这些推理模式出现频率越来越高,最终性能也会随之增长SFT 尽管让推理模式取得了远高于 RL 的增加,但转换成的性能提高相较于 RL 有限

  这项研究通过强化学习探索了长上下文推理大模型的开发。其首先提出长上下文推理强化学习范式,并发现次优的训练效率和不稳定的优化过程等关键问题。

  我们的分析揭示了长上下文推理强化学习的三项关键洞察:渐进式上下文扩展对实现稳定适应的重要作用、优先强化学习对最优性能的必要性,以及强化学习训练过程中长文本推理模式的增加对性能提升的促进作用。