Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

SkillDiffuser: Interpretable Hierarchical Planning via Skill Abstractions in Diffusion-Based Task Execution

About

Diffusion models have demonstrated strong potential for robotic trajectory planning. However, generating coherent trajectories from high-level instructions remains challenging, especially for long-range composition tasks requiring multiple sequential skills. We propose SkillDiffuser, an end-to-end hierarchical planning framework integrating interpretable skill learning with conditional diffusion planning to address this problem. At the higher level, the skill abstraction module learns discrete, human-understandable skill representations from visual observations and language instructions. These learned skill embeddings are then used to condition the diffusion model to generate customized latent trajectories aligned with the skills. This allows generating diverse state trajectories that adhere to the learnable skills. By integrating skill learning with conditional trajectory generation, SkillDiffuser produces coherent behavior following abstract instructions across diverse tasks. Experiments on multi-task robotic manipulation benchmarks like Meta-World and LOReL demonstrate state-of-the-art performance and human-interpretable skill representations from SkillDiffuser. More visualization results and information could be found on our website.

Zhixuan Liang, Yao Mu, Hengbo Ma, Masayoshi Tomizuka, Mingyu Ding, Ping Luo• 2023

Related benchmarks

TaskDatasetResultRank
Robotic ManipulationCalvin ABCD→D
Success Rate (1 Inst)94.4
26
Dexterous Hand ControlAdroit
Overall Avg Success Rate63
13
Robotic ManipulationLOReL Sawyer Dataset (test)
Close Drawer Success Rate9.532
6
Robotic ManipulationMimicGen (test)
Square Success Rate75
6
Robotic ManipulationMimicGen
Square75
6
Robot ManipulationLOReL Sawyer (unseen multi-step composition)
Success Rate25.21
4
Language-conditioned robot task executionLOReL Sawyer seen rephrasals
Success Rate43.65
3
Language-conditioned robot task executionLOReL Sawyer unseen noun rephrasals
Success Rate36.01
3
Language-conditioned robot task executionLOReL Sawyer unseen verb rephrasals
Success Rate0.367
3
Language-conditioned robot task executionLOReL Sawyer unseen noun + verb rephrasals
Success Rate42.02
3
Showing 10 of 13 rows

Other info

Code

Follow for update