Equivariant Descriptor Fields: SE(3)-Equivariant Energy-Based Models for End-to-End Visual Robotic Manipulation Learning
Not on the map. Why?
Abstract (arXiv)
End-to-end learning for visual robotic manipulation is known to suffer from sample inefficiency, requiring large numbers of demonstrations. The spatial roto-translation equivariance, or the SE(3)-equivariance can be exploited to improve the sample efficiency for learning robotic manipulation. In this paper, we present SE(3)-equivariant models for visual robotic manipulation from point clouds that can be trained fully end-to-end. By utilizing the representation theory of the Lie group, we construct novel SE(3)-equivariant energy-based models that allow highly sample efficient end-to-end learning. We show that our models can learn from scratch without prior knowledge and yet are highly sample efficient (5~10 demonstrations are enough). Furthermore, we show that our models can generalize to tasks with (i) previously unseen target object poses, (ii) previously unseen target object instances of the category, and (iii) previously unseen visual distractors. We experiment with 6-DoF robotic manipulation tasks to validate our models' sample efficiency and generalizability. Codes are available at: https://github.com/tomato1mule/edf
Key facts. Cited by 56 papers in the map (data through 14 Oct 2026). Builds on 36 of its 85 references that are in the map. First posted on arXiv 16 Jun 2022.
Builds on
36 of 85 references.
References · 36
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2013Auto-Encoding Variational Bayescited by 4,053
- 2018Dynamic Graph CNN for Learning on Point Cloudscited by 2,223
- 2018Relational inductive biases, deep learning, and graph networkscited by 1,360
- 2016Group Equivariant Convolutional Networkscited by 1,165
- 2015End-to-End Training of Deep Visuomotor Policiescited by 875
- 2018Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point cloudscited by 746
- 2018QT-Opt: Scalable Deep Reinforcement Learning for Vision-Based Robotic Manipulationcited by 743
- 2020SE(3)-Transformers: 3D Roto-Translation Equivariant Attention Networkscited by 607
- 2016Stein Variational Gradient Descent: A General Purpose Bayesian Inference Algorithmcited by 507
- 2021Implicit Behavioral Cloningcited by 452
- 2020Transporter Networks: Rearranging the Visual World for Robotic Manipulationcited by 344
24 more
- 2020Point-GNN: Graph Neural Network for 3D Object Detection in a Point Cloudcited by 335
- 2021Vector Neurons: A General Framework for SO(3)-Equivariant Networkscited by 305
- 2021Neural Descriptor Fields: SE(3)-Equivariant Object Representations for Manipulationcited by 209
- 2016A Theory of Generative ConvNetcited by 174
- 2018Dense Object Nets: Learning Dense Visual Object Descriptors By and For Robotic Manipulationcited by 142
- 2019Unsupervised Learning of Object Keypoints for Perception and Controlcited by 109
- 2019Self-supervised 6D Object Pose Estimation for Robot Manipulationcited by 101
- 2022SO(2)-Equivariant Reinforcement Learningcited by 84
- 2021Independent SE(3)-Equivariant Models for End-to-End Rigid Protein Dockingcited by 82
- 2020Learning to Rearrange Deformable Cables, Fabrics, and Bags with Goal-Conditioned Transporter Networkscited by 78
- 2016Multi-view Self-supervised Deep Learning for 6D Pose Estimation in the Amazon Picking Challengecited by 70
- 2018RGCNN: Regularized Graph CNN for Point Cloud Segmentationcited by 70
- 2016Cooperative Training of Descriptor and Generator Networkscited by 60
- 2020Generative PointNet: Deep Energy-Based Learning on Unordered Point Sets for 3D Generation, Reconstruction and Classificationcited by 57
- 2018Learning Descriptor Networks for 3D Shape Synthesis and Analysiscited by 56
- 2022Equivariant Transporter Networkcited by 46
- 2021Leveraging SE(3) Equivariance for Self-Supervised Category-Level Object Pose Estimationcited by 45
- 2022A Tale of Two Flows: Cooperative Learning of Langevin Flow and Normalizing Flow Toward Energy-Based Modelcited by 44
- 2020Learning Energy-Based Model with Variational Auto-Encoder as Amortized Samplercited by 43
- 2022Convergence of the Riemannian Langevin Algorithmcited by 25
- 2016Learning to Draw Samples: With Application to Amortized MLE for Generative Adversarial Learningcited by 21
- 2019Energy-Based Continuous Inverse Optimal Controlcited by 14
- 2021Learning Equivariant Energy Based Models with Equivariant Stein Variational Gradient Descentcited by 10
- 2021Hierarchical Primitive Composition: Simultaneous Activation of Skills with Inconsistent Action Dimensions in Multiple Hierarchiescited by 3
Built on by
Citing papers per year: 2022: 1; 2023: 13; 2024: 19; 2025: 16; 2026: 7.
56 papers.
Sort:
By area:
- 2023Learning to Act from Actionless Videos through Dense Correspondencescited by 164
- 2024EquiBot: SIM(3)-Equivariant Diffusion Policy for Generalizable and Data Efficient Learningcited by 75
- 2024Equivariant Diffusion Policycited by 66
- 2023Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulationcited by 52
- 2022SE(3)-Equivariant Relational Rearrangement with Neural Descriptor Fieldscited by 51
- 2025Diffusion Models for Robotic Manipulation: A Surveycited by 41
- 2023EquivAct: SIM(3)-Equivariant Visuomotor Policies beyond Rigid Object Manipulationcited by 37
- 2024A Survey of Embodied Learning for Object-Centric Robotic Manipulationcited by 35
- 2024Fourier Transporter: Bi-Equivariant Robotic Manipulation in 3Dcited by 33
- 2024HardNet: Hard-Constrained Neural Networks with Universal Approximation Guaranteescited by 31
- 2023Leveraging SE(3) Equivariance for Learning 3D Geometric Shape Assemblycited by 28
- 2024ActionFlow: Equivariant, Accurate, and Efficient Policies with Spatially Symmetric Flow Matchingcited by 26
38 more
- 2023SparseDFF: Sparse-View Feature Distillation for One-Shot Dexterous Manipulationcited by 22
- 2023Local Neural Descriptor Fields: Locally Conditioned Object Representations for Manipulationcited by 21
- 2023A General Theory of Correct, Incorrect, and Extrinsic Equivariancecited by 19
- 2023Banana: Banach Fixed-Point Network for Pointcloud Segmentation with Inter-Part Equivariancecited by 19
- 2023EFEM: Equivariant Neural Field Expectation Maximization for 3D Object Segmentation Without Scene Supervisioncited by 18
- 2024Imagination Policy: Using Generative Point Cloud Models for Learning Manipulation Policiescited by 18
- 2023One-shot Imitation Learning via Interaction Warpingcited by 16
- 2025A Survey of Robotic Navigation and Manipulation with Physics Simulators in the Era of Embodied AIcited by 16
- 2024Leveraging Locality to Boost Sample Efficiency in Robotic Manipulationcited by 13
- 2024RiEMann: Near Real-Time SE(3)-Equivariant Robot Manipulation without Point Cloud Segmentationcited by 12
- 2025AnyPlace: Learning Generalized Object Placement for Robot Manipulationcited by 11
- 2023NeuSE: Neural SE(3)-Equivariant Embedding for Long-Term Object-Based Simultaneous Localization and Mappingcited by 10
- 2025SE(3)-Equivariant Robot Learning and Control: A Tutorial Surveycited by 10
- 2023Denoising Heat-inspired Diffusion with Insulators for Collision Free Motion Planningcited by 8
- 2024EqvAfford: SE(3) Equivariance for Point-Level Affordance Learningcited by 6
- 2024MATCH POLICY: A Simple Pipeline from Point Cloud Registration to Manipulation Policiescited by 6
- 2024SE(3) Equivariant Ray Embeddings for Implicit Multi-View Depth Estimationcited by 6
- 2025Constraint-Preserving Data Generation for Visuomotor Policy Learningcited by 6
- 2023Unified framework for diffusion generative models in SO(3): applications in computer vision and astrophysicscited by 5
- 2024Multi-task real-robot data with gaze attention for dual-arm fine manipulationcited by 5
- 2024Equivariant Offline Reinforcement Learningcited by 5
- 2024Neural Attention Field: Emerging Point Relevance in 3D Scenes for One-Shot Dexterous Graspingcited by 5
- 2025A Practical Guide for Incorporating Symmetry in Diffusion Policycited by 5
- 2024Constrained 6-DoF Grasp Generation on Complex Shapes for Improved Dual-Arm Manipulationcited by 4
- 2024SE(3)-bi-equivariant Transformers for Point Cloud Assemblycited by 4
- 2025ADPro: a Test-time Adaptive Diffusion Policy via Manifold-constrained Denoising and Task-aware Initialization for Robotic Manipulationcited by 4
- 2025EfficientFlow: Efficient Equivariant Flow Policy Learning for Embodied AIcited by 4
- 2025Partially Equivariant Reinforcement Learning in Symmetry-Breaking Environmentscited by 3
- 2025MatrixNet: Learning over symmetry groups using learned group representationscited by 2
- 2025Coarse-to-Fine 3D Keyframe Transportercited by 2
- 2025Equivariant Flow Matching for Point Cloud Assemblycited by 2
- 2025Enforcing convex constraints in Graph Neural Networkscited by 2
- 2024BiEquiFormer: Bi-Equivariant Representations for Global Point Cloud Registrationcited by 1
- 2025Equivariant Volumetric Graspingcited by 1
- 2025Symmetry-Aware Steering of Equivariant Diffusion Policies: Benefits and Limitscited by 1
- 2024SPLIT: SE(3)-diffusion via Local Geometry-based Score Prediction for 3D Scene-to-Pose-Set Matching Problemscited by 0
- 2025Multi-Group Equivariant Augmentation for Reinforcement Learning in Robot Manipulationcited by 0
- 2026Group-Invariant Unsupervised Skill Discovery: Symmetry-aware Skill Representations for Generalizable Behaviorcited by 0
50 most cited of 56 All 56 (JSON)
- 2026Phase-Conditioned Affordance Fields for SE(3)-Equivariant Visuomotor Policies12 Oct 2026
- 2026Equivariant Visual-Tactile Diffusion Policy for Contact-Rich Manipulation2 Oct 2026
- 2026EquivDP3: A SIM(3)-Invariant Point-Cloud Encoder for Data-Efficient Humanoid Loco-Manipulation29 Sep 2026
- 2026Pix2Act: Image-Space Manipulation Policies with Equivariant Augmentation13 Jul 2026
- 2026Robustness of Robotic Manipulation: Foundations and Frontiers30 Jun 2026
- 2026Beyond World-Frame Action Heads: Motion-Centric Action Frames for Vision-Language-Action Models12 May 2026
- 2026Group-Invariant Unsupervised Skill Discovery: Symmetry-aware Skill Representations for Generalizable Behavior20 Jan 2026
- 2025Symmetry-Aware Steering of Equivariant Diffusion Policies: Benefits and Limits12 Dec 2025
- 2025EfficientFlow: Efficient Equivariant Flow Policy Learning for Embodied AI1 Dec 2025
- 2025Partially Equivariant Reinforcement Learning in Symmetry-Breaking Environments30 Nov 2025
- 2025Enforcing convex constraints in Graph Neural Networks13 Oct 2025
- 2025Multi-Group Equivariant Augmentation for Reinforcement Learning in Robot Manipulation15 Aug 2025
38 more
- 2025ADPro: a Test-time Adaptive Diffusion Policy via Manifold-constrained Denoising and Task-aware Initialization for Robotic Manipulation8 Aug 2025
- 2025Constraint-Preserving Data Generation for Visuomotor Policy Learning5 Aug 2025
- 2025Equivariant Volumetric Grasping24 Jul 2025
- 2025Equivariant Flow Matching for Point Cloud Assembly24 May 2025
- 2025A Practical Guide for Incorporating Symmetry in Diffusion Policy19 May 2025
- 2025A Survey of Robotic Navigation and Manipulation with Physics Simulators in the Era of Embodied AI1 May 2025
- 2025Diffusion Models for Robotic Manipulation: A Survey11 Apr 2025
- 2025SE(3)-Equivariant Robot Learning and Control: A Tutorial Survey12 Mar 2025
- 2025AnyPlace: Learning Generalized Object Placement for Robot Manipulation6 Feb 2025
- 2025Coarse-to-Fine 3D Keyframe Transporter3 Feb 2025
- 2025MatrixNet: Learning over symmetry groups using learned group representations16 Jan 2025
- 2024SPLIT: SE(3)-diffusion via Local Geometry-based Score Prediction for 3D Scene-to-Pose-Set Matching Problems15 Nov 2024
- 2024SE(3) Equivariant Ray Embeddings for Implicit Multi-View Depth Estimation11 Nov 2024
- 2024Neural Attention Field: Emerging Point Relevance in 3D Scenes for One-Shot Dexterous Grasping30 Oct 2024
- 2024HardNet: Hard-Constrained Neural Networks with Universal Approximation Guarantees14 Oct 2024
- 2024MATCH POLICY: A Simple Pipeline from Point Cloud Registration to Manipulation Policies23 Sep 2024
- 2024ActionFlow: Equivariant, Accurate, and Efficient Policies with Spatially Symmetric Flow Matching6 Sep 2024
- 2024A Survey of Embodied Learning for Object-Centric Robotic Manipulation21 Aug 2024
- 2024EqvAfford: SE(3) Equivariance for Point-Level Affordance Learning4 Aug 2024
- 2024SE(3)-bi-equivariant Transformers for Point Cloud Assembly12 Jul 2024
- 2024BiEquiFormer: Bi-Equivariant Representations for Global Point Cloud Registration11 Jul 2024
- 2024Equivariant Diffusion Policy1 Jul 2024
- 2024EquiBot: SIM(3)-Equivariant Diffusion Policy for Generalizable and Data Efficient Learning1 Jul 2024
- 2024Equivariant Offline Reinforcement Learning20 Jun 2024
- 2024Imagination Policy: Using Generative Point Cloud Models for Learning Manipulation Policies17 Jun 2024
- 2024Leveraging Locality to Boost Sample Efficiency in Robotic Manipulation15 Jun 2024
- 2024Constrained 6-DoF Grasp Generation on Complex Shapes for Improved Dual-Arm Manipulation6 Apr 2024
- 2024RiEMann: Near Real-Time SE(3)-Equivariant Robot Manipulation without Point Cloud Segmentation28 Mar 2024
- 2024Fourier Transporter: Bi-Equivariant Robotic Manipulation in 3D22 Jan 2024
- 2024Multi-task real-robot data with gaze attention for dual-arm fine manipulation15 Jan 2024
- 2023Unified framework for diffusion generative models in SO(3): applications in computer vision and astrophysics18 Dec 2023
- 2023SparseDFF: Sparse-View Feature Distillation for One-Shot Dexterous Manipulation25 Oct 2023
- 2023EquivAct: SIM(3)-Equivariant Visuomotor Policies beyond Rigid Object Manipulation24 Oct 2023
- 2023Denoising Heat-inspired Diffusion with Insulators for Collision Free Motion Planning19 Oct 2023
- 2023Learning to Act from Actionless Videos through Dense Correspondences12 Oct 2023
- 2023Leveraging SE(3) Equivariance for Learning 3D Geometric Shape Assembly13 Sep 2023
- 2023Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulation6 Sep 2023
- 2023One-shot Imitation Learning via Interaction Warping21 Jun 2023
50 newest of 56 All 56 (JSON)
- 2024EquiBot: SIM(3)-Equivariant Diffusion Policy for Generalizable and Data Efficient Learningcited by 75
- 2024Equivariant Diffusion Policycited by 66
- 2023Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulationcited by 52
- 2025Diffusion Models for Robotic Manipulation: A Surveycited by 41
- 2024ActionFlow: Equivariant, Accurate, and Efficient Policies with Spatially Symmetric Flow Matchingcited by 26
- 2023Denoising Heat-inspired Diffusion with Insulators for Collision Free Motion Planningcited by 8
- 2023Unified framework for diffusion generative models in SO(3): applications in computer vision and astrophysicscited by 5
- 2025A Practical Guide for Incorporating Symmetry in Diffusion Policycited by 5
- 2025ADPro: a Test-time Adaptive Diffusion Policy via Manifold-constrained Denoising and Task-aware Initialization for Robotic Manipulationcited by 4
- 2025Equivariant Flow Matching for Point Cloud Assemblycited by 2
- 2025Symmetry-Aware Steering of Equivariant Diffusion Policies: Benefits and Limitscited by 1
- 2024SPLIT: SE(3)-diffusion via Local Geometry-based Score Prediction for 3D Scene-to-Pose-Set Matching Problemscited by 0
12 most cited of 14 All 14 (JSON)
- 2026Equivariant Visual-Tactile Diffusion Policy for Contact-Rich Manipulation2 Oct 2026
- 2026EquivDP3: A SIM(3)-Invariant Point-Cloud Encoder for Data-Efficient Humanoid Loco-Manipulation29 Sep 2026
- 2025Symmetry-Aware Steering of Equivariant Diffusion Policies: Benefits and Limits12 Dec 2025
- 2025ADPro: a Test-time Adaptive Diffusion Policy via Manifold-constrained Denoising and Task-aware Initialization for Robotic Manipulation8 Aug 2025
- 2025Equivariant Flow Matching for Point Cloud Assembly24 May 2025
- 2025A Practical Guide for Incorporating Symmetry in Diffusion Policy19 May 2025
- 2025Diffusion Models for Robotic Manipulation: A Survey11 Apr 2025
- 2024SPLIT: SE(3)-diffusion via Local Geometry-based Score Prediction for 3D Scene-to-Pose-Set Matching Problems15 Nov 2024
- 2024ActionFlow: Equivariant, Accurate, and Efficient Policies with Spatially Symmetric Flow Matching6 Sep 2024
- 2024Equivariant Diffusion Policy1 Jul 2024
- 2024EquiBot: SIM(3)-Equivariant Diffusion Policy for Generalizable and Data Efficient Learning1 Jul 2024
- 2023Unified framework for diffusion generative models in SO(3): applications in computer vision and astrophysics18 Dec 2023
12 newest of 14 All 14 (JSON)
- 2023One-shot Imitation Learning via Interaction Warpingcited by 16
- 2024RiEMann: Near Real-Time SE(3)-Equivariant Robot Manipulation without Point Cloud Segmentationcited by 12
- 2025SE(3)-Equivariant Robot Learning and Control: A Tutorial Surveycited by 10
- 2024Multi-task real-robot data with gaze attention for dual-arm fine manipulationcited by 5
- 2024Equivariant Offline Reinforcement Learningcited by 5
- 2025Coarse-to-Fine 3D Keyframe Transportercited by 2
- 2025SE(3)-Equivariant Robot Learning and Control: A Tutorial Survey12 Mar 2025
- 2025Coarse-to-Fine 3D Keyframe Transporter3 Feb 2025
- 2024Equivariant Offline Reinforcement Learning20 Jun 2024
- 2024RiEMann: Near Real-Time SE(3)-Equivariant Robot Manipulation without Point Cloud Segmentation28 Mar 2024
- 2024Multi-task real-robot data with gaze attention for dual-arm fine manipulation15 Jan 2024
- 2023One-shot Imitation Learning via Interaction Warping21 Jun 2023
- 2023Learning to Act from Actionless Videos through Dense Correspondencescited by 164
- 2022SE(3)-Equivariant Relational Rearrangement with Neural Descriptor Fieldscited by 51
- 2023EquivAct: SIM(3)-Equivariant Visuomotor Policies beyond Rigid Object Manipulationcited by 37
- 2024A Survey of Embodied Learning for Object-Centric Robotic Manipulationcited by 35
- 2024Fourier Transporter: Bi-Equivariant Robotic Manipulation in 3Dcited by 33
- 2024HardNet: Hard-Constrained Neural Networks with Universal Approximation Guaranteescited by 31
- 2023Leveraging SE(3) Equivariance for Learning 3D Geometric Shape Assemblycited by 28
- 2023Local Neural Descriptor Fields: Locally Conditioned Object Representations for Manipulationcited by 21
- 2023A General Theory of Correct, Incorrect, and Extrinsic Equivariancecited by 19
- 2023Banana: Banach Fixed-Point Network for Pointcloud Segmentation with Inter-Part Equivariancecited by 19
- 2023EFEM: Equivariant Neural Field Expectation Maximization for 3D Object Segmentation Without Scene Supervisioncited by 18
- 2024Imagination Policy: Using Generative Point Cloud Models for Learning Manipulation Policiescited by 18
12 most cited of 32 All 32 (JSON)
- 2026Phase-Conditioned Affordance Fields for SE(3)-Equivariant Visuomotor Policies12 Oct 2026
- 2026Pix2Act: Image-Space Manipulation Policies with Equivariant Augmentation13 Jul 2026
- 2026Robustness of Robotic Manipulation: Foundations and Frontiers30 Jun 2026
- 2026Group-Invariant Unsupervised Skill Discovery: Symmetry-aware Skill Representations for Generalizable Behavior20 Jan 2026
- 2025EfficientFlow: Efficient Equivariant Flow Policy Learning for Embodied AI1 Dec 2025
- 2025Enforcing convex constraints in Graph Neural Networks13 Oct 2025
- 2025Constraint-Preserving Data Generation for Visuomotor Policy Learning5 Aug 2025
- 2025Equivariant Volumetric Grasping24 Jul 2025
- 2025A Survey of Robotic Navigation and Manipulation with Physics Simulators in the Era of Embodied AI1 May 2025
- 2025AnyPlace: Learning Generalized Object Placement for Robot Manipulation6 Feb 2025
- 2025MatrixNet: Learning over symmetry groups using learned group representations16 Jan 2025
- 2024SE(3) Equivariant Ray Embeddings for Implicit Multi-View Depth Estimation11 Nov 2024
12 newest of 32 All 32 (JSON)