Skill or Luck? Return Decomposition via Advantage Functions

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Pan, Hsiao-Ru, Schölkopf, Bernhard
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866911780227252224
author Pan, Hsiao-Ru
Schölkopf, Bernhard
author_facet Pan, Hsiao-Ru
Schölkopf, Bernhard
contents Learning from off-policy data is essential for sample-efficient reinforcement learning. In the present work, we build on the insight that the advantage function can be understood as the causal effect of an action on the return, and show that this allows us to decompose the return of a trajectory into parts caused by the agent's actions (skill) and parts outside of the agent's control (luck). Furthermore, this decomposition enables us to naturally extend Direct Advantage Estimation (DAE) to off-policy settings (Off-policy DAE). The resulting method can learn from off-policy trajectories without relying on importance sampling techniques or truncating off-policy actions. We draw connections between Off-policy DAE and previous methods to demonstrate how it can speed up learning and when the proposed off-policy corrections are important. Finally, we use the MinAtar environments to illustrate how ignoring off-policy corrections can lead to suboptimal policy optimization performance.
format Preprint
id arxiv_https___arxiv_org_abs_2402_12874
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Skill or Luck? Return Decomposition via Advantage Functions
Pan, Hsiao-Ru
Schölkopf, Bernhard
Machine Learning
Learning from off-policy data is essential for sample-efficient reinforcement learning. In the present work, we build on the insight that the advantage function can be understood as the causal effect of an action on the return, and show that this allows us to decompose the return of a trajectory into parts caused by the agent's actions (skill) and parts outside of the agent's control (luck). Furthermore, this decomposition enables us to naturally extend Direct Advantage Estimation (DAE) to off-policy settings (Off-policy DAE). The resulting method can learn from off-policy trajectories without relying on importance sampling techniques or truncating off-policy actions. We draw connections between Off-policy DAE and previous methods to demonstrate how it can speed up learning and when the proposed off-policy corrections are important. Finally, we use the MinAtar environments to illustrate how ignoring off-policy corrections can lead to suboptimal policy optimization performance.
title Skill or Luck? Return Decomposition via Advantage Functions
topic Machine Learning
url https://arxiv.org/abs/2402.12874