ObstaDiff: Generalizable Diffusion Policy Learning via Obstacle-aware Representations

arXiv:2609.10918v1 Announce Type: cross Abstract: Imitation learning has achieved impressive results in robotic manipulation, yet most existing approaches assume clean backgrounds and lack explicit mechanisms for obstacle-aware motion generation. Extending such policies to cluttered, real-world scenes with unstructured obstacles remains a key generalization challenge. We present ObstaDiff, a decomposed diffusion-policy framework with a lightweight obstacle-aware visual encoder. ObstaDiff extracts a structured target-obstacle-background representation, enabling the downstream alignment policy to generate end-effector trajectories toward a target-centered bottleneck pose while reasoning about surrounding obstacles. We evaluate ObstaDiff on 61 real-robot greenhouse trials per method (366 executions in total). ObstaDiff achieves 75.41% average task success and 8.20% average obstacle collision rate, outperforming representative imitation-learning baselines and improving generalization in cluttered agricultural scenes.

Related

• Training-Free Imitation Learning with Closed-Form Diffusion Policies
• SpatialVAM:Spatial-Aware Multi-View Video Diffusion as a Data-Efficient Robot Policy
• Expressivity and Statistical Trade-offs in Diffusion Policy Learning
• TMRL: Diffusion Timestep-Modulated Pretraining Enables Exploration for Efficient Policy Finetuning

Source: arXiv cs.LG | 2026-09-11