Equivariant Diffusion Policy

Abstract

Recent work has shown diffusion models are an effective approach to learning the multimodal distributions arising from demonstration data in behavior cloning. However, a drawback of this approach is the need to learn a denoising function, which is significantly more complex than learning an explicit policy. In this work, we propose Equivariant Diffusion Policy, a novel diffusion policy learning method that leverages domain symmetries to obtain better sample efficiency and generalization in the denoising function. We theoretically analyze the \(\mathrm\{SO\}(2)\) symmetry of full 6-DoF control and characterize when a diffusion model is \(\mathrm\{SO\}(2)\)-equivariant. We furthermore evaluate the method empirically on a set of 12 simulation tasks in MimicGen, and show that it obtains a success rate that is, on average, 21.9% higher than the baseline Diffusion Policy. We also evaluate the method on a real-world system to show that effective policies can be learned with relatively few tra

Stats

• citations0
• S2 citations—
• github stars0
• HF likes0
• heat score0.00
• arxiv keywang2024equivariant

Related papers

• Streaming Diffusion Policy: Fast Policy Synthesis With Variable Noise Diffusion Models (2024)0.00
• Don't Start From Scratch: Behavioral Refinement Via Interpolant-based Policy Diffusion (2024)9.28
• Diffusion Policy Through Conditional Proximal Policy Optimization (2026)0.00
• Goal-conditioned Imitation Learning Using Score-based Diffusion Policies (2023)13.74
• Policy-guided Diffusion (2024)0.00
• Dichotomous Diffusion Policy Optimization (2025)0.00
• Policy Representation Via Diffusion Probability Model For Reinforcement Learning (2023)0.00
• Sparse Diffusion Policy: A Sparse, Reusable, And Flexible Policy For Robot Learning (2024)0.00