DRL-Pytorch

DRL-Pytorch

PyTorch实现的深度强化学习算法集合

DRL-Pytorch项目提供多种常用深度强化学习算法的PyTorch实现,包括Q-learning、DQN变体、PPO、DDPG、TD3和SAC等。代码结构清晰统一,便于研究人员和开发者比较不同算法。项目还包含详细使用说明、依赖列表和学习资源推荐,有助于快速入门和实践。

强化学习PyTorchDRL算法深度学习人工智能Github开源项目
<div align=center> <img src="https://yellow-cdn.veclightyear.com/0a4dffa0/11f3e200-0e1d-4086-98f5-0cb10ff2087a.png" width=500 /> </div> <div align=center> 流行深度强化学习算法的清晰、健壮和统一的PyTorch实现 </div> <div align=center> <img src="https://img.shields.io/badge/Python-blue" /> <img src="https://img.shields.io/badge/Pytorch-ff69b4" /> <img src="https://img.shields.io/badge/DRL-blueviolet" /> </div> <br/> <br/>

0.星标历史

<div align="left"> <img width="70%" height="auto" src="https://api.star-history.com/svg?repos=XinJingHao/Deep-Reinforcement-Learning-Algorithms-with-Pytorch&type=Date"> </div> <br/>

1.依赖项

除非另有说明,本仓库使用以下Python依赖:

gymnasium==0.29.1 numpy==1.26.1 pytorch==2.1.0 python==3.11.5
<br/>

2.如何使用我的代码

进入你想使用的算法文件夹,运行main.py从头开始训练:

python main.py

更多详细信息,请查看相应算法文件夹中的README.md文件。

<br/>

3.代码的单独链接

<br/>

4.推荐的深度强化学习资源

4.1 模拟环境:

  • gymgymnasium(轻量级且标准的深度强化学习环境;易于入门;速度较慢):
<div align="left"> <img width="60%" height="auto" src="https://yellow-cdn.veclightyear.com/0a4dffa0/64aab302-222f-4c7a-9ed5-49344898f5b8.gif"> </div> <br/>
  • Isaac Gym(NVIDIA的物理模拟环境;GPU加速;超快):
<div align="left"> <img width="60%" height="auto" src="https://yellow-cdn.veclightyear.com/0a4dffa0/c791f9a6-20fa-415d-bba6-e6004e49f2c8.gif"> </div> <br/>
  • Sparrow(轻量级移动机器人模拟器;适合深度强化学习):
<div align="left"> <img width="60%" height="auto" src="https://yellow-cdn.veclightyear.com/0a4dffa0/3d5617f6-3e5c-440d-b42a-8bea14e8f39d.gif"> </div> <br/>
  • ROS(流行且全面的机器人物理模拟器;较重且速度慢):
<div align="left"> <img width="60%" height="auto" src="https://yellow-cdn.veclightyear.com/0a4dffa0/4d94a18d-ab4f-4319-a34d-db3a6371eea3.gif"> </div> <br/>
  • Webots(流行的机器人物理模拟器;比ROS更快;真实性稍低):
<div align="left"> <img width="60%" height="auto" src="https://yellow-cdn.veclightyear.com/0a4dffa0/8839c464-0e6a-46f3-b794-657b59bb4c30.gif"> </div> <br/>

4.2 书籍:

4.3 在线课程:

4.4 博客:

<br/>

5. 重要论文

DQN: Mnih V, Kavukcuoglu K, Silver D, 等. 通过深度强化学习实现人类水平的控制[J]. 自然, 2015, 518(7540): 529-533.

Double DQN: Van Hasselt H, Guez A, Silver D. 具有双Q学习的深度强化学习[C]//AAAI人工智能会议论文集. 2016, 30(1).

Duel DQN: Wang, Ziyu, 等. "用于深度强化学习的决斗网络架构." 国际机器学习会议. PMLR, 2016.

PER: Schaul T, Quan J, Antonoglou I, 等. 优先经验回放[J]. arXiv预印本 arXiv:1511.05952, 2015.

C51: Bellemare M G, Dabney W, Munos R. 强化学习的分布式视角[C]//国际机器学习会议. PMLR, 2017: 449-458.

NoisyNet DQN: Fortunato M, Azar M G, Piot B, 等. 用于探索的噪声网络[J]. arXiv预印本 arXiv:1706.10295, 2017.

PPO: Schulman J, Wolski F, Dhariwal P, 等. 近端策略优化算法[J]. arXiv预印本 arXiv:1707.06347, 2017.

DDPG: Lillicrap T P, Hunt J J, Pritzel A, 等. 使用深度强化学习进行连续控制[J]. arXiv预印本 arXiv:1509.02971, 2015.

TD3: Fujimoto S, Hoof H, Meger D. 解决演员-评论家方法中的函数逼近误差[C]//国际机器学习会议. PMLR, 2018: 1587-1596.

SAC: Haarnoja T, Zhou A, Abbeel P, 等. 软演员-评论家:离策略最大熵深度强化学习与随机演员[C]//国际机器学习会议. PMLR, 2018: 1861-1870.

ASL: 一小时内训练真实世界局部路径规划器:通过部分解耦强化学习和矢量化多样性

<br/>

6. 我的代码训练曲线:

Q学习:

<img src="https://yellow-cdn.veclightyear.com/0a4dffa0/6b0a896f-e5b6-46da-86b8-19dadbe6e197.svg" width=320>

决斗双重DQN:

CartPoleLunarLander
<img src="https://yellow-cdn.veclightyear.com/0a4dffa0/140bc2df-be25-44c6-ac5d-a262f2ed08cd.png" width="320" height="200"><img src="https://yellow-cdn.veclightyear.com/0a4dffa0/ef6e8f06-1269-4f28-a085-a3ec511df817.png" width="320" height="200">

Atari游戏上的噪声对偶DDQN:

乒乓球耐力赛车
<img src="https://yellow-cdn.veclightyear.com/0a4dffa0/4714fdc6-905b-4193-aa2c-4c6ad12f9c9f.png" width="320" height="200"><img src="https://yellow-cdn.veclightyear.com/0a4dffa0/ebd32444-3b92-43f0-aab5-88b2d580e96b.png" width="320" height="200">
<br/>

优先级DQN/DDQN:

平衡杆月球着陆器
<img src="https://yellow-cdn.veclightyear.com/0a4dffa0/b3ad2a6a-abb9-4e47-aac8-12b7f23d0fb4.svg" width="320" height="200"><img src="https://yellow-cdn.veclightyear.com/0a4dffa0/eed8a6e5-bbbc-4d32-9749-a0e1efc34577.svg" width="320" height="200">
<br/>

分类DQN:

平衡杆月球着陆器
<img src="https://yellow-cdn.veclightyear.com/0a4dffa0/23eaae5e-1466-49d6-bc94-51831b6f4c9c.svg" width="320" height="200"><img src="https://yellow-cdn.veclightyear.com/0a4dffa0/d205ab6c-b73a-4dce-9747-9c83f1be4cd5.svg" width="320" height="200">
<br/>

噪声网络DQN:

平衡杆月球着陆器
<img src="https://yellow-cdn.veclightyear.com/0a4dffa0/ae219a82-9215-47bf-9b79-8475368bae05.png" width="320" height="200"><img src="https://yellow-cdn.veclightyear.com/0a4dffa0/594103f0-b2b0-4532-a24f-356e9de57203.png" width="320" height="200">
<br/>

离散PPO:

<img src="https://yellow-cdn.veclightyear.com/0a4dffa0/0a0c7a7e-7e33-46cd-a72a-96499e412ca3.jpg" width=700>

连续PPO:

<img src="https://yellow-cdn.veclightyear.com/0a4dffa0/9091ac74-e8e4-46d2-a589-2abdbae6e421.jpg">

DDPG:

钟摆连续月球着陆器
<img src="https://yellow-cdn.veclightyear.com/0a4dffa0/79947025-eabc-4b30-9ecb-50c28f12e6c5.svg" width="320" height="200"><img src="https://yellow-cdn.veclightyear.com/0a4dffa0/c8841a81-1922-4ce6-a69b-1b88d4fd8d3b.svg" width="320" height="200">
<br/>

TD3:

<img src="https://yellow-cdn.veclightyear.com/0a4dffa0/aad2bb98-3ff8-4896-a526-24be3f3aa315.png" width=700>

连续SAC:

<img src="https://yellow-cdn.veclightyear.com/0a4dffa0/8e946117-1ca7-4c74-b445-b2dcc22d099a.jpg" width=700>

离散SAC:

<img src="https://yellow-cdn.veclightyear.com/0a4dffa0/03e935f8-b93b-4b36-94ea-8fc272094076.jpg" width=700>

演员-共享者-学习者 (ASL):

<div align="left"> <img width="70%" height="auto" src="https://yellow-cdn.veclightyear.com/0a4dffa0/98a7620c-fe53-4844-b597-b72da53f3055.svg"> </div>

编辑推荐精选

Vora

Vora

免费创建高清无水印Sora视频

Vora是一个免费创建高清无水印Sora视频的AI工具

Refly.AI

Refly.AI

最适合小白的AI自动化工作流平台

无需编码,轻松生成可复用、可变现的AI自动化工作流

酷表ChatExcel

酷表ChatExcel

大模型驱动的Excel数据处理工具

基于大模型交互的表格处理系统,允许用户通过对话方式完成数据整理和可视化分析。系统采用机器学习算法解析用户指令,自动执行排序、公式计算和数据透视等操作,支持多种文件格式导入导出。数据处理响应速度保持在0.8秒以内,支持超过100万行数据的即时分析。

AI工具酷表ChatExcelAI智能客服AI营销产品使用教程
TRAE编程

TRAE编程

AI辅助编程,代码自动修复

Trae是一种自适应的集成开发环境(IDE),通过自动化和多元协作改变开发流程。利用Trae,团队能够更快速、精确地编写和部署代码,从而提高编程效率和项目交付速度。Trae具备上下文感知和代码自动完成功能,是提升开发效率的理想工具。

AI工具TraeAI IDE协作生产力转型热门
AIWritePaper论文写作

AIWritePaper论文写作

AI论文写作指导平台

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

AI辅助写作AI工具AI论文工具论文写作智能生成大纲数据安全AI助手热门
博思AIPPT

博思AIPPT

AI一键生成PPT,就用博思AIPPT!

博思AIPPT,新一代的AI生成PPT平台,支持智能生成PPT、AI美化PPT、文本&链接生成PPT、导入Word/PDF/Markdown文档生成PPT等,内置海量精美PPT模板,涵盖商务、教育、科技等不同风格,同时针对每个页面提供多种版式,一键自适应切换,完美适配各种办公场景。

AI办公办公工具AI工具博思AIPPTAI生成PPT智能排版海量精品模板AI创作热门
潮际好麦

潮际好麦

AI赋能电商视觉革命,一站式智能商拍平台

潮际好麦深耕服装行业,是国内AI试衣效果最好的软件。使用先进AIGC能力为电商卖家批量提供优质的、低成本的商拍图。合作品牌有Shein、Lazada、安踏、百丽等65个国内外头部品牌,以及国内10万+淘宝、天猫、京东等主流平台的品牌商家,为卖家节省将近85%的出图成本,提升约3倍出图效率,让品牌能够快速上架。

iTerms

iTerms

企业专属的AI法律顾问

iTerms是法大大集团旗下法律子品牌,基于最先进的大语言模型(LLM)、专业的法律知识库和强大的智能体架构,帮助企业扫清合规障碍,筑牢风控防线,成为您企业专属的AI法律顾问。

SimilarWeb流量提升

SimilarWeb流量提升

稳定高效的流量提升解决方案,助力品牌曝光

稳定高效的流量提升解决方案,助力品牌曝光

Sora2视频免费生成

Sora2视频免费生成

最新版Sora2模型免费使用,一键生成无水印视频

最新版Sora2模型免费使用,一键生成无水印视频

下拉加载更多