No Regrets: Investigating and Improving Regret Approximations for Curriculum Discovery

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Rutherford, Alexander, Beukman, Michael, Willi, Timon, Lacerda, Bruno, Hawes, Nick, Foerster, Jakob
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910677686288384
author Rutherford, Alexander
Beukman, Michael
Willi, Timon
Lacerda, Bruno
Hawes, Nick
Foerster, Jakob
author_facet Rutherford, Alexander
Beukman, Michael
Willi, Timon
Lacerda, Bruno
Hawes, Nick
Foerster, Jakob
contents What data or environments to use for training to improve downstream performance is a longstanding and very topical question in reinforcement learning. In particular, Unsupervised Environment Design (UED) methods have gained recent attention as their adaptive curricula promise to enable agents to be robust to in- and out-of-distribution tasks. This work investigates how existing UED methods select training environments, focusing on task prioritisation metrics. Surprisingly, despite methods aiming to maximise regret in theory, the practical approximations do not correlate with regret but with success rate. As a result, a significant portion of an agent's experience comes from environments it has already mastered, offering little to no contribution toward enhancing its abilities. Put differently, current methods fail to predict intuitive measures of ``learnability.'' Specifically, they are unable to consistently identify those scenarios that the agent can sometimes solve, but not always. Based on our analysis, we develop a method that directly trains on scenarios with high learnability. This simple and intuitive approach outperforms existing UED methods in several binary-outcome environments, including the standard domain of Minigrid and a novel setting closely inspired by a real-world robotics problem. We further introduce a new adversarial evaluation procedure for directly measuring robustness, closely mirroring the conditional value at risk (CVaR). We open-source all our code and present visualisations of final policies here: https://github.com/amacrutherford/sampling-for-learnability.
format Preprint
id arxiv_https___arxiv_org_abs_2408_15099
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle No Regrets: Investigating and Improving Regret Approximations for Curriculum Discovery
Rutherford, Alexander
Beukman, Michael
Willi, Timon
Lacerda, Bruno
Hawes, Nick
Foerster, Jakob
Machine Learning
Artificial Intelligence
Robotics
What data or environments to use for training to improve downstream performance is a longstanding and very topical question in reinforcement learning. In particular, Unsupervised Environment Design (UED) methods have gained recent attention as their adaptive curricula promise to enable agents to be robust to in- and out-of-distribution tasks. This work investigates how existing UED methods select training environments, focusing on task prioritisation metrics. Surprisingly, despite methods aiming to maximise regret in theory, the practical approximations do not correlate with regret but with success rate. As a result, a significant portion of an agent's experience comes from environments it has already mastered, offering little to no contribution toward enhancing its abilities. Put differently, current methods fail to predict intuitive measures of ``learnability.'' Specifically, they are unable to consistently identify those scenarios that the agent can sometimes solve, but not always. Based on our analysis, we develop a method that directly trains on scenarios with high learnability. This simple and intuitive approach outperforms existing UED methods in several binary-outcome environments, including the standard domain of Minigrid and a novel setting closely inspired by a real-world robotics problem. We further introduce a new adversarial evaluation procedure for directly measuring robustness, closely mirroring the conditional value at risk (CVaR). We open-source all our code and present visualisations of final policies here: https://github.com/amacrutherford/sampling-for-learnability.
title No Regrets: Investigating and Improving Regret Approximations for Curriculum Discovery
topic Machine Learning
Artificial Intelligence
Robotics
url https://arxiv.org/abs/2408.15099