Adaptive Diffusion Constrained Sampling for Bimanual Robot Manipulation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tong, Haolei, Zhang, Yuezhe, Lueth, Sophie, Chalvatzaki, Georgia
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917292962480128
author Tong, Haolei
Zhang, Yuezhe
Lueth, Sophie
Chalvatzaki, Georgia
author_facet Tong, Haolei
Zhang, Yuezhe
Lueth, Sophie
Chalvatzaki, Georgia
contents Coordinated multi-arm manipulation requires satisfying multiple simultaneous geometric constraints across high-dimensional configuration spaces, which poses a significant challenge for traditional planning and control methods. In this work, we propose Adaptive Diffusion Constrained Sampling (ADCS), a generative framework that flexibly integrates both equality (e.g., relative and absolute pose constraints) and structured inequality constraints (e.g., proximity to object surfaces) into an energy-based diffusion model. Equality constraints are modeled using dedicated energy networks trained on pose differences in Lie algebra space, while inequality constraints are represented via Signed Distance Functions (SDFs) and encoded into learned constraint embeddings, allowing the model to reason about complex spatial regions. A key innovation of our method is a Transformer-based architecture that learns to weight constraint-specific energy functions at inference time, enabling flexible and context-aware constraint integration. Moreover, we adopt a two-phase sampling strategy that improves precision and sample diversity by combining Langevin dynamics with resampling and density-aware re-weighting. Experimental results on dual-arm manipulation tasks show that ADCS significantly improves sample diversity and generalization across settings demanding precise coordination and adaptive constraint handling.
format Preprint
id arxiv_https___arxiv_org_abs_2505_13667
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Adaptive Diffusion Constrained Sampling for Bimanual Robot Manipulation
Tong, Haolei
Zhang, Yuezhe
Lueth, Sophie
Chalvatzaki, Georgia
Robotics
Coordinated multi-arm manipulation requires satisfying multiple simultaneous geometric constraints across high-dimensional configuration spaces, which poses a significant challenge for traditional planning and control methods. In this work, we propose Adaptive Diffusion Constrained Sampling (ADCS), a generative framework that flexibly integrates both equality (e.g., relative and absolute pose constraints) and structured inequality constraints (e.g., proximity to object surfaces) into an energy-based diffusion model. Equality constraints are modeled using dedicated energy networks trained on pose differences in Lie algebra space, while inequality constraints are represented via Signed Distance Functions (SDFs) and encoded into learned constraint embeddings, allowing the model to reason about complex spatial regions. A key innovation of our method is a Transformer-based architecture that learns to weight constraint-specific energy functions at inference time, enabling flexible and context-aware constraint integration. Moreover, we adopt a two-phase sampling strategy that improves precision and sample diversity by combining Langevin dynamics with resampling and density-aware re-weighting. Experimental results on dual-arm manipulation tasks show that ADCS significantly improves sample diversity and generalization across settings demanding precise coordination and adaptive constraint handling.
title Adaptive Diffusion Constrained Sampling for Bimanual Robot Manipulation
topic Robotics
url https://arxiv.org/abs/2505.13667