如何提升强化学习(RL)的训练效率,尤其在长程智能体(Agent)与代码生成任务中,始终是业界亟待突破的核心难题。
当前,主流的大模型强化学习(LLM RL)流水线大多采用同步、批量式的处理方式。简而言之,必须等待一整批数据全部生成完毕,才能触发模型更新,这种模式自然限制了效率的提升。虽然异步RL能够缓解同步等待的瓶颈,但现有方法更多聚焦于系统吞吐量,对训练稳定性及最终任务效果的关注仍显不足。此外,像GRPO这类方法依赖于组式采样,很难直接迁移到异步Agent训练场景中。
针对这一痛点,清华大学KEG实验室的研究团队提出了一种全新方法——单rollout异步优化(Single-rollout Asynchronous Optimization,SAO)。其核心思路十分直接:用单个rollout采样替代GRPO的组式采样。即,每个输入任务仅生成一条rollout,生成后立即用于训练,从而降低离策略影响,增强模型的泛化能力。

研究结果表明,SAO能够稳定训练至1000步,在Agent编码与数学推理基准上,表现持续优于GRPO及其各类变体。同时,这种单rollout的强化学习方法在模拟在线学习任务中,也展现出了对动态环境的良好适应能力。
这项工作为高效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的系统仍需负责任的发布和持续监控。
