Saved in:
Bibliographic Details
Main Authors: He, Shuncheng, Zhang, Hongchang, Shao, Jianzhun, Jiang, Yuhang, Ji, Xiangyang
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2506.19643
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909659059716096
author He, Shuncheng
Zhang, Hongchang
Shao, Jianzhun
Jiang, Yuhang
Ji, Xiangyang
author_facet He, Shuncheng
Zhang, Hongchang
Shao, Jianzhun
Jiang, Yuhang
Ji, Xiangyang
contents Offline reinforcement learning (RL) recently gains growing interests from RL researchers. However, the performance of offline RL suffers from the out-of-distribution problem, which can be corrected by feedback in online RL. Previous offline RL research focuses on restricting the offline algorithm in in-distribution even in-sample action sampling. In contrast, fewer work pays attention to the influence of the batch data. In this paper, we first build a bridge over the batch data and the performance of offline RL algorithms theoretically, from the perspective of model-based offline RL optimization. We draw a conclusion that, with mild assumptions, the distance between the state-action pair distribution generated by the behavioural policy and the distribution generated by the optimal policy, accounts for the performance gap between the policy learned by model-based offline RL and the optimal policy. Secondly, we reveal that in task-agnostic settings, a series of policies trained by unsupervised RL can minimize the worst-case regret in the performance gap. Inspired by the theoretical conclusions, UDG (Unsupervised Data Generation) is devised to generate data and select proper data for offline training under tasks-agnostic settings. Empirical results demonstrate that UDG can outperform supervised data generation on solving unknown tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2506_19643
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Unsupervised Data Generation for Offline Reinforcement Learning: A Perspective from Model
He, Shuncheng
Zhang, Hongchang
Shao, Jianzhun
Jiang, Yuhang
Ji, Xiangyang
Machine Learning
Offline reinforcement learning (RL) recently gains growing interests from RL researchers. However, the performance of offline RL suffers from the out-of-distribution problem, which can be corrected by feedback in online RL. Previous offline RL research focuses on restricting the offline algorithm in in-distribution even in-sample action sampling. In contrast, fewer work pays attention to the influence of the batch data. In this paper, we first build a bridge over the batch data and the performance of offline RL algorithms theoretically, from the perspective of model-based offline RL optimization. We draw a conclusion that, with mild assumptions, the distance between the state-action pair distribution generated by the behavioural policy and the distribution generated by the optimal policy, accounts for the performance gap between the policy learned by model-based offline RL and the optimal policy. Secondly, we reveal that in task-agnostic settings, a series of policies trained by unsupervised RL can minimize the worst-case regret in the performance gap. Inspired by the theoretical conclusions, UDG (Unsupervised Data Generation) is devised to generate data and select proper data for offline training under tasks-agnostic settings. Empirical results demonstrate that UDG can outperform supervised data generation on solving unknown tasks.
title Unsupervised Data Generation for Offline Reinforcement Learning: A Perspective from Model
topic Machine Learning
url https://arxiv.org/abs/2506.19643