MAgent

GitHub
1.8k 333 较难 1 次阅读 1周前MITAgent开发框架
AI 解读 由 AI 自动生成,仅供参考

MAgent 是一个专为多智能体强化学习(MARL)设计的开源研究平台。与传统平台仅关注单个或少量智能体不同,MAgent 的核心突破在于能够支持从数百到数百万量级的大规模智能体协同训练与仿真,有效解决了现有工具在处理超大规模群体智能时面临的扩展性瓶颈。

该平台特别适合人工智能研究人员、算法工程师以及对群体智能感兴趣的高校师生使用。用户既可以基于规则编写算法,也能结合 TensorFlow 或 MXNet 等深度学习框架开发复杂模型。MAgent 内置了追捕、聚集和战斗等多种经典场景示例,并提供了参数共享 DQN、DRQN 及 A2C 等基线算法实现,帮助用户快速验证想法。其独特的技术亮点在于高效的底层架构,能够在普通显卡上支撑海量智能体的并行交互,为研究人工集体智能提供了坚实的实验基础。

需要注意的是,原项目目前已停止维护,官方建议有需求的用户转向其继任版本 MAgent2,以获得更好的安装体验和持续的技术支持。无论是探索群体协作机制,还是测试大规模分布式策略,MAgent 都是一个极具价值的入门与科研工具。

使用场景

某智慧城市研究团队正在开发大规模无人机集群的协同交通疏导系统,需要模拟成千上万个智能体在复杂路况下的实时协作。

没有 MAgent 时

  • 规模瓶颈严重:传统强化学习平台仅支持单代理或少量代理,一旦智能体数量超过几百个,仿真环境立即崩溃或计算延迟极高,无法模拟真实的城市级无人机规模。
  • 协作逻辑难以涌现:由于缺乏专门的多智能体交互机制,开发者只能硬编码规则让无人机避障,导致群体无法自发形成高效的“蜂群”疏导策略,拥堵缓解效果差。
  • 算法验证成本高:每次测试新的协同算法都需要重写底层通信和渲染模块,且无法复用现有的深度强化学习基线(如 DQN、A2C),研发周期长达数月。

使用 MAgent 后

  • 百万级规模支撑:MAgent 专为大规模设计,轻松支撑从数百到数百万智能体的并行训练与推理,完美复现了城市上空千架无人机同时调度的宏大场景。
  • 集体智能自然涌现:利用其内置的多智能体强化学习环境,无人机集群通过自我博弈学会了复杂的包围、分流和接力引导策略,无需人工预设具体动作规则。
  • 开箱即用的基线算法:直接调用 MAgent 集成的参数共享 DQN 和 A2C 等算法模板,团队将原本数月的算法验证工作缩短至几天内完成,并快速迭代出最优模型。

MAgent 通过突破智能体数量上限并提供标准化的多智能体训练环境,让研究者能够从微观的单点控制跨越到宏观的集体智能涌现研究。

运行环境要求

操作系统
  • Linux
  • macOS
GPU

非运行必需,但训练示例建议在 NVIDIA GTX1080-Ti 或同等显卡上运行(显存不足时可调整 batch_size)

内存

未说明(提及若发生内存溢出错误需减小 infer_batch_size)

依赖
notes该项目已不再维护,建议使用其维护分支 MAgent2(支持 pip 安装)。macOS 用户安装 websocketpp 时可能遇到 homebrew 问题,需参考特定 issue 解决。基线算法基于 TensorFlow 和 MXNet 实现。
python2.7 或 3.x
cmake
libboost-system-dev (Linux) / boost@1.55 (macOS)
libjsoncpp-dev
libwebsocketpp-dev
TensorFlow
MXNet
MAgent hero image

快速开始

该项目已不再维护

请参阅 https://github.com/Farama-Foundation/MAgent2,这是一个可使用 pip 安装的、仍在维护的本项目的分支。

MAgent

MAgent 是一个用于多智能体强化学习的研究平台。与以往专注于单智能体或少数几个智能体强化学习研究的平台不同,MAgent 的目标是支持从数百个智能体扩展到数百万个智能体的强化学习研究。

系统要求

MAgent 支持运行 Python 2.7 或 Python 3 的 Linux 和 OS X 系统。 我们不对您的智能体结构做任何假设。 您可以编写基于规则的算法,也可以使用深度学习框架。

在 Linux 上安装

git clone git@github.com:geek-ai/MAgent.git
cd MAgent

sudo apt-get install cmake libboost-system-dev libjsoncpp-dev libwebsocketpp-dev

bash build.sh
export PYTHONPATH=$(pwd)/python:$PYTHONPATH

在 OS X 上安装

注意:使用 Homebrew 安装 websocketpp 存在问题,请参考 #17

git clone git@github.com:geek-ai/MAgent.git
cd MAgent

brew install cmake llvm boost@1.55
brew install jsoncpp argp-standalone
brew tap david-icracked/homebrew-websocketpp
brew install --HEAD david-icracked/websocketpp/websocketpp
brew link --force boost@1.55

bash build.sh
export PYTHONPATH=$(pwd)/python:$PYTHONPATH

文档

快速入门

示例

以下任务的训练时间在 GTX1080-Ti 显卡上约为 1 天。 如果出现内存不足错误,可以在模型中将 infer_batch_size 调小。

注意:您应在本仓库的根目录下运行以下示例,不要进入 examples/ 目录。

训练

上面视频中展示的三个示例。 每 10 轮会保存一次视频文件,您可以使用 render 功能观看这些视频。

  • 追逐

    python examples/train_pursuit.py --train
    
  • 收集

    python examples/train_gather.py --train
    
  • 战斗

    python examples/train_battle.py --train
    

游戏

一个与战斗智能体互动的游戏。您将扮演一名将军,指挥您的士兵。

  • 战斗游戏
    python examples/show_battle_game.py
    

基线算法

基线算法包括参数共享的 DQN、DRQN 和 A2C,它们分别用 TensorFlow 和 MXNet 实现。 在我们的大规模共享和网格世界环境中,DQN 表现最佳。

致谢

非常感谢 Tianqi Chen 提供的宝贵建议。

常见问题

相似工具推荐

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|昨天
开发框架图像Agent

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

140.4k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

107.7k|★★☆☆☆|3天前
开发框架图像Agent

Deep-Live-Cam

Deep-Live-Cam 是一款专注于实时换脸与视频生成的开源工具,用户仅需一张静态照片,即可通过“一键操作”实现摄像头画面的即时变脸或制作深度伪造视频。它有效解决了传统换脸技术流程繁琐、对硬件配置要求极高以及难以实时预览的痛点,让高质量的数字内容创作变得触手可及。 这款工具不仅适合开发者和技术研究人员探索算法边界,更因其极简的操作逻辑(仅需三步:选脸、选摄像头、启动),广泛适用于普通用户、内容创作者、设计师及直播主播。无论是为了动画角色定制、服装展示模特替换,还是制作趣味短视频和直播互动,Deep-Live-Cam 都能提供流畅的支持。 其核心技术亮点在于强大的实时处理能力,支持口型遮罩(Mouth Mask)以保留使用者原始的嘴部动作,确保表情自然精准;同时具备“人脸映射”功能,可同时对画面中的多个主体应用不同面孔。此外,项目内置了严格的内容安全过滤机制,自动拦截涉及裸露、暴力等不当素材,并倡导用户在获得授权及明确标注的前提下合规使用,体现了技术发展与伦理责任的平衡。

88.9k|★★★☆☆|今天
开发框架图像Agent

NextChat

NextChat 是一款轻量且极速的 AI 助手,旨在为用户提供流畅、跨平台的大模型交互体验。它完美解决了用户在多设备间切换时难以保持对话连续性,以及面对众多 AI 模型不知如何统一管理的痛点。无论是日常办公、学习辅助还是创意激发,NextChat 都能让用户随时随地通过网页、iOS、Android、Windows、MacOS 或 Linux 端无缝接入智能服务。 这款工具非常适合普通用户、学生、职场人士以及需要私有化部署的企业团队使用。对于开发者而言,它也提供了便捷的自托管方案,支持一键部署到 Vercel 或 Zeabur 等平台。 NextChat 的核心亮点在于其广泛的模型兼容性,原生支持 Claude、DeepSeek、GPT-4 及 Gemini Pro 等主流大模型,让用户在一个界面即可自由切换不同 AI 能力。此外,它还率先支持 MCP(Model Context Protocol)协议,增强了上下文处理能力。针对企业用户,NextChat 提供专业版解决方案,具备品牌定制、细粒度权限控制、内部知识库整合及安全审计等功能,满足公司对数据隐私和个性化管理的高标准要求。

87.6k|★★☆☆☆|昨天
开发框架语言模型

ML-For-Beginners

ML-For-Beginners 是由微软推出的一套系统化机器学习入门课程,旨在帮助零基础用户轻松掌握经典机器学习知识。这套课程将学习路径规划为 12 周,包含 26 节精炼课程和 52 道配套测验,内容涵盖从基础概念到实际应用的完整流程,有效解决了初学者面对庞大知识体系时无从下手、缺乏结构化指导的痛点。 无论是希望转型的开发者、需要补充算法背景的研究人员,还是对人工智能充满好奇的普通爱好者,都能从中受益。课程不仅提供了清晰的理论讲解,还强调动手实践,让用户在循序渐进中建立扎实的技能基础。其独特的亮点在于强大的多语言支持,通过自动化机制提供了包括简体中文在内的 50 多种语言版本,极大地降低了全球不同背景用户的学习门槛。此外,项目采用开源协作模式,社区活跃且内容持续更新,确保学习者能获取前沿且准确的技术资讯。如果你正寻找一条清晰、友好且专业的机器学习入门之路,ML-For-Beginners 将是理想的起点。

85k|★★☆☆☆|昨天
图像数据工具视频