S-RL Toolbox: Environments, Datasets and Evaluation Metrics for State Representation Learning
Antonin Raffin, Ashley Hill, René Traoré, Timothée Lesort, Natalia Díaz-Rodríguez, David Filliat
Code Available — Be the first to reproduce this paper.
ReproduceCode
- github.com/araffin/robotics-rl-srlOfficialIn papernone★ 0
- github.com/eric-erki/robotics-rl-srlnone★ 0
- github.com/kalifou/robotics-rl-srlnone★ 0
- github.com/araffin/srl-zoopytorch★ 0
Abstract
State representation learning aims at learning compact representations from raw observations in robotics and control applications. Approaches used for this objective are auto-encoders, learning forward models, inverse dynamics or learning using generic priors on the state characteristics. However, the diversity in applications and methods makes the field lack standard evaluation datasets, metrics and tasks. This paper provides a set of environments, data generators, robotic control tasks, metrics and tools to facilitate iterative state representation learning and evaluation in reinforcement learning settings.