Skip to content

RLeXplore provides stable baselines of exploration methods in reinforcement learning, such as intrinsic curiosity module (ICM), random network distillation (RND) and rewarding impact-driven exploration (RIDE).

License

RLE-Foundation/RLeXplore

Repository files navigation



RLeXplore: Benchmarking Intrinsic Reward Shaping in Reinforcement Learning

RLeXplore is a unified, highly-modularized and plug-and-play toolkit that currently provides high-quality and reliable implementations of eight representative intrinsic reward algorithms. It used to be challenging to compare intrinsic reward algorithms due to various confounding factors, including distinct implementations, optimization strategies, and evaluation methodologies. Therefore, RLeXplore is designed to provide unified and standardized procedures for constructing, computing, and optimizing intrinsic reward modules.

The workflow of RLeXplore is illustrated as follows:

Installation

  • with pip recommended

Open a terminal and install rllte with pip:

pip install rllte-core
  • with git

Open a terminal and clone the repository from GitHub with git:

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

After that, run the following command to install package and dependencies:

pip install -e . 

Finally, you can invoke the intrinsic reward module by:

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

Module List

Type Modules
Count-based PseudoCounts, RND, E3B
Curiosity-driven ICM, Disagreement, RIDE
Memory-based NGU
Information theory-based RE3

Tutorials

Click the following links to get the code notebook:

  1. Quick Start
  2. RLeXplore with RLLTE
  3. RLeXplore with Stable-Baselines3
  4. RLeXplore with CleanRL
  5. Custom Intrinsic Rewards

About

RLeXplore provides stable baselines of exploration methods in reinforcement learning, such as intrinsic curiosity module (ICM), random network distillation (RND) and rewarding impact-driven exploration (RIDE).

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published