MARTI:多智能体强化学习训练与推理框架
让 LLM 多智能体系统通过强化学习进行可扩展训练与树搜索增强推理,覆盖辩论、智能体链、混合智能体等工作流。
- Star 数
- ★ 574
- 最近更新
- 2 个月前
- License
- MIT
- 主语言
- Python
- FA 评分
- 40/100 · 缺口较多
30 秒速览
- 运行形态
- 可在哪里用
- 通用 · 跨平台
- 费用
- 免费,无需付费服务
- 上手难度
- 高 · 需要较重的基础设施
- 开始前需要
- 典型场景
- 研究人员想在多智能体辩论或混合智能体设置下,用 GRPO/PPO 对 LLM 做可复现的强化学习训练实验
- 不适合
- 只有单卡或消费级显卡的个人开发者
- 需要开箱即用推理服务或 Docker 部署的团队
- 希望免维护依赖栈、不想自己搭建 OpenRLHF/Ray/vLLM 的用户
这个 Agent 能做什么,适合哪些场景?
MARTI 是清华大学与上海人工智能实验室开源的 LLM 多智能体强化学习(MARL)训练与推理框架,已被 ICLR 2026 接收。框架遵循“集中式多智能体交互 + 分布式策略训练”原则,包含多智能体世界(Multi-Agent World)、集中式奖励(Centralized Rewarding)和单智能体训练器(Single Agent Trainer)三大模块,底层基于 OpenRLHF,并支持 vLLM v1 引擎与 Hybrid Engine。最新版本 MARTI-v2 引入树搜索增强强化学习(MARS²),面向代码生成等复杂推理任务,提供异步多智能体 MCTS 训练,并集成 GSPO 损失、TIS 修正、动态数据过滤和超长序列缓冲等训练技术,支持最高 32K token 序列与异构多智能体(如 Qwen3-8B + AreaL-boba-2-8B)联合训练。框架内置辩论、智能体链、混合智能体等图式工作流,实验性支持 AutoGen 与 CAMEL 第三方框架。在 LCB 代码生成基准上,多智能体 MCTS 使 Qwen3-8B 相对基础模型提升 8.0%、相对 Vanilla GRPO 提升 4.4%。注意这是一套研究性训练框架,而非开箱即用的推理服务。
MARTI 读取配置与训练脚本(examples/ 目录下的 bash 脚本),通过 OpenRLHF 基础设施与 vLLM 引擎执行多智能体强化学习训练。集中式的 Multi-Agent World 模块负责智能体间交互与奖励分配,Centralized Rewarding 模块实现信用分配与奖励塑形,Single Agent Trainer 对每个智能体的策略进行分布式训练。支持 PPO、GRPO、REINFORCE++、TTRL 等 RL 算法,以及辩论、Chain-of-Agents、Mixture-of-Agents、Review-RL 等工作流。MARTI-v2 的 MARS² 模块执行异步多智能体 MCTS 树搜索,通过自适应节点扩展与精炼探索解空间,产出高质量推理轨迹用于策略更新,支持 GSPO 序列级优化、TIS 采样修正、按智能体的动态样本过滤和超长 token 惩罚缓冲。训练产物是经过多智能体 RL 优化的模型权重(如 Qwen3-8B/14B、DeepCoder-14B 等)。
- 研究人员想在多智能体辩论或混合智能体设置下,用 GRPO/PPO 对 LLM 做可复现的强化学习训练实验
- 算法工程师需要训练代码生成模型,希望用 MCTS 树搜索代替朴素 rollout 来提升 LCB 类基准的 Pass@1
- 团队要联合训练多个异构模型(如 Qwen3-8B + AreaL-boba-2-8B),各智能体拥有独立角色、训练策略和样本过滤
- 已有 AutoGen 或 CAMEL 工作流的用户,想把现有编排接入 RL 训练管线(实验性支持)
- 需要异步工具调用(Async Tool Use)和多轮智能体 RL(如代码/数学多轮 RL)的研究场景
如何安装或部署这个 Agent?
安装步骤(来自仓库):
- 克隆仓库并安装依赖:
git clone https://github.com/TsinghuaC3I/MARTI.git
cd MARTI
pip install -r requirements.txt- 按仓库说明配置依赖,包括 OpenRLHF、Ray 和 vLLM。
硬件门槛:单智能体训练约需 8×80G GPU;多智能体训练每个智能体约需 8×80G GPU。仓库未提供 Docker 镜像或托管服务,需自行准备 GPU 集群环境。
如何使用这个 Agent?
在脚本中设置路径后运行对应示例脚本:
ROOT_DIR="/path/to/MARTI"
MODEL_DIR="/path/to/models"# 单智能体 MCTS 训练(MARS²)
bash examples/mars2/run_train_single_mcts.sh# 多智能体 MCTS 训练
bash examples/mars2/run_train_multi_mcts.sh# 异步多轮代码 RL(单智能体)
bash examples/single-agent/run_train_code_async.sh# 多智能体辩论
bash examples/multi-agent/run_train_mad.sh
# Chain-of-Agents (MathChat)
bash examples/multi-agent/run_train_mathchat.sh
# Review-RL
bash examples/reviewrl/run_train_reviewrl_async.sh文档位于 docs/ 目录(Overview、Workflows Integration、Reward and Training、Experiments)。
这个 Agent 有哪些优点和局限?
- ICLR 2026 论文支撑,并提供 arXiv 技术报告(2602.07848),方法有同行评审背书
- 多智能体训练与单智能体 RL 在同一框架内统一,内置信用分配与奖励塑形,省去自行拼装管线
- MARS² 提供异步多智能体 MCTS 训练,LCB 基准上多智能体相对 Vanilla GRPO 提升最高 4.4%,且支持 32K 超长序列与异构智能体训练
- 支持 PPO、GRPO、REINFORCE++、TTRL 多种 RL 算法,并实验性兼容 AutoGen/CAMEL 现有编排
- 硬件门槛高:单智能体约需 8×80G GPU,多智能体每个智能体再需 8×80G,论文实验使用 3 节点各 8×H200,个人或小团队难以复现
- 研究性框架,AutoGen/CAMEL 集成为实验性,生产化程度未知
- 无托管服务或 Docker 部署方案,需要自行维护 OpenRLHF/Ray/vLLM 依赖栈
- 对模型家族有具体实验证据(Qwen3、AreaL-boba、DeepCoder、DeepSeek-R1、Gemma、Llama 出现在主题标签中),其他模型的实际效果缺乏公开验证
这个 Agent 与同类方案有什么区别?
MARTI 明确构建于单智能体 RL 框架 OpenRLHF 与 verl 之上,与它们的关系是扩展而非替代:OpenRLHF/verl 面向单智能体策略训练,MARTI 在此之上增加集中式多智能体交互、奖励分配与多智能体树搜索(MARS²)。README 未提及其他具名多智能体 RL 竞品。
与相关度最高的同类 agent 并排比较关键指标。
| Agent | 源码审查 | 形态 / 费用 | Star | 最近更新 | 主语言 | 完整支持的平台 |
|---|---|---|---|---|---|---|
| MARTI:多智能体强化学习训练与推理框架 当前 | 40 · 缺口较多 | 命令行工具免费 | ★ 574 | 2 个月前 | Python | — |
| AgentsMeetRL——智能体强化学习资源大全 | 57 · 缺口较多 | Agent 插件 / 技能免费 | ★ 1.9k | 2 天前 | HTML | — |
| Hands-On Modern RL 实战教程 | 68 · 存在缺口 | 命令行工具免费 | ★ 4.9k | 16 天前 | Jupyter Notebook | — |
| rLLM:大语言模型智能体强化学习框架 | 54 · 缺口较多 | 命令行工具免费 + 模型费 | ★ 5.9k | 23 天前 | Python | OpenAI API |
FollowAgents 如何评估这个 Agent?
查看各维度的扣分理由
代码将 Agent 执行并发限制在信号量内,并将提示、标签、采样参数和生成结果传入执行及结果队列,提供了有限的数据流线索,因此最小权限和透明度各给 1 分;材料没有展示用户确认、敏感数据保护或回滚机制,故相应得 0 分。依赖清单列出固定或有下限的部分版本,但没有漏洞审查或完整锁定,依赖安全给 1 分。执行涉及模型生成、Ray 远程实例和队列写入,外部效果有代码迹象但范围说明有限,给 1 分。README 列出 OpenRLHF、vLLM、Ray 等基础项目,并提供引用和贡献者信息,来源归属给 2 分。
Agent 抽象、异步执行流程和 GAIA 答案评分代码支持了有限的内部一致性,但执行路径依赖 reset/step 返回特定键及 rewards.item(),未见相应接口校验,故自洽性给 1 分。依赖清单和安装步骤存在,但关键依赖未全部固定版本且外部基础设施要求较多,可用性给 1 分。执行代码对缺失字段、生成失败等没有明显的用户可读错误处理,失败消息给 1 分。
README 覆盖单/多 Agent、代码与数学任务、工作流和树搜索训练等目标场景,给 2 分;但能力边界和实验性集成仅简要提及,给 1 分。示例脚本明确入口和训练场景,但何时选择各模式的触发条件不充分,给 1 分。硬件需求和模型路径说明了部分环境要求,然而集群、依赖安装和配置细节仍需外部指引,环境适配给 1 分。
README 有目录、模块文档链接和分组示例,信息架构给 2 分。安装仅提供 requirements 安装命令并提示另行配置依赖,故安装说明给 1 分。模块与任务命名总体清晰,给 2 分;示例覆盖多种训练入口但没有 FAQ,给 2 分。实验性集成和硬件需求透露了部分限制,但局限清单不完整,给 1 分。仓库含完整 MIT 许可证,给 3 分。新闻日期构成简要更新记录,但未提供版本化发布说明,给 1 分。README 指定项目负责人、贡献者和联系邮箱,维护责任给 2 分。
代码在执行后汇总提示、标签、观察 token、奖励、分数、动作范围和日志,输出可供训练流程使用,给 2 分。框架将多 Agent 工作流与强化学习训练结合,README 给出了明确的使用场景和能力主张,边际价值给 2 分;但训练要求约 8 张 80GB GPU,且多 Agent 要求按 Agent 配置节点,成本效益说明有限,给 1 分。
README 对功能、硬件需求和基准结果给出具体描述,但材料中缺少与这些主张逐项对应的配置、原始数据或可复核过程,主张可追溯性给 1 分。README、执行代码、评分代码、依赖清单和许可证提供了不同类型的相互补充证据,但没有测试或文档交叉验证执行行为,跨源佐证给 1 分。部分陈述明确作为框架特性或实验结果提出,但结果仍主要是 README 中的作者陈述,事实与推断区分不充分,给 1 分。
- 源码中未见:执行前用户确认开启或自行加上执行前确认;先在沙箱或测试环境跑通,确认行为后再接入真实数据。
- 源码中未见:敏感信息处理使用专用、低权限、可随时吊销的 API 密钥,不要复用生产凭据,也不要让密钥出现在日志里。
- 源码中未见:回滚或恢复路径运行前先备份,或在 git 分支、快照上操作,确保改动可以撤销。
- 这是基于所给静态材料的低置信度审查;未执行代码,也未独立核实 README 中的实验结果。
- 运行依赖 Ray、vLLM、模型与 GPU 集群;README 给出的硬件门槛较高,部署和成本需要谨慎评估。
- 执行代码依赖 Agent 的 reset/step 返回约定,并会把输入和标签传递给执行实例;接入真实数据前应审查数据处理与错误路径。
常见问题
最低硬件要求是多少?
它是一个可以直接部署的推理服务吗?
支持哪些模型和 RL 算法?
能否接入我已有的 AutoGen 或 CAMEL 工作流?
有没有社区项目基于它?
谢谢反馈,我们会据此改进页面。