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

白日梦AI提供专注于AI视频生成的多样化功能,包括文生视频、动态画面和形象生成等,帮助用户快速上手,创造专业级内容。

Project Cover

有言AI

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

Project Cover

Kimi

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

Project Cover

讯飞绘镜

讯飞绘镜是一个支持从创意到完整视频创作的智能平台,用户可以快速生成视频素材并创作独特的音乐视频和故事。平台提供多样化的主题和精选作品,帮助用户探索创意灵感。

Project Cover

讯飞文书

讯飞文书依托讯飞星火大模型,为文书写作者提供从素材筹备到稿件撰写及审稿的全程支持。通过录音智记和以稿写稿等功能,满足事务性工作的高频需求,帮助撰稿人节省精力,提高效率,优化工作与生活。

Project Cover

阿里绘蛙

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

Project Cover

AIWritePaper论文写作

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

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