Dynamics-Guided Diffusion Model
for Robot Manipulator Design
Task-specific Designs without Task-specific Training
We present Dynamics-Guided Diffusion Model, a data-driven framework for generating manipulator geometry designs for a given manipulation task. Instead of training different design models for each task, our approach employs a learned dynamics network shared across tasks. For a new manipulation task, we first decompose it into a collection of individual motion targets which we call target interaction profile, where each individual motion can be modeled by the shared dynamics network. The design objective constructed from the target and predicted interaction profiles provides a gradient to guide the refinement of finger geometry for the task. This refinement process is executed as a classifier-guided diffusion process, where the design objective acts as the classifier guidance. We evaluate our framework on various manipulation tasks, under the sensor-less setting using only an open-loop parallel jaw motion. Our generated designs outperform optimization-based and unguided diffusion baselines relatively by 31.5% and 45.3% on average manipulation success rate. With the ability to generate a design within 0.8 seconds, our framework could facilitate rapid design iteration and enhance the adoption of data-driven approaches for robotic mechanism design.
Paper
Latest version: arXiv or here.
Code
Team
1 Stanford University 2 Columbia University
Technical Summary Video (with audio)
Results
Pose ConvergenceThe goal of pose convergence is to design fingers that always reorient a target object to a specified orientation when closing the gripper in parallel. As you can imagine, this manipulator can be quite useful in industrial settings such as assembly lines. When objects are fed in with different poses we can automatically align them to the same pose. Then the following robot can just blindly manipulate the object.
More examples of pose convergence manipulators in the real world:
Rotate
Shift Up/Down
Shift Left/Right
Multi-object Results
Our framework also allows designing manipulators for a set of objects to achieve a task.Rotate Clockwise
BibTeX
@article{xu2024dynamics,
title={Dynamics-Guided Diffusion Model for Robot Manipulator Design},
author={Xu, Xiaomeng and Ha, Huy and Song, Shuran},
journal={arXiv preprint arXiv:2402.15038},
year={2024}
}
Contact
If you have any questions, please feel free to contact Xiaomeng Xu.