Saved in:
Bibliographic Details
Main Authors: Li, Shangzhe, Zhang, Xinhua
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2402.00807
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914963068551168
author Li, Shangzhe
Zhang, Xinhua
author_facet Li, Shangzhe
Zhang, Xinhua
contents Batch offline data have been shown considerably beneficial for reinforcement learning. Their benefit is further amplified by upsampling with generative models. In this paper, we consider a novel opportunity where interaction with environment is feasible, but only restricted to observations, i.e., \textit{no reward} feedback is available. This setting is broadly applicable, as simulators or even real cyber-physical systems are often accessible, while in contrast reward is often difficult or expensive to obtain. As a result, the learner must make good sense of the offline data to synthesize an efficient scheme of querying the transition of state. Our method first leverages online interactions to generate high-return trajectories via conditional diffusion models. They are then blended with the original offline trajectories through a stitching algorithm, and the resulting augmented data can be applied generically to downstream reinforcement learners. Superior empirical performance is demonstrated over state-of-the-art data augmentation methods that are extended to utilize state-only interactions.
format Preprint
id arxiv_https___arxiv_org_abs_2402_00807
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Augmenting Offline Reinforcement Learning with State-only Interactions
Li, Shangzhe
Zhang, Xinhua
Machine Learning
Batch offline data have been shown considerably beneficial for reinforcement learning. Their benefit is further amplified by upsampling with generative models. In this paper, we consider a novel opportunity where interaction with environment is feasible, but only restricted to observations, i.e., \textit{no reward} feedback is available. This setting is broadly applicable, as simulators or even real cyber-physical systems are often accessible, while in contrast reward is often difficult or expensive to obtain. As a result, the learner must make good sense of the offline data to synthesize an efficient scheme of querying the transition of state. Our method first leverages online interactions to generate high-return trajectories via conditional diffusion models. They are then blended with the original offline trajectories through a stitching algorithm, and the resulting augmented data can be applied generically to downstream reinforcement learners. Superior empirical performance is demonstrated over state-of-the-art data augmentation methods that are extended to utilize state-only interactions.
title Augmenting Offline Reinforcement Learning with State-only Interactions
topic Machine Learning
url https://arxiv.org/abs/2402.00807