ITPub博客

首页 > 人工智能 > 人工智能 > Agent(智能体) Lightning:轻量级强化学习智能体训练框架

Agent(智能体) Lightning:轻量级强化学习智能体训练框架

人工智能 作者:ITPUB社群 时间:1970-01-01 08:33:46 0 删除 编辑

发布于 2026 年 10 月 7 日

作者:Zhiyuan He,研究软件开发工程师 II;Yuqing Yang,首席研究软件开发经理

概览

  • 驾驭式智能体强化学习(Harnessed Agentic RL):微软亚洲研究院提出了一种训练范式,让部署时使用的同一智能体框架(agent harness)直接参与强化学习,无需在训练框架内重新实现智能体。
  • 轻量设计:Agent Lightning v1.0 以约 3,500 行代码实现了完整的智能体强化学习控制平面。
  • 原生 Kubernetes(容器编排平台) 支持:智能体以标准 Kubernetes 作业的形式运行在自管理集群、云 Kubernetes 或本地基础设施上,不依赖付费的商业沙箱服务。
  • 数据高效的训练配方:一个端到端编码智能体流水线将 Qwen3.5-9B 在 SWE-bench Verified 上的 Pass@1 从 41.8% 提升至 56.4%,提升了 14.6 个百分点,且仅使用了约 6,000 条基于开源数据集的训练样本。

AI 智能体已从单一模型演变为由模型、工具和执行环境构建的复杂全栈系统。其能力越来越依赖于在模型之外协调它们的智能体框架(agent harness)。强化学习(RL)是一种让 AI 系统通过试错进行学习的方法,其行为由奖励和惩罚引导。RL 可以让这些智能体变得更好,但大多数智能体 RL 系统要求开发者在训练框架内重新实现智能体。这成本高昂,而且意味着被训练的智能体并非最终部署的那个智能体。

为解决这一问题,微软亚洲研究院的研究人员提出了驾驭式智能体强化学习(Harnessed Agentic RL)训练范式,并开源了完全重建的 Agent Lightning v1.0(在新标签页中打开)。与原始版本相比,Agent Lightning v1.0 更强调保持轻量、与真实框架集成,以及提供完整、可复现的智能体 RL 训练流水线。

Agent Lightning v1.0 围绕驾驭式智能体强化学习进行了重建,关键改进包括:

  • 轻量:整个框架约 3,500 行代码。Agent Lightning v1.0 在一个足够小且清晰的代码库中实现了完整的驾驭式智能体强化学习系统,便于理解、修改和扩展。
  • 在真实智能体框架上训练:智能体通过 Agent Lightning v1.0 中的大语言模型(LLM(大语言模型))代理(proxy)访问模型,现有框架代码保持不变。
  • 原生 Kubernetes 支持:智能体直接以 Kubernetes 作业的形式运行,无需外部商业沙箱服务。自管理集群和本地基础设施均可支持大规模 rollout。
  • 完整的编码智能体训练示例:一个基于 Qwen3.5-9B 的端到端流水线将 SWE-bench Verified 上的 Pass@1 从 41.8% 提升至 56.4%,绝对提升 14.6 个百分点,且仅使用了约 6,000 条训练样本。

传统智能体强化学习的局限

传统智能体强化学习假设训练框架拥有与环境交互的循环。在 ReAct 风格的循环中,模型生成一个动作,环境返回一个观察结果,观察结果被追加到上下文中,模型再生成下一个动作,因此整个 rollout 映射为一条连续的 token 轨迹。早期的 RL 系统如 verl、AReaL 和 slime 就是这样构建的,这意味着训练一个智能体需要在 RL 框架内重建其循环。

真实的框架已经超出了这一假设。诸如 mini-SWE-agent、OpenHands、OpenCode、Claude Code 和 Codex 等编码智能体各自带有自己的上下文管理、工具协议、执行逻辑和依赖,通用智能体系统也是如此。为训练而重建一个智能体成本高昂,而且重建后的智能体行为可能不再与部署的智能体相同。

Agent Lightning 采取了不同的路径。它在智能体与模型之间放置了一个 LLM 代理(proxy)。智能体继续像以前一样运行:只需将之前调用模型 API(应用程序编程接口) 的端点指向 Agent Lightning,训练框架就能观察并记录其模型调用。在 v1.0 中,研究人员更进一步,正式将这一范式定义为驾驭式智能体强化学习(Harnessed Agentic RL):部署时使用哪个智能体框架,训练时就由哪个框架直接参与强化学习(图 1)。

使用真实 harness 进行训练时的四个挑战

Harnessed Agentic RL 与传统 agentic RL 的一个核心区别在于,环境交互循环由 agent harness 处理,而非训练框架。训练系统只能观察到一系列 LLM 请求与响应对,因此单次 rollout 可能被拆分为数量不定的训练样本。这带来了四个关键挑战:

  • 重新分词与样本合并:Harness 将上下文保留为文本,但 RL 训练需要 rollout 期间采样的 token ID。将文本再次通过 chat template 和 tokenizer 可能导致 token 边界偏移,因此相邻调用并不总能合并为一个样本。
  • 优势计算:重新分词、子智能体和上下文摘要可能将一次 rollout 拆分为多个样本。直接在样本级别计算基线和优势会导致产生更多样本的 rollout 被重复计数,从而改变 rollout 级别原有的统计关系。
  • 损失归一化:按样本数量平均损失会使产生更多样本的 rollout 获得更高权重。由于样本数量往往只是 harness 行为的产物,损失归一化也必须避免受其扭曲。
  • 训练后端调度:样本数量和长度只有在 harness 完成后才可知,而 GPU(图形处理器) 数量和数据/张量并行配置通常是固定的。后端必须将可变工作负载映射到固定资源上。

用 3,500 行代码构建完整的 agent RL 控制平面

在系统设计上,Agent Lightning v1.0 将简洁性作为首要原则。整个框架约 3,500 行代码,包含三个核心组件:API Gateway、Rollout Controller 和 Customized Trainer(图 2)。

API gateway 存储 rollout、模型和事件,并充当 OpenAI 兼容的 LLM 代理。它将 harness 的每次模型调用关联到对应的 rollout,并记录训练所需的 prompt、响应和对数概率。rollout controller 启动并管理 agent 执行,可以作为本地进程或标准 Kubernetes job 运行,使 agent 执行与 trainer 保持分离。customized trainer 基于 verl 构建,负责创建 rollout、等待其完成、收集样本,并通过 sample adapter 组装最终训练样本。因此,对于现有的 agent harness,只需将模型端点指向 Agent Lightning proxy,通常就足以快速接入 RL 训练。

共置异步 RL

不同 agent 的 rollout 时间差异很大。同步 RL 会等待批次中最慢的 agent,导致 GPU 空闲,而完全异步 RL 提高了利用率,但需要为 rollout 和训练分别配置 GPU 池。为此,Agent Lightning v1.0 引入了 Collocated Async RL,使 rollout 和模型更新共享同一组 GPU。

一旦系统收集到足够的 rollout,更新便开始:API Gateway 暂停接受新请求,并等待已在进行的请求完成,更新完成后 rollout 恢复。整个状态转换对外部 agent harness 是透明的。在实验中,该方法相比同步 RL 实现了约 2 倍的端到端加速,同时使用的 GPU 少于传统异步 RL(图 3)。

在 Kubernetes 上运行 agent

收集足够的 rollout 意味着同时运行大量 agent,这会消耗大量 CPU、内存和计算资源。其他 Harnessed Agentic RL 框架通常将这些 agent 托管在 Modal Sandbox 或 E2B 等商业沙箱服务上,成本会随规模迅速攀升。而 Agent Lightning v1.0 将它们作为标准 Kubernetes job 运行,复用现有的自管理集群、云 Kubernetes 或本地基础设施(图 4)。现有计算资源得到更高效利用,大规模 rollout 成本更低,整个流水线保持开源且可复现。

6,000 个训练样本,性能提升 14.6 分

为测试该方法,研究人员在 SWE-smith、mini-SWE-agent 和 Qwen3.5-9B 上构建了完整流水线,涵盖数据清洗、环境构建、奖励黑客防护以及 RL 训练。训练集包含约 6,000 个样本,无需大规模算力。仅 RL 训练就将 Qwen3.5-9B 在 SWE-bench Verified 上的表现从 41.8% 提升至 56.4%,提高了 14.6 个百分点。

编码智能体实验进一步证实了此前对两个挑战的分析:优势计算和损失归一化。与样本级处理相比,rollout 级优势结合 rollout 级归一化可获得更高的验证奖励,并在训练过程中使策略熵保持更稳定(图 5)。


转载声明:本文为转载整理,原文版权归原作者及原站点所有。

原文来源:Microsoft Research Blog

原创地址:https://www.microsoft.com/en-us/research/blog/agent-lightning-v1-0-a-3500-line-lightweight-agentic-rl-framework-for-training-agents-with-real-harnesses/

原文作者:Brenda Potts;如有侵权,请联系管理员删除。

请登录后发表评论 登录
全部评论

注册时间:2022-11-21

  • 博文量
    1122
  • 访问量
    547409