Reinforcement Learning from Delayed Observations via World Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Karamzade, Armin, Kim, Kyungmin, Kalsi, Montek, Fox, Roy
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910501885181952
author Karamzade, Armin
Kim, Kyungmin
Kalsi, Montek
Fox, Roy
author_facet Karamzade, Armin
Kim, Kyungmin
Kalsi, Montek
Fox, Roy
contents In standard reinforcement learning settings, agents typically assume immediate feedback about the effects of their actions after taking them. However, in practice, this assumption may not hold true due to physical constraints and can significantly impact the performance of learning algorithms. In this paper, we address observation delays in partially observable environments. We propose leveraging world models, which have shown success in integrating past observations and learning dynamics, to handle observation delays. By reducing delayed POMDPs to delayed MDPs with world models, our methods can effectively handle partial observability, where existing approaches achieve sub-optimal performance or degrade quickly as observability decreases. Experiments suggest that one of our methods can outperform a naive model-based approach by up to 250%. Moreover, we evaluate our methods on visual delayed environments, for the first time showcasing delay-aware reinforcement learning continuous control with visual observations.
format Preprint
id arxiv_https___arxiv_org_abs_2403_12309
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Reinforcement Learning from Delayed Observations via World Models
Karamzade, Armin
Kim, Kyungmin
Kalsi, Montek
Fox, Roy
Machine Learning
Artificial Intelligence
In standard reinforcement learning settings, agents typically assume immediate feedback about the effects of their actions after taking them. However, in practice, this assumption may not hold true due to physical constraints and can significantly impact the performance of learning algorithms. In this paper, we address observation delays in partially observable environments. We propose leveraging world models, which have shown success in integrating past observations and learning dynamics, to handle observation delays. By reducing delayed POMDPs to delayed MDPs with world models, our methods can effectively handle partial observability, where existing approaches achieve sub-optimal performance or degrade quickly as observability decreases. Experiments suggest that one of our methods can outperform a naive model-based approach by up to 250%. Moreover, we evaluate our methods on visual delayed environments, for the first time showcasing delay-aware reinforcement learning continuous control with visual observations.
title Reinforcement Learning from Delayed Observations via World Models
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2403.12309