SeMOPO: Learning High-quality Model and Policy from Low-quality Offline Visual Datasets

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wan, Shenghua, Chen, Ziyuan, Gan, Le, Feng, Shuai, Zhan, De-Chuan
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929385511059456
author Wan, Shenghua
Chen, Ziyuan
Gan, Le
Feng, Shuai
Zhan, De-Chuan
author_facet Wan, Shenghua
Chen, Ziyuan
Gan, Le
Feng, Shuai
Zhan, De-Chuan
contents Model-based offline reinforcement Learning (RL) is a promising approach that leverages existing data effectively in many real-world applications, especially those involving high-dimensional inputs like images and videos. To alleviate the distribution shift issue in offline RL, existing model-based methods heavily rely on the uncertainty of learned dynamics. However, the model uncertainty estimation becomes significantly biased when observations contain complex distractors with non-trivial dynamics. To address this challenge, we propose a new approach - \emph{Separated Model-based Offline Policy Optimization} (SeMOPO) - decomposing latent states into endogenous and exogenous parts via conservative sampling and estimating model uncertainty on the endogenous states only. We provide a theoretical guarantee of model uncertainty and performance bound of SeMOPO. To assess the efficacy, we construct the Low-Quality Vision Deep Data-Driven Datasets for RL (LQV-D4RL), where the data are collected by non-expert policy and the observations include moving distractors. Experimental results show that our method substantially outperforms all baseline methods, and further analytical experiments validate the critical designs in our method. The project website is \href{https://sites.google.com/view/semopo}{https://sites.google.com/view/semopo}.
format Preprint
id arxiv_https___arxiv_org_abs_2406_09486
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle SeMOPO: Learning High-quality Model and Policy from Low-quality Offline Visual Datasets
Wan, Shenghua
Chen, Ziyuan
Gan, Le
Feng, Shuai
Zhan, De-Chuan
Computer Vision and Pattern Recognition
Artificial Intelligence
Model-based offline reinforcement Learning (RL) is a promising approach that leverages existing data effectively in many real-world applications, especially those involving high-dimensional inputs like images and videos. To alleviate the distribution shift issue in offline RL, existing model-based methods heavily rely on the uncertainty of learned dynamics. However, the model uncertainty estimation becomes significantly biased when observations contain complex distractors with non-trivial dynamics. To address this challenge, we propose a new approach - \emph{Separated Model-based Offline Policy Optimization} (SeMOPO) - decomposing latent states into endogenous and exogenous parts via conservative sampling and estimating model uncertainty on the endogenous states only. We provide a theoretical guarantee of model uncertainty and performance bound of SeMOPO. To assess the efficacy, we construct the Low-Quality Vision Deep Data-Driven Datasets for RL (LQV-D4RL), where the data are collected by non-expert policy and the observations include moving distractors. Experimental results show that our method substantially outperforms all baseline methods, and further analytical experiments validate the critical designs in our method. The project website is \href{https://sites.google.com/view/semopo}{https://sites.google.com/view/semopo}.
title SeMOPO: Learning High-quality Model and Policy from Low-quality Offline Visual Datasets
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2406.09486