Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Allen, Cameron, Kirtland, Aaron, Tao, Ruo Yu, Lobel, Sam, Scott, Daniel, Petrocelli, Nicholas, Gottesman, Omer, Parr, Ronald, Littman, Michael L., Konidaris, George
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866915021350502400
author Allen, Cameron
Kirtland, Aaron
Tao, Ruo Yu
Lobel, Sam
Scott, Daniel
Petrocelli, Nicholas
Gottesman, Omer
Parr, Ronald
Littman, Michael L.
Konidaris, George
author_facet Allen, Cameron
Kirtland, Aaron
Tao, Ruo Yu
Lobel, Sam
Scott, Daniel
Petrocelli, Nicholas
Gottesman, Omer
Parr, Ronald
Littman, Michael L.
Konidaris, George
contents Reinforcement learning algorithms typically rely on the assumption that the environment dynamics and value function can be expressed in terms of a Markovian state representation. However, when state information is only partially observable, how can an agent learn such a state representation, and how can it detect when it has found one? We introduce a metric that can accomplish both objectives, without requiring access to -- or knowledge of -- an underlying, unobservable state space. Our metric, the $λ$-discrepancy, is the difference between two distinct temporal difference (TD) value estimates, each computed using TD($λ$) with a different value of $λ$. Since TD($λ{=}0$) makes an implicit Markov assumption and TD($λ{=}1$) does not, a discrepancy between these estimates is a potential indicator of a non-Markovian state representation. Indeed, we prove that the $λ$-discrepancy is exactly zero for all Markov decision processes and almost always non-zero for a broad class of partially observable environments. We also demonstrate empirically that, once detected, minimizing the $λ$-discrepancy can help with learning a memory function to mitigate the corresponding partial observability. We then train a reinforcement learning agent that simultaneously constructs two recurrent value networks with different $λ$ parameters and minimizes the difference between them as an auxiliary loss. The approach scales to challenging partially observable domains, where the resulting agent frequently performs significantly better (and never performs worse) than a baseline recurrent agent with only a single value network.
format Preprint
id arxiv_https___arxiv_org_abs_2407_07333
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy
Allen, Cameron
Kirtland, Aaron
Tao, Ruo Yu
Lobel, Sam
Scott, Daniel
Petrocelli, Nicholas
Gottesman, Omer
Parr, Ronald
Littman, Michael L.
Konidaris, George
Machine Learning
Artificial Intelligence
Reinforcement learning algorithms typically rely on the assumption that the environment dynamics and value function can be expressed in terms of a Markovian state representation. However, when state information is only partially observable, how can an agent learn such a state representation, and how can it detect when it has found one? We introduce a metric that can accomplish both objectives, without requiring access to -- or knowledge of -- an underlying, unobservable state space. Our metric, the $λ$-discrepancy, is the difference between two distinct temporal difference (TD) value estimates, each computed using TD($λ$) with a different value of $λ$. Since TD($λ{=}0$) makes an implicit Markov assumption and TD($λ{=}1$) does not, a discrepancy between these estimates is a potential indicator of a non-Markovian state representation. Indeed, we prove that the $λ$-discrepancy is exactly zero for all Markov decision processes and almost always non-zero for a broad class of partially observable environments. We also demonstrate empirically that, once detected, minimizing the $λ$-discrepancy can help with learning a memory function to mitigate the corresponding partial observability. We then train a reinforcement learning agent that simultaneously constructs two recurrent value networks with different $λ$ parameters and minimizes the difference between them as an auxiliary loss. The approach scales to challenging partially observable domains, where the resulting agent frequently performs significantly better (and never performs worse) than a baseline recurrent agent with only a single value network.
title Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2407.07333