TEA: Trajectory Encoding Augmentation for Robust and Transferable Policies in Offline Reinforcement Learning

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Ormancı, Batıkan Bora, Swazinna, Phillip, Udluft, Steffen, Runkler, Thomas A.
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866912204994904064
author Ormancı, Batıkan Bora
Swazinna, Phillip
Udluft, Steffen
Runkler, Thomas A.
author_facet Ormancı, Batıkan Bora
Swazinna, Phillip
Udluft, Steffen
Runkler, Thomas A.
contents In this paper, we investigate offline reinforcement learning (RL) with the goal of training a single robust policy that generalizes effectively across environments with unseen dynamics. We propose a novel approach, Trajectory Encoding Augmentation (TEA), which extends the state space by integrating latent representations of environmental dynamics obtained from sequence encoders, such as AutoEncoders. Our findings show that incorporating these encodings with TEA improves the transferability of a single policy to novel environments with new dynamics, surpassing methods that rely solely on unmodified states. These results indicate that TEA captures critical, environment-specific characteristics, enabling RL agents to generalize effectively across dynamic conditions.
format Preprint
id arxiv_https___arxiv_org_abs_2411_19133
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TEA: Trajectory Encoding Augmentation for Robust and Transferable Policies in Offline Reinforcement Learning
Ormancı, Batıkan Bora
Swazinna, Phillip
Udluft, Steffen
Runkler, Thomas A.
Machine Learning
In this paper, we investigate offline reinforcement learning (RL) with the goal of training a single robust policy that generalizes effectively across environments with unseen dynamics. We propose a novel approach, Trajectory Encoding Augmentation (TEA), which extends the state space by integrating latent representations of environmental dynamics obtained from sequence encoders, such as AutoEncoders. Our findings show that incorporating these encodings with TEA improves the transferability of a single policy to novel environments with new dynamics, surpassing methods that rely solely on unmodified states. These results indicate that TEA captures critical, environment-specific characteristics, enabling RL agents to generalize effectively across dynamic conditions.
title TEA: Trajectory Encoding Augmentation for Robust and Transferable Policies in Offline Reinforcement Learning
topic Machine Learning
url https://arxiv.org/abs/2411.19133