ARTICLE · 人工智能

在 Amazon SageMaker AI 上进行多轮强化学习:实战指南 | Amazon Web Services

作者:Daniel Quang 来源:AWS Artificial Intelligence 2026-10-24 01:48 17 分钟 8 阅读 4167 字
AI 与智能应用模型训练与推理强化学习AI AgentAgent编排
一语总结

AWS 提供了一份全面的实战指南,介绍如何使用 Bedrock AgentCore 在 Amazon SageMaker AI 上实现多轮强化学习(MTRL),并在 Wordle 游戏任务上展示了显著的求解率提升。

AI 总结

本实战指南详细介绍了如何在 Amazon SageMaker AI 上使用多轮强化学习(MTRL)来训练智能体模型,使其能够执行多步动作,且奖励取决于最终结果。作者以 Wordle 作为教学示例,讲解了结合 SageMaker AI 进行训练、Amazon Bedrock AgentCore 托管环境和奖励函数的架构。文章涵盖了前提条件、IAM 角色、通过 CLI 部署智能体、准备数据集、启动训练、通过 MLflow 进行监控以及相关的定价动态。在测试运行中,MTRL 显著提升了 GPT-OSS-20B 模型的求解率,同时说明了推理努力配置和 rollout token 定价等实际考量因素。

核心要点
  1. 多轮强化学习解决智能体任务序列问题。

    与监督微调或单轮强化学习不同,MTRL 在多步序列上训练模型,其中每个动作都依赖于先前的反馈,且只有最终结果才能提供可靠的奖励信号。

  2. 架构将训练与环境托管分离。

    SageMaker AI 通过 GRPO 和 LoRA 管理训练作业和策略模型更新,而 Amazon Bedrock AgentCore 则托管智能体运行时、交互环境和评分逻辑。

  3. 训练成本和性能在很大程度上取决于 token 维度。

    定价按预填充、采样和训练 token 三个维度扩展,其中训练环节主导成本。控制推理努力和轨迹长度对于管理开支至关重要。

  4. 严格的 IAM 和运行时配置可防止静默失败。

    正确设置专用训练角色主体(`job.sagemaker.amazonaws.com`)、正确的 AgentCore 部署包装器以及结构化负载,对于成功执行 rollout 至关重要。

在 Amazon SageMaker AI 上进行多轮强化学习:实战指南 | Amazon Web Services

本实战指南详细介绍了如何在 Amazon SageMaker AI 上使用多轮强化学习(MTRL)来训练智能体模型,使其能够执行多步动作,且奖励取决于最终结果。作者以 Wordle 作为教学示例,讲解了结合 SageMaker AI 进行训练、Amazon Bedrock AgentCore 托管环境和奖励函数的架构。文章涵盖了前提条件、IAM 角色、通过 CLI 部署智能体、准备数据集、启动训练、通过 MLflow 进行监控以及相关的定价动态。在测试运行中,MTRL 显著提升了 GPT-OSS-20B 模型的求解率,同时说明了推理努力配置和 rollout token 定价等实际考量因素。

文章金句

"

大多数微调教模型针对单个提示生成更好的答案。智能体任务则不同。

"

SageMaker AI 负责训练作业和策略模型。AgentCore 托管智能体、Wordle 环境和奖励函数。

"

训练维度占主导地位并随轨迹长度扩展,因此更短的游戏和更简洁的推理可以降低费用。