Epistemically-guided forward-backward exploration

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Urpí, Núria Armengol, Vlastelica, Marin, Martius, Georg, Coros, Stelian
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866911044653285376
author Urpí, Núria Armengol
Vlastelica, Marin
Martius, Georg
Coros, Stelian
author_facet Urpí, Núria Armengol
Vlastelica, Marin
Martius, Georg
Coros, Stelian
contents Zero-shot reinforcement learning is necessary for extracting optimal policies in absence of concrete rewards for fast adaptation to future problem settings. Forward-backward representations (FB) have emerged as a promising method for learning optimal policies in absence of rewards via a factorization of the policy occupancy measure. However, up until now, FB and many similar zero-shot reinforcement learning algorithms have been decoupled from the exploration problem, generally relying on other exploration algorithms for data collection. We argue that FB representations should fundamentally be used for exploration in order to learn more efficiently. With this goal in mind, we design exploration policies that arise naturally from the FB representation that minimize the posterior variance of the FB representation, hence minimizing its epistemic uncertainty. We empirically demonstrate that such principled exploration strategies improve sample complexity of the FB algorithm considerably in comparison to other exploration methods. Code is publicly available at https://sites.google.com/view/fbee-url.
format Preprint
id arxiv_https___arxiv_org_abs_2507_05477
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Epistemically-guided forward-backward exploration
Urpí, Núria Armengol
Vlastelica, Marin
Martius, Georg
Coros, Stelian
Machine Learning
Artificial Intelligence
Zero-shot reinforcement learning is necessary for extracting optimal policies in absence of concrete rewards for fast adaptation to future problem settings. Forward-backward representations (FB) have emerged as a promising method for learning optimal policies in absence of rewards via a factorization of the policy occupancy measure. However, up until now, FB and many similar zero-shot reinforcement learning algorithms have been decoupled from the exploration problem, generally relying on other exploration algorithms for data collection. We argue that FB representations should fundamentally be used for exploration in order to learn more efficiently. With this goal in mind, we design exploration policies that arise naturally from the FB representation that minimize the posterior variance of the FB representation, hence minimizing its epistemic uncertainty. We empirically demonstrate that such principled exploration strategies improve sample complexity of the FB algorithm considerably in comparison to other exploration methods. Code is publicly available at https://sites.google.com/view/fbee-url.
title Epistemically-guided forward-backward exploration
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2507.05477