Bridging State and History Representations: Understanding Self-Predictive RL

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Ni, Tianwei, Eysenbach, Benjamin, Seyedsalehi, Erfan, Ma, Michel, Gehring, Clement, Mahajan, Aditya, Bacon, Pierre-Luc
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866914763592695808
author Ni, Tianwei
Eysenbach, Benjamin
Seyedsalehi, Erfan
Ma, Michel
Gehring, Clement
Mahajan, Aditya
Bacon, Pierre-Luc
author_facet Ni, Tianwei
Eysenbach, Benjamin
Seyedsalehi, Erfan
Ma, Michel
Gehring, Clement
Mahajan, Aditya
Bacon, Pierre-Luc
contents Representations are at the core of all deep reinforcement learning (RL) methods for both Markov decision processes (MDPs) and partially observable Markov decision processes (POMDPs). Many representation learning methods and theoretical frameworks have been developed to understand what constitutes an effective representation. However, the relationships between these methods and the shared properties among them remain unclear. In this paper, we show that many of these seemingly distinct methods and frameworks for state and history abstractions are, in fact, based on a common idea of self-predictive abstraction. Furthermore, we provide theoretical insights into the widely adopted objectives and optimization, such as the stop-gradient technique, in learning self-predictive representations. These findings together yield a minimalist algorithm to learn self-predictive representations for states and histories. We validate our theories by applying our algorithm to standard MDPs, MDPs with distractors, and POMDPs with sparse rewards. These findings culminate in a set of preliminary guidelines for RL practitioners.
format Preprint
id arxiv_https___arxiv_org_abs_2401_08898
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Bridging State and History Representations: Understanding Self-Predictive RL
Ni, Tianwei
Eysenbach, Benjamin
Seyedsalehi, Erfan
Ma, Michel
Gehring, Clement
Mahajan, Aditya
Bacon, Pierre-Luc
Machine Learning
Artificial Intelligence
Representations are at the core of all deep reinforcement learning (RL) methods for both Markov decision processes (MDPs) and partially observable Markov decision processes (POMDPs). Many representation learning methods and theoretical frameworks have been developed to understand what constitutes an effective representation. However, the relationships between these methods and the shared properties among them remain unclear. In this paper, we show that many of these seemingly distinct methods and frameworks for state and history abstractions are, in fact, based on a common idea of self-predictive abstraction. Furthermore, we provide theoretical insights into the widely adopted objectives and optimization, such as the stop-gradient technique, in learning self-predictive representations. These findings together yield a minimalist algorithm to learn self-predictive representations for states and histories. We validate our theories by applying our algorithm to standard MDPs, MDPs with distractors, and POMDPs with sparse rewards. These findings culminate in a set of preliminary guidelines for RL practitioners.
title Bridging State and History Representations: Understanding Self-Predictive RL
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2401.08898