Skip to content

Implement Meta-RL REPTILE algorithm #321

@zxnga

Description

@zxnga

Hello !

I would like to propose a meta-RL extension for the repository to be able to use the REPTILE meta-learning algorithm (https://arxiv.org/pdf/1803.02999) along with SB3 algorithms that would act as the inner loop agents.

Motivation:

Currently, there is only few and outdated options for model-free Meta-RL. In particular:

  • Many Meta-RL implementations are either research-only or focused on supervised learning experimentations that make them hard to be easily transfered to RL tasks
  • MAML was removed from RLlib
  • SB3 already has the core inner loop algorithms which simplify the development of some of the meta RL algorithms

Specifically i already implemented an SB3 extension for the REPTILE algorithm for my thesis on a separate repo (https://github.com/zxnga/reptile_sb3) that could also easily be extended to the FOMAML version of the algorithm.

If the idea aligns with the project’s, I’d be happy to open a draft PR as i already have a clean workable solution the reproduces the results of the paper on supervised tasks.

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions