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>

编辑推荐精选

iTerms

iTerms

企业专属的AI法律顾问

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

SimilarWeb流量提升

SimilarWeb流量提升

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

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

Sora2视频免费生成

Sora2视频免费生成

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

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

Transly

Transly

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

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

讯飞绘文

讯飞绘文

选题、配图、成文,一站式创作,让内容运营更高效

讯飞绘文,一个AI集成平台,支持写作、选题、配图、排版和发布。高效生成适用于各类媒体的定制内容,加速品牌传播,提升内容营销效果。

热门AI辅助写作AI工具讯飞绘文内容运营AI创作个性化文章多平台分发AI助手
TRAE编程

TRAE编程

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

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

AI工具TraeAI IDE协作生产力转型热门
商汤小浣熊

商汤小浣熊

最强AI数据分析助手

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

imini AI

imini AI

像人一样思考的AI智能体

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

Keevx

Keevx

AI数字人视频创作平台

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

即梦AI

即梦AI

一站式AI创作平台

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

下拉加载更多