Generative Trajectory Stitching through Diffusion Composition

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Luo, Yunhao, Mishra, Utkarsh A., Du, Yilun, Xu, Danfei
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913818775388160
author Luo, Yunhao
Mishra, Utkarsh A.
Du, Yilun
Xu, Danfei
author_facet Luo, Yunhao
Mishra, Utkarsh A.
Du, Yilun
Xu, Danfei
contents Effective trajectory stitching for long-horizon planning is a significant challenge in robotic decision-making. While diffusion models have shown promise in planning, they are limited to solving tasks similar to those seen in their training data. We propose CompDiffuser, a novel generative approach that can solve new tasks by learning to compositionally stitch together shorter trajectory chunks from previously seen tasks. Our key insight is modeling the trajectory distribution by subdividing it into overlapping chunks and learning their conditional relationships through a single bidirectional diffusion model. This allows information to propagate between segments during generation, ensuring physically consistent connections. We conduct experiments on benchmark tasks of various difficulties, covering different environment sizes, agent state dimension, trajectory types, training data quality, and show that CompDiffuser significantly outperforms existing methods.
format Preprint
id arxiv_https___arxiv_org_abs_2503_05153
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Generative Trajectory Stitching through Diffusion Composition
Luo, Yunhao
Mishra, Utkarsh A.
Du, Yilun
Xu, Danfei
Robotics
Artificial Intelligence
Machine Learning
Effective trajectory stitching for long-horizon planning is a significant challenge in robotic decision-making. While diffusion models have shown promise in planning, they are limited to solving tasks similar to those seen in their training data. We propose CompDiffuser, a novel generative approach that can solve new tasks by learning to compositionally stitch together shorter trajectory chunks from previously seen tasks. Our key insight is modeling the trajectory distribution by subdividing it into overlapping chunks and learning their conditional relationships through a single bidirectional diffusion model. This allows information to propagate between segments during generation, ensuring physically consistent connections. We conduct experiments on benchmark tasks of various difficulties, covering different environment sizes, agent state dimension, trajectory types, training data quality, and show that CompDiffuser significantly outperforms existing methods.
title Generative Trajectory Stitching through Diffusion Composition
topic Robotics
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2503.05153