Saved in:
Bibliographic Details
Main Author: He, Shenghong
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2502.06491
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909600597409792
author He, Shenghong
author_facet He, Shenghong
contents Model-based offline reinforcement learning (MORL) aims to learn a policy by exploiting a dynamics model derived from an existing dataset. Applying conservative quantification to the dynamics model, most existing works on MORL generate trajectories that approximate the real data distribution to facilitate policy learning by using current information (e.g., the state and action at time step $t$). However, these works neglect the impact of historical information on environmental dynamics, leading to the generation of unreliable trajectories that may not align with the real data distribution. In this paper, we propose a new MORL algorithm \textbf{R}eliability-guaranteed \textbf{T}ransformer (RT), which can eliminate unreliable trajectories by calculating the cumulative reliability of the generated trajectory (i.e., using a weighted variational distance away from the real data). Moreover, by sampling candidate actions with high rewards, RT can efficiently generate high-return trajectories from the existing offline data. We theoretically prove the performance guarantees of RT in policy learning, and empirically demonstrate its effectiveness against state-of-the-art model-based methods on several benchmark tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2502_06491
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Model-Based Offline Reinforcement Learning with Reliability-Guaranteed Sequence Modeling
He, Shenghong
Machine Learning
Artificial Intelligence
Model-based offline reinforcement learning (MORL) aims to learn a policy by exploiting a dynamics model derived from an existing dataset. Applying conservative quantification to the dynamics model, most existing works on MORL generate trajectories that approximate the real data distribution to facilitate policy learning by using current information (e.g., the state and action at time step $t$). However, these works neglect the impact of historical information on environmental dynamics, leading to the generation of unreliable trajectories that may not align with the real data distribution. In this paper, we propose a new MORL algorithm \textbf{R}eliability-guaranteed \textbf{T}ransformer (RT), which can eliminate unreliable trajectories by calculating the cumulative reliability of the generated trajectory (i.e., using a weighted variational distance away from the real data). Moreover, by sampling candidate actions with high rewards, RT can efficiently generate high-return trajectories from the existing offline data. We theoretically prove the performance guarantees of RT in policy learning, and empirically demonstrate its effectiveness against state-of-the-art model-based methods on several benchmark tasks.
title Model-Based Offline Reinforcement Learning with Reliability-Guaranteed Sequence Modeling
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2502.06491