xTED: Cross-Domain Adaptation via Diffusion-Based Trajectory Editing

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Niu, Haoyi, Chen, Qimao, Liu, Tenglong, Li, Jianxiong, Zhou, Guyue, Zhang, Yi, Hu, Jianming, Zhan, Xianyuan
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866912949397880832
author Niu, Haoyi
Chen, Qimao
Liu, Tenglong
Li, Jianxiong
Zhou, Guyue
Zhang, Yi
Hu, Jianming
Zhan, Xianyuan
author_facet Niu, Haoyi
Chen, Qimao
Liu, Tenglong
Li, Jianxiong
Zhou, Guyue
Zhang, Yi
Hu, Jianming
Zhan, Xianyuan
contents Reusing pre-collected data from different domains is an appealing solution for decision-making tasks, especially when data in the target domain are limited. Existing cross-domain policy transfer methods mostly aim at learning domain correspondences or corrections to facilitate policy learning, such as learning task/domain-specific discriminators, representations, or policies. This design philosophy often results in heavy model architectures or task/domain-specific modeling, lacking flexibility. This reality makes us wonder: can we directly bridge the domain gaps universally at the data level, instead of relying on complex downstream cross-domain policy transfer procedures? In this study, we propose the Cross-Domain Trajectory EDiting (xTED) framework that employs a specially designed diffusion model for cross-domain trajectory adaptation. Our proposed model architecture effectively captures the intricate dependencies among states, actions, and rewards, as well as the dynamics patterns within target data. Edited by adding noises and denoising with the pre-trained diffusion model, source domain trajectories can be transformed to align with target domain properties while preserving original semantic information. This process effectively corrects underlying domain gaps, enhancing state realism and dynamics reliability in source data, and allowing flexible integration with various single-domain and cross-domain downstream policy learning methods. Despite its simplicity, xTED demonstrates superior performance in extensive simulation and real-robot experiments.
format Preprint
id arxiv_https___arxiv_org_abs_2409_08687
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle xTED: Cross-Domain Adaptation via Diffusion-Based Trajectory Editing
Niu, Haoyi
Chen, Qimao
Liu, Tenglong
Li, Jianxiong
Zhou, Guyue
Zhang, Yi
Hu, Jianming
Zhan, Xianyuan
Robotics
Machine Learning
Reusing pre-collected data from different domains is an appealing solution for decision-making tasks, especially when data in the target domain are limited. Existing cross-domain policy transfer methods mostly aim at learning domain correspondences or corrections to facilitate policy learning, such as learning task/domain-specific discriminators, representations, or policies. This design philosophy often results in heavy model architectures or task/domain-specific modeling, lacking flexibility. This reality makes us wonder: can we directly bridge the domain gaps universally at the data level, instead of relying on complex downstream cross-domain policy transfer procedures? In this study, we propose the Cross-Domain Trajectory EDiting (xTED) framework that employs a specially designed diffusion model for cross-domain trajectory adaptation. Our proposed model architecture effectively captures the intricate dependencies among states, actions, and rewards, as well as the dynamics patterns within target data. Edited by adding noises and denoising with the pre-trained diffusion model, source domain trajectories can be transformed to align with target domain properties while preserving original semantic information. This process effectively corrects underlying domain gaps, enhancing state realism and dynamics reliability in source data, and allowing flexible integration with various single-domain and cross-domain downstream policy learning methods. Despite its simplicity, xTED demonstrates superior performance in extensive simulation and real-robot experiments.
title xTED: Cross-Domain Adaptation via Diffusion-Based Trajectory Editing
topic Robotics
Machine Learning
url https://arxiv.org/abs/2409.08687