arXiv:2507.19766cs.CLcs.AI2025-07被引 11

通过分段强化学习,显著提升大模型长序列推理能力。

UloRL:An Ultra-Long Output Reinforcement Learning Approach for Advancing Large Language Models' Reasoning Abilities

  • 将超长输出分解为短片段,缓解训练延迟问题。
  • 采用动态掩码机制,避免熵崩溃,提升训练稳定性。
  • 在30B模型上实现85.1%的竞赛准确率,超越更大模型。

近期大语言模型的发展表明,基于可验证奖励的强化学习(RLVR)可通过延长输出序列来增强推理能力。然而,传统强化学习框架在处理超长输出时存在效率低下问题,主要源于长尾序列分布和训练中的熵崩溃现象。为此,我们提出一种超长输出强化学习(UloRL)方法,以推进大语言模型的推理能力。具体而言,我们将超长输出解码划分为短片段,从而缓解由长尾样本导致的延迟;同时引入对已掌握正向标记(MPTs)的动态掩码机制,防止熵崩溃。实验结果表明,该方法效果显著:在Qwen3-30B-A3B模型上,分段滚动生成使训练速度提升2.06倍;使用128k-token输出的强化训练使模型在AIME2025上的表现从70.9%提升至85.1%,在BeyondAIME上从50.7%提升至61.9%,甚至超过参数量更大的Qwen3-235B-A22B模型,取得显著进步。这些发现凸显了该方法在实现超长序列生成与推理能力提升方面的潜力。代码与模型将向社区开源。

原文摘要 · Abstract (English)

Recent advances in large language models (LLMs) have highlighted the potential of reinforcement learning with verifiable rewards (RLVR) to enhance reasoning capabilities through extended output sequences. However, traditional RL frameworks face inefficiencies when handling ultra-long outputs due to long-tail sequence distributions and entropy collapse during training. To address these challenges, we propose an Ultra-Long Output Reinforcement Learning (UloRL) approach for advancing large language models' reasoning abilities. Specifically, we divide ultra long output decoding into short segments, enabling efficient training by mitigating delays caused by long-tail samples. Additionally, we introduce dynamic masking of well-Mastered Positive Tokens (MPTs) to prevent entropy collapse. Experimental results demonstrate the effectiveness of our approach. On the Qwen3-30B-A3B model, RL with segment rollout achieved 2.06x increase in training speed, while RL training with 128k-token outputs improves the model's performance on AIME2025 from 70.9\% to 85.1\% and on BeyondAIME from 50.7\% to 61.9\%, even surpassing Qwen3-235B-A22B with remarkable gains. These findings underscore the potential of our methods to advance the reasoning capabilities of LLMs with ultra-long sequence generation. We will release our code and model for further use by the community.

强化学习长序列生成推理能力大模型

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