MarkTechPost更新于

阿里巴巴Qwen推出QwenGyre:面向XLong-Horizon智能体的弹性强化学习RL

阿里Qwen发布QwenGyre弹性强化学习框架,用于训练跨越数小时、处理近1M token的超长时程编程智能体。对比Colocate最高提速1.85倍,NL2RepoBench从52.5%升至58.5%,但无公开代码。

阿里巴巴的Qwen团队推出了 QwenGyre,这是一个用于极长时程智能体训练的弹性强化学习RL框架。QwenGyre是阿里巴巴的一个在线强化学习系统,通过未经修改的黑盒harness训练编码智能体。它针对运行数小时并处理接近1M token的rollout。 QwenGyre弹性RL框架 被用于Qwen的旗舰Qwen 3.8 2.4T模型。它基于2个核心思想。它在rollout和训练之间调度空闲的GPU。它还将分支的智能体历史转化为有界的训练样本。

TL;DR

  • 规模: 这是一个训练框架,而非模型。在Qwen 3.6 122B和Qwen 3.8 2.4T(每个Qwen为A95B)上测试,每次rollout约700K token。
  • 运行环境: 多节点GPU集群。122B的运行使用了24到32个节点;2.4T的运行使用了192个节点(4个cell,每个48)。GPU型号未披露。Megatron负责训练,SGLang服务rollout。
  • 性能: 在相同分数下,比Async快1.38倍到1.78倍,比Colocate快1.21倍到1.85倍。
  • 最佳: 在TerminalBench上比Colocate端到端提速1.85倍。
  • 最差: 在2.4T模型上仅比Colocate快1.21倍,其中许多查询接近超时运行。
  • 结论:
    • 最佳:NL2RepoBench在48步内从52.5%提升到58.5%。
    • 最差:论文未提供公开的代码发布链接。

QwenGyre解决了什么问题?

研究团队将“XLong-horizon”任务定义为跨越数小时并包含数百次模型和工具交互的智能体运行。例如根据规格说明生成完整代码库,如 NL2Repo-Bench.

在这个规模下出现了两个问题:

  • 首先,rollout存在长尾现象。在Colocate设置中,训练需要等待最慢的执行,同时GPU处于空闲状态。在Async设置中,固定的GPU池无法相互借出闲置算力。
  • 其次,像 Claude Code 这样的harness会压缩上下文、派生子智能体并进行重试。一次执行会变成一棵分支树,而不是单一轨迹。展开每条路径会重复共享前缀并推高训练成本。

Elastic 调度器是如何工作的?

QwenGyre 将 GPU 划分为共享同一训练布局的单元(cell)。每个单元既可以提供 rollout 服务,也可以承载完整的训练副本。核心单元拥有权威权重和优化器状态。卫星单元随后加入训练。

调度器跟踪一个推广水位:已排队和正在运行的执行。随着执行完成,一旦剩余的单元仍能服务所有实时工作,单元便切换到训练状态。模型调用通过代理重新路由,因此测试框架持续运行。

关键机制:

  • 在重路由过程中,KV-cache块通过RDMA迁移,从而减少前缀重计算。
  • 卫星通过 Mooncake 的 Transfer Engine 拉取核心权重并加入正在进行的批次。
  • 流式训练让单元格能够在数据就绪时申领微步骤。
  • 平均角色切换耗时 8.52 秒(rollout 到 training)和 3.46 秒(training 到 rollout)。

Trajectory Processor 如何构建训练样本?

一个代理记录精确的 token ID 和行为对数概率,使用 token-in, token-out(TITO). 调用以共享前缀的轨迹树形式存储.

  • 部分评分: 超时的执行仍会基于其保留的工作区进行评分。评估器失败会被排除,绝不会被视为零奖励。
  • 有界采样: 每个执行最多保留 5 条轨迹,排序为 main > main-summary > sub-agent > sub-agent-summary。
  • 去重: 共享目标只计一次,且只有策略生成的 token 会被训练。
  • 公平加权: 损失在每个执行内部取平均,因此额外的分支不会增加其权重。

上限设为5的方案在匹配未设上限训练得分的同时,将前向-反向传播时间缩短了25.2%。训练使用改进版 GSPO ,每组16次rollout。

QwenGyre有多快?基准测试结果

所有运行均使用Claude Code 2.1.220作为测试框架,GPU预算相同且陈旧度匹配。

工作负载(模型)对比Async对比Colocate
NL2RepoBench(Qwen 3.6 122B,48步)1.42倍至1.53倍1.36倍至1.47倍
DeepSWE(Qwen 3.6 122B,24步)1.569倍至1.572倍1.610倍至1.824倍
TerminalBench(Qwen 3.6 122B,24步)1.383倍至1.430倍1.809倍至1.849倍
NL2RepoBench(Qwen 3.8 2.4T,48步)1.78倍1.21倍

在2.4T模型上,QwenGyre用75.42小时完成了48步。Async耗时134.55小时,Colocate耗时91.47小时。评估通过率从52.48%提升至58.54%。

QwenGyre与其他智能体RL框架相比如何?

特性QwenGyreveRLAReaLAgent Lightning
开发者阿里巴巴Qwen字节跳动Seed + 社区清华大学IIIS + 蚂蚁集团微软
许可证未公开Apache-2.0Apache-2.0MIT
GPU放置弹性单元,可在滚动更新中途切换灵活的设备映射,HybridEngine重分片完全异步未公开
黑盒评测框架是,通过代理未公开是,通过替换base_url是,几乎零代码改动
训练 / 回滚(rollout)Megatron / SGLangFSDP, Megatron / vLLM, SGLangMegatron, FSDP, Archon / vLLM, SGLang未公开
公开宣称的最大规模Qwen 3.8 2.4T671B模型未公开未公开
标称加速比相比Colocate最高可达1.85倍未公开相比同步模式为2.77倍(v0.3)未公开
公开代码未公开是是是的

加速来自各项目自身的配置,不能直接进行比较。

关键要点

  • QwenGyre 在 rollout 和训练之间迁移 GPU,而无需停止在线 agent。
  • 轨迹树将每次执行的样本数上限设为 5,并且只对共享前缀训练一次。
  • 超时的运行仍会根据其已完成的部分工作获得评分。
  • 比 Colocate 快最高 1.85 倍,比 Async 快 1.78 倍。
  • Qwen 3.8 2.4T 在 NL2RepoBench 上的成绩在 48 步内从 52.5% 提升到 58.5%。


请查看 完整 的 arXiv 上的 QwenGyre 论文 以及 Qwen 的 发布公告。所有功劳归于该项目的的研究者。另外,欢迎在 Twitter 上关注我们,别忘了加入我们的 150k+ML SubReddit 并订阅 我们的新闻通讯。等等!你用 telegram 吗? 现在你也可以在 telegram 上加入我们了。

这篇文章 Alibaba Qwen 推出 QwenGyre:面向超长程智能体(XLong-Horizon Agents)的弹性强化学习(RL) 首发于 MarkTechPost.

原始出处

MarkTechPost

内容说明

原始发布及相关权利归来源方。

机器翻译 · 请以原文为准