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>

编辑推荐精选

商汤小浣熊

商汤小浣熊

最强AI数据分析助手

小浣熊家族Raccoon,您的AI智能助手,致力于通过先进的人工智能技术,为用户提供高效、便捷的智能服务。无论是日常咨询还是专业问题解答,小浣熊都能以快速、准确的响应满足您的需求,让您的生活更加智能便捷。

imini AI

imini AI

像人一样思考的AI智能体

imini 是一款超级AI智能体,能根据人类指令,自主思考、自主完成、并且交付结果的AI智能体。

Keevx

Keevx

AI数字人视频创作平台

Keevx 一款开箱即用的AI数字人视频创作平台,广泛适用于电商广告、企业培训与社媒宣传,让全球企业与个人创作者无需拍摄剪辑,就能快速生成多语言、高质量的专业视频。

即梦AI

即梦AI

一站式AI创作平台

提供 AI 驱动的图片、视频生成及数字人等功能,助力创意创作

扣子-AI办公

扣子-AI办公

AI办公助手,复杂任务高效处理

AI办公助手,复杂任务高效处理。办公效率低?扣子空间AI助手支持播客生成、PPT制作、网页开发及报告写作,覆盖科研、商业、舆情等领域的专家Agent 7x24小时响应,生活工作无缝切换,提升50%效率!

TRAE编程

TRAE编程

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

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

AI工具TraeAI IDE协作生产力转型热门
蛙蛙写作

蛙蛙写作

AI小说写作助手,一站式润色、改写、扩写

蛙蛙写作—国内先进的AI写作平台,涵盖小说、学术、社交媒体等多场景。提供续写、改写、润色等功能,助力创作者高效优化写作流程。界面简洁,功能全面,适合各类写作者提升内容品质和工作效率。

AI辅助写作AI工具蛙蛙写作AI写作工具学术助手办公助手营销助手AI助手
问小白

问小白

全能AI智能助手,随时解答生活与工作的多样问题

问小白,由元石科技研发的AI智能助手,快速准确地解答各种生活和工作问题,包括但不限于搜索、规划和社交互动,帮助用户在日常生活中提高效率,轻松管理个人事务。

热门AI助手AI对话AI工具聊天机器人
Transly

Transly

实时语音翻译/同声传译工具

Transly是一个多场景的AI大语言模型驱动的同声传译、专业翻译助手,它拥有超精准的音频识别翻译能力,几乎零延迟的使用体验和支持多国语言可以让你带它走遍全球,无论你是留学生、商务人士、韩剧美剧爱好者,还是出国游玩、多国会议、跨国追星等等,都可以满足你所有需要同传的场景需求,线上线下通用,扫除语言障碍,让全世界的语言交流不再有国界。

讯飞智文

讯飞智文

一键生成PPT和Word,让学习生活更轻松

讯飞智文是一个利用 AI 技术的项目,能够帮助用户生成 PPT 以及各类文档。无论是商业领域的市场分析报告、年度目标制定,还是学生群体的职业生涯规划、实习避坑指南,亦或是活动策划、旅游攻略等内容,它都能提供支持,帮助用户精准表达,轻松呈现各种信息。

AI办公办公工具AI工具讯飞智文AI在线生成PPTAI撰写助手多语种文档生成AI自动配图热门
下拉加载更多