Skill-Aware Diffusion for Generalizable Robotic Manipulation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Huang, Aoshen, Chen, Jiaming, Cheng, Jiyu, Song, Ran, Pan, Wei, Zhang, Wei
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866908770601271296
author Huang, Aoshen
Chen, Jiaming
Cheng, Jiyu
Song, Ran
Pan, Wei
Zhang, Wei
author_facet Huang, Aoshen
Chen, Jiaming
Cheng, Jiyu
Song, Ran
Pan, Wei
Zhang, Wei
contents Robust generalization in robotic manipulation is crucial for robots to adapt flexibly to diverse environments. Existing methods usually improve generalization by scaling data and networks, but model tasks independently and overlook skill-level information. Observing that tasks within the same skill share similar motion patterns, we propose Skill-Aware Diffusion (SADiff), which explicitly incorporates skill-level information to improve generalization. SADiff learns skill-specific representations through a skill-aware encoding module with learnable skill tokens, and conditions a skill-constrained diffusion model to generate object-centric motion flow. A skill-retrieval transformation strategy further exploits skill-specific trajectory priors to refine the mapping from 2D motion flow to executable 3D actions. Furthermore, we introduce IsaacSkill, a high-fidelity dataset containing fundamental robotic skills for comprehensive evaluation and sim-to-real transfer. Experiments in simulation and real-world settings show that SADiff achieves good performance and generalization across various manipulation tasks. Code, data, and videos are available at https://sites.google.com/view/sa-diff.
format Preprint
id arxiv_https___arxiv_org_abs_2601_11266
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Skill-Aware Diffusion for Generalizable Robotic Manipulation
Huang, Aoshen
Chen, Jiaming
Cheng, Jiyu
Song, Ran
Pan, Wei
Zhang, Wei
Robotics
Robust generalization in robotic manipulation is crucial for robots to adapt flexibly to diverse environments. Existing methods usually improve generalization by scaling data and networks, but model tasks independently and overlook skill-level information. Observing that tasks within the same skill share similar motion patterns, we propose Skill-Aware Diffusion (SADiff), which explicitly incorporates skill-level information to improve generalization. SADiff learns skill-specific representations through a skill-aware encoding module with learnable skill tokens, and conditions a skill-constrained diffusion model to generate object-centric motion flow. A skill-retrieval transformation strategy further exploits skill-specific trajectory priors to refine the mapping from 2D motion flow to executable 3D actions. Furthermore, we introduce IsaacSkill, a high-fidelity dataset containing fundamental robotic skills for comprehensive evaluation and sim-to-real transfer. Experiments in simulation and real-world settings show that SADiff achieves good performance and generalization across various manipulation tasks. Code, data, and videos are available at https://sites.google.com/view/sa-diff.
title Skill-Aware Diffusion for Generalizable Robotic Manipulation
topic Robotics
url https://arxiv.org/abs/2601.11266