Project Icon

RLeXplore

统一模块化工具包助力内在动机强化学习研究

RLeXplore是一个统一的模块化工具包,实现了八种代表性内在奖励算法。它通过标准化的程序解决了内在奖励算法比较中的混淆因素,包括实现、优化和评估方法的差异。该工具包支持多种内在奖励类型,如基于计数、好奇心驱动、基于记忆和信息论。RLeXplore提供了简便的安装方法、详细教程和基准测试结果,为内在动机强化学习研究提供了有力支持。



RLeXplore:加速内在动机强化学习研究

RLeXplore是一个统一、高度模块化和即插即用的工具包,目前提供了八种代表性内在奖励算法的高质量、可靠实现。由于各种混淆因素,包括不同的实现、优化策略和评估方法,比较内在奖励算法一直具有挑战性。因此,RLeXplore旨在为构建、计算和优化内在奖励模块提供统一和标准化的程序。

RLeXplore的工作流程如下图所示:

安装

  • 使用pip 推荐

打开终端,使用pip安装rllte

conda create -n rllte python=3.8
pip install rllte-core 
  • 使用git

打开终端,从GitHub使用git克隆仓库:

git clone https://github.com/RLE-Foundation/rllte.git
pip install -e .

现在你可以通过以下方式调用内在奖励模块:

from rllte.xplore.reward import ICM, RIDE, ...

模块列表

类型模块
基于计数PseudoCounts, RND, E3B
好奇心驱动ICM, Disagreement, RIDE
基于记忆NGU
基于信息理论RE3

教程

点击以下链接获取代码笔记本:

  1. 快速开始
  2. RLeXplore与RLLTE
  3. RLeXplore与Stable-Baselines3
  4. RLeXplore与CleanRL
  5. 探索混合内在奖励
  6. 自定义内在奖励

基准测试结果

  • RLLTE的PPO+RLeXplore超级马里奥兄弟上的表现:
  • CleanRL的PPO+RLeXplore的RND蒙特祖玛的复仇上的表现:

引用我们

在出版物中引用此仓库:

@article{yuan_roger2024rlexplore,
  title={RLeXplore: Accelerating Research in Intrinsically-Motivated Reinforcement Learning},
  author={Yuan, Mingqi and Castanyer, Roger Creus and Li, Bo and Jin, Xin and Berseth, Glen and Zeng, Wenjun},
  journal={arXiv preprint arXiv:2405.19548},
  year={2024}
}
项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

稿定AI

稿定设计 是一个多功能的在线设计和创意平台,提供广泛的设计工具和资源,以满足不同用户的需求。从专业的图形设计师到普通用户,无论是进行图片处理、智能抠图、H5页面制作还是视频剪辑,稿定设计都能提供简单、高效的解决方案。该平台以其用户友好的界面和强大的功能集合,帮助用户轻松实现创意设计。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号