位置:首页 > 产业资讯 > 清华团队单rollout异步优化突破强化学习千步稳定训练

清华团队单rollout异步优化突破强化学习千步稳定训练

时间:2026-07-19  |  作者:云端旅人  |  阅读:0

如何提升强化学习(RL)的效率,在长程Agent和coding任务中,依然是绕不开的核心难题。

目前,主流的大模型强化学习(LLM RL)流水线大多采用同步、批量式的处理流程。简单来说,就是得等一整批数据都生成好了,才能进行模型更新。这种模式效率提升自然受限。

虽然异步RL能缓解同步等待的问题,但现有方法更多是盯着系统吞吐量,对训练稳定性和最终任务效果关注不够。另外,像GRPO这类方法依赖组式采样,也很难直接套用到异步Agent训练上。

针对这个痛点,清华大学KEG实验室的研究团队提出了一种新方法——单rollout异步优化(Single-rollout Asynchronous Optimization,SAO)。核心思路很直接:用单个rollout采样替代GRPO的组式采样。也就是说,每个输入任务只生成一条rollout,生成完立刻就能用于训练,从而降低离策略影响,提升模型的泛化能力。

研究结果显示,SAO能稳定训练到1000步。在Agent编码数学推理基准上,表现持续优于GRPO及其各种变体。同时,这种单rollout的RL方法在模拟在线学习任务中,也展现出了对动态环境的适应能力。

这项工作为高效RL训练提供了一条可行路径,并已部署到GLM-5.2(750B-A40B)的Agent RL训练流水线中。

图|SAO 在推理和代码基准上的表现。

SAO是如何稳定异步训练的?

整体框架上,SAO采用单 rollout 异步优化:每个输入任务只生成一条训练 rollout,完成后立即进入训练。为了让这个流程平稳运行,SAO设计了几个关键机制:直接双边重要性采样、价值模型强化,以及多轮Agent轨迹的专门处理,同时扩大了价值模型预训练数据规模。

具体来看:

  • 1.直接双边重要性采样(DIS):主要用来减少异步训练中的离策略偏差。由于rollout生成和模型训练不同步,SAO不需要保存大量历史模型,而是直接利用rollout阶段记录的token对数概率来计算重要性采样比。如果某个token的比率超出设定区间,就不参与梯度更新,相当于过滤掉那些“跑偏”的信号。
  • 2.提高价值模型更新频率:目的是降低单 rollout 优化中的梯度估计方差。单rollout训练的梯度方差天然更高,因此需要更可靠的价值模型来估计优势信号。SAO把策略模型和价值模型的更新频率解耦了,策略模型每更新一次,价值模型就更新两次,以此来压低训练方差。
  • 3.固定注意力模块参数:负责稳定价值模型训练。研究团队发现,价值模型的波动主要来自全注意力层,而混合专家层(MoE)相对稳定。所以,SAO在训练价值模型时,固定了注意力模块的参数,只优化MoE投影层,从而减少价值模型训练中的波动。
  • 4.跳过观察信息的 token 级 GAE:用于减少多轮 Agent 轨迹中的环境反馈噪声。在多轮轨迹中,模型动作和环境反馈会交替出现。如果环境观察信息直接参与优势计算,就会引入噪声。因此,SAO会跳过环境反馈的观察信息token,只在模型生成的动作之间传播优势信号。

图|带有单 rollout 设计的 SAO 概览。

SAO的实验结果如何?

实验结果显示,SAO在数学推理代码任务模拟在线学习中都有更好表现。在多个基准上,它超过了对应基线模型和GRPO系列方法,并且能在较长训练过程中保持稳定。具体结果如下:

数学推理和代码任务

SAO的评估覆盖了带Python工具的奥赛数学推理复杂数学问答真实代码修复等场景。结果显示,SAO在这些任务中都超过了对应基线模型和GRPO方法。在AIME2025上,SAO达到了97.3%的准确率,而GRPO只有84.2%

图|数学推理基准上的实验结果(准确率%)。

代码任务SWE-Bench Verified上,SAO的准确率达到29.8%,也高于基线模型的23.0%和GRPO(w/ DIS)的27.0%

图|SWE-Bench Verified 上的实验结果。

训练稳定性

训练曲线看,SAO在不同基准上基本保持领先训练过程也更稳定。普通GRPO较早出现性能崩溃;加入DIS后,GRPO可以稳定训练,这说明直接双边重要性采样能有效过滤偏离过大的token更新。相比之下,SAO在保持稳定的同时,训练中后期评估表现还在继续提升。

图|SAO 与 GRPO(w/ DIS)在训练过程中的性能比较。

训练动态

实验结果揭示,SAO的稳定性与其框架设计密切相关。更频繁的价值模型更新,让价值估计与真实回报更一致;固定注意力模块参数后,价值模型训练更平稳;而DIS则通过过滤偏离过大的token更新,控制了离策略影响。相比之下,基于价值模型的VAPO对照方法,如果不使用DIS,大约在90 step时就会出现训练崩溃。

图|异步单 rollout 强化学习的训练动态。

消融实验

研究团队进一步检验了各项框架设计的作用。他们分别调整了SAO的关键组件:减少价值模型更新次数、改用全参数价值模型训练,并加入了未使用DIS的VAPO对照和滑动平均基线。结果显示,这些设置的表现都低于完整SAO。在BeyondAIME上,完整SAO的准确率为74.8%,而去掉更快价值更新后,准确率降至69.8%。

图|价值模型训练策略和 critic 更新频率的消融结果。

在线学习模拟

研究团队通过模拟在线写作任务,测试了SAO在非平稳环境中的适应能力。任务中,奖励偏好会在可爱风、中二风和古典风之间切换。结果显示,SAO能在奖励偏好切换后快速对齐新的目标风格,并在训练奖励上保持更高水平。相比之下,滑动平均基线存在明显的适应滞后,收敛水平也更低

图|变化写作风格偏好下的在线学习模拟。

不足和未来方向

研究团队指出,尽管SAO在多类Agent任务中表现稳定,它仍存在适用范围、部署基础设施和真实在线应用方面的限制。若要扩展到更广泛的场景,还需要解决以下几个问题:

  • 1.验证更广泛的适用性:当前实验主要集中在基于Qwen3-30B-A3B的大规模Agent推理、代码任务和模拟在线写作任务上。未来,仍需验证SAO是否能迁移到更小模型非Agent类型的RLHF场景,以及奖励更密集rollout更短的任务环境中。
  • 2.完善部署基础设施:SAO依赖训练好的价值模型rollout对数概率。未来若要实际部署,需要训练基础设施在异步生成过程中,能可靠地保存这些概率信息。
  • 3.完善真实在线适应的安全保障:在线学习实验仍停留在受控的模拟环境。研究团队指出,未来若要用于真实用户场景,还需要更强的安全防护监控机制隐私审查
  • 4.加强发布与监控机制:研究团队指出,如果系统缺少适当的数据过滤访问控制评估机制,SAO相关能力可能被用于优化有害目标。未来,基于SAO的系统仍需要负责任的发布和持续监控。

免责声明:文中图文均来自网络,如有侵权请联系删除,心愿游戏发布此文仅为传递信息,不代表心愿游戏认同其观点或证实其描述。

相关文章

更多

精选合集

更多

大家都在玩

热门话题

大家都在看

更多