Shelving, Stacking, Hanging: Relational Pose Diffusion for Multi-modal Rearrangement
Not on the map. Why?
Abstract (arXiv)
We propose a system for rearranging objects in a scene to achieve a desired object-scene placing relationship, such as a book inserted in an open slot of a bookshelf. The pipeline generalizes to novel geometries, poses, and layouts of both scenes and objects, and is trained from demonstrations to operate directly on 3D point clouds. Our system overcomes challenges associated with the existence of many geometrically-similar rearrangement solutions for a given scene. By leveraging an iterative pose de-noising training procedure, we can fit multi-modal demonstration data and produce multi-modal outputs while remaining precise and accurate. We also show the advantages of conditioning on relevant local geometric features while ignoring irrelevant global structure that harms both generalization and precision. We demonstrate our approach on three distinct rearrangement tasks that require handling multi-modality and generalization over object shape and pose in both simulation and the real world. Project website, code, and videos: https://anthonysimeonov.github.io/rpdiff-multi-modal/
Key facts. Cited by 29 papers in the map (data through 7 Oct 2026). Builds on 75 of its 94 references that are in the map. First posted on arXiv 10 Jul 2023.
Builds on
75 of 94 references.
References · 75
- 2017Attention Is All You Needcited by 59,932
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2016PointNet: Deep Learning on Point Sets for 3D Classification and Segmentationcited by 5,223
- 2017PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Spacecited by 4,384
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 2,994
- 2015ShapeNet: An Information-Rich 3D Model Repositorycited by 2,880
- 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusioncited by 2,730
- 2018Dynamic Graph CNN for Learning on Point Cloudscited by 2,220
- 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,339
- 2018On the Continuity of Rotation Representations in Neural Networkscited by 1,186
38 more
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,023
- 2020Convolutional Occupancy Networkscited by 760
- 2021Diffusion Probabilistic Models for 3D Point Cloud Generationcited by 734
- 2022Point-E: A System for Generating 3D Point Clouds from Complex Promptscited by 694
- 2020WaveGrad: Estimating Gradients for Waveform Generationcited by 620
- 2022Perceiver-Actor: A Multi-Task Transformer for Robotic Manipulationcited by 545
- 2021Neural Fields in Visual Computing and Beyondcited by 539
- 2022Is Conditional Generative Modeling all you need for Decision-Making?cited by 504
- 2021Implicit Behavioral Cloningcited by 442
- 2021RMA: Rapid Motor Adaptation for Legged Robotscited by 430
- 2020Transporter Networks: Rearranging the Visual World for Robotic Manipulationcited by 342
- 2021Vector Neurons: A General Framework for SO(3)-Equivariant Networkscited by 304
- 2022Behavior Transformers: Cloning k modes with one stonecited by 299
- 2020CosyPose: Consistent multi-view multi-object 6D pose estimationcited by 287
- 2018DeepIM: Deep Iterative Matching for 6D Pose Estimationcited by 283
- 2022Cold Diffusion: Inverting Arbitrary Image Transforms Without Noisecited by 275
- 2021How to Train Your Energy-Based Modelscited by 271
- 20196-DOF GraspNet: Variational Grasp Generation for Object Manipulationcited by 237
- 2023Diffusion-based Generation, Optimization, and Planning in 3D Scenescited by 237
- 2023Reduce, Reuse, Recycle: Compositional Generation with Energy-Based Diffusion Models and MCMCcited by 216
- 2021Contact-GraspNet: Efficient 6-DoF Grasp Generation in Cluttered Scenescited by 213
- 2021Neural Descriptor Fields: SE(3)-Equivariant Object Representations for Manipulationcited by 204
- 2022MegaPose: 6D Pose Estimation of Novel Objects via Render & Comparecited by 198
- 2023On the Importance of Noise Scheduling for Diffusion Modelscited by 193
- 2023Inversion by Direct Iteration: An Alternative to Denoising Diffusion for Image Restorationcited by 164
- 2021A System for General In-Hand Object Re-Orientationcited by 155
- 2020Rearrangement: A Challenge for Embodied AIcited by 153
- 2019kPAM: KeyPoint Affordances for Category-Level Robotic Manipulationcited by 152
- 2022SE(3)-DiffusionFields: Learning smooth cost functions for joint grasp and motion optimization through diffusioncited by 125
- 2017Robotic Pick-and-Place of Novel Objects in Clutter with Multi-Affordance Grasping and Cross-Domain Image Matchingcited by 124
- 2015Human Pose Estimation with Iterative Error Feedbackcited by 121
- 2021Coarse-to-Fine Q-attention: Efficient Learning for Visual Robotic Manipulation via Discretisationcited by 120
- 2018A micro Lie theory for state estimation in roboticscited by 117
- 2019Self-Supervised Correspondence in Visuomotor Policy Learningcited by 106
- 2020Normalization Techniques in Training DNNs: Methodology, Analysis and Applicationcited by 90
- 2021Q-attention: Enabling Efficient Learning for Vision-based Robotic Manipulationcited by 85
- 2023LEGO-Net: Learning Regular Rearrangements of Objects in Roomscited by 75
- 2021Implicit-PDF: Non-Parametric Representation of Probability Distributions on the Rotation Manifoldcited by 65
50 most cited of 75 All 75 (JSON)
Built on by
Citing papers per year: 2023: 5; 2024: 16; 2025: 8.
29 papers.
Sort:
By area:
- 20243D Diffuser Actor: Policy Diffusion with 3D Scene Representationscited by 278
- 2025HybridVLA: Collaborative Diffusion and Autoregression in a Unified Vision-Language-Action Modelcited by 176
- 2024RVT-2: Learning Precise Manipulation from Few Demonstrationscited by 135
- 20243D Diffusion Policy: Generalizable Visuomotor Policy Learning via Simple 3D Representationscited by 115
- 2024Towards Synergistic, Generalized, and Efficient Dual-System for Robotic Manipulationcited by 71
- 2023Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulationcited by 51
- 2024DexDiffuser: Generating Dexterous Grasps with Diffusion Modelscited by 46
- 2025Diffusion Models for Robotic Manipulation: A Surveycited by 39
- 2024Scaling Diffusion Policy in Transformer to 1 Billion Parameters for Robotic Manipulationcited by 38
- 2024Deep Generative Models in Robotics: A Survey on Learning from Multimodal Demonstrationscited by 35
- 2024Imagination Policy: Using Generative Point Cloud Models for Learning Manipulation Policiescited by 18
- 2024Discrete Policy: Learning Disentangled Action Space for Multi-Task Robotic Manipulationcited by 17
17 more
- 2023Dream2Real: Zero-Shot 3D Object Rearrangement with Vision-Language Modelscited by 16
- 20243D-MVP: 3D Multiview Pretraining for Robotic Manipulationcited by 10
- 2025AnyPlace: Learning Generalized Object Placement for Robot Manipulationcited by 10
- 20254D Visual Pre-training for Robot Learningcited by 8
- 2023LVDiffusor: Distilling Functional Rearrangement Priors from Large Models into Diffusorcited by 7
- 2024MATCH POLICY: A Simple Pipeline from Point Cloud Registration to Manipulation Policiescited by 6
- 2023Out of Sight, Still in Mind: Reasoning and Planning about Unobserved Objects with Video Tracking Enabled Memory Modelscited by 5
- 20236-DoF Stability Field via Diffusion Modelscited by 2
- 2024Non-rigid Relative Placement through 3D Dense Diffusioncited by 2
- 2024Precise Pick-and-Place using Score-Based Diffusion Networkscited by 1
- 2024Implicit Contact Diffuser: Sequential Contact Reasoning with Latent Point Cloud Diffusioncited by 1
- 2024Particle-based 6D Object Pose Estimation from Point Clouds using Diffusion Modelscited by 1
- 2025Simultaneous Pick and Place Detection by Combining SE(3) Diffusion Models with Differential Kinematicscited by 1
- 2025GOPLA: Generalizable Object Placement Learning via Synthetic Augmentation of Human Arrangementcited by 1
- 2024DAP: Diffusion-based Affordance Prediction for Multi-modality Storagecited by 0
- 2025Generating Stable Placements via Physics-guided Diffusion Modelscited by 0
- 2025Placeit! A Framework for Learning Robot Object Placement Skillscited by 0
- 2025GOPLA: Generalizable Object Placement Learning via Synthetic Augmentation of Human Arrangement16 Oct 2025
- 2025Placeit! A Framework for Learning Robot Object Placement Skills10 Oct 2025
- 2025Generating Stable Placements via Physics-guided Diffusion Models25 Sep 2025
- 20254D Visual Pre-training for Robot Learning24 Aug 2025
- 2025Simultaneous Pick and Place Detection by Combining SE(3) Diffusion Models with Differential Kinematics28 Apr 2025
- 2025Diffusion Models for Robotic Manipulation: A Survey11 Apr 2025
- 2025HybridVLA: Collaborative Diffusion and Autoregression in a Unified Vision-Language-Action Model13 Mar 2025
- 2025AnyPlace: Learning Generalized Object Placement for Robot Manipulation6 Feb 2025
- 2024Particle-based 6D Object Pose Estimation from Point Clouds using Diffusion Models1 Dec 2024
- 2024Non-rigid Relative Placement through 3D Dense Diffusion25 Oct 2024
- 2024Implicit Contact Diffuser: Sequential Contact Reasoning with Latent Point Cloud Diffusion21 Oct 2024
- 2024Towards Synergistic, Generalized, and Efficient Dual-System for Robotic Manipulation10 Oct 2024
17 more
- 2024Discrete Policy: Learning Disentangled Action Space for Multi-Task Robotic Manipulation27 Sep 2024
- 2024MATCH POLICY: A Simple Pipeline from Point Cloud Registration to Manipulation Policies23 Sep 2024
- 2024Scaling Diffusion Policy in Transformer to 1 Billion Parameters for Robotic Manipulation22 Sep 2024
- 2024Precise Pick-and-Place using Score-Based Diffusion Networks15 Sep 2024
- 2024DAP: Diffusion-based Affordance Prediction for Multi-modality Storage31 Aug 2024
- 2024Deep Generative Models in Robotics: A Survey on Learning from Multimodal Demonstrations8 Aug 2024
- 20243D-MVP: 3D Multiview Pretraining for Robotic Manipulation26 Jun 2024
- 2024Imagination Policy: Using Generative Point Cloud Models for Learning Manipulation Policies17 Jun 2024
- 2024RVT-2: Learning Precise Manipulation from Few Demonstrations12 Jun 2024
- 20243D Diffusion Policy: Generalizable Visuomotor Policy Learning via Simple 3D Representations6 Mar 2024
- 20243D Diffuser Actor: Policy Diffusion with 3D Scene Representations16 Feb 2024
- 2024DexDiffuser: Generating Dexterous Grasps with Diffusion Models5 Feb 2024
- 2023Dream2Real: Zero-Shot 3D Object Rearrangement with Vision-Language Models7 Dec 2023
- 2023LVDiffusor: Distilling Functional Rearrangement Priors from Large Models into Diffusor3 Dec 2023
- 20236-DoF Stability Field via Diffusion Models26 Oct 2023
- 2023Out of Sight, Still in Mind: Reasoning and Planning about Unobserved Objects with Video Tracking Enabled Memory Models26 Sep 2023
- 2023Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulation6 Sep 2023
- 20243D Diffuser Actor: Policy Diffusion with 3D Scene Representationscited by 278
- 20243D Diffusion Policy: Generalizable Visuomotor Policy Learning via Simple 3D Representationscited by 115
- 2023Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulationcited by 51
- 2024DexDiffuser: Generating Dexterous Grasps with Diffusion Modelscited by 46
- 2025Diffusion Models for Robotic Manipulation: A Surveycited by 39
- 2024Scaling Diffusion Policy in Transformer to 1 Billion Parameters for Robotic Manipulationcited by 38
- 2024Discrete Policy: Learning Disentangled Action Space for Multi-Task Robotic Manipulationcited by 17
- 20254D Visual Pre-training for Robot Learningcited by 8
- 2023LVDiffusor: Distilling Functional Rearrangement Priors from Large Models into Diffusorcited by 7
- 20236-DoF Stability Field via Diffusion Modelscited by 2
- 2024Particle-based 6D Object Pose Estimation from Point Clouds using Diffusion Modelscited by 1
- 2025Simultaneous Pick and Place Detection by Combining SE(3) Diffusion Models with Differential Kinematicscited by 1
12 most cited of 13 All 13 (JSON)
- 2025Generating Stable Placements via Physics-guided Diffusion Models25 Sep 2025
- 20254D Visual Pre-training for Robot Learning24 Aug 2025
- 2025Simultaneous Pick and Place Detection by Combining SE(3) Diffusion Models with Differential Kinematics28 Apr 2025
- 2025Diffusion Models for Robotic Manipulation: A Survey11 Apr 2025
- 2024Particle-based 6D Object Pose Estimation from Point Clouds using Diffusion Models1 Dec 2024
- 2024Discrete Policy: Learning Disentangled Action Space for Multi-Task Robotic Manipulation27 Sep 2024
- 2024Scaling Diffusion Policy in Transformer to 1 Billion Parameters for Robotic Manipulation22 Sep 2024
- 20243D Diffusion Policy: Generalizable Visuomotor Policy Learning via Simple 3D Representations6 Mar 2024
- 20243D Diffuser Actor: Policy Diffusion with 3D Scene Representations16 Feb 2024
- 2024DexDiffuser: Generating Dexterous Grasps with Diffusion Models5 Feb 2024
- 2023LVDiffusor: Distilling Functional Rearrangement Priors from Large Models into Diffusor3 Dec 2023
- 20236-DoF Stability Field via Diffusion Models26 Oct 2023
12 newest of 13 All 13 (JSON)
- 2024RVT-2: Learning Precise Manipulation from Few Demonstrationscited by 135
- 2024Imagination Policy: Using Generative Point Cloud Models for Learning Manipulation Policiescited by 18
- 20243D-MVP: 3D Multiview Pretraining for Robotic Manipulationcited by 10
- 2025AnyPlace: Learning Generalized Object Placement for Robot Manipulationcited by 10
- 2024MATCH POLICY: A Simple Pipeline from Point Cloud Registration to Manipulation Policiescited by 6
- 2023Out of Sight, Still in Mind: Reasoning and Planning about Unobserved Objects with Video Tracking Enabled Memory Modelscited by 5
- 2024Non-rigid Relative Placement through 3D Dense Diffusioncited by 2
- 2024Precise Pick-and-Place using Score-Based Diffusion Networkscited by 1
- 2024Implicit Contact Diffuser: Sequential Contact Reasoning with Latent Point Cloud Diffusioncited by 1
- 2025GOPLA: Generalizable Object Placement Learning via Synthetic Augmentation of Human Arrangementcited by 1
- 2024DAP: Diffusion-based Affordance Prediction for Multi-modality Storagecited by 0
- 2025Placeit! A Framework for Learning Robot Object Placement Skillscited by 0
- 2025GOPLA: Generalizable Object Placement Learning via Synthetic Augmentation of Human Arrangement16 Oct 2025
- 2025Placeit! A Framework for Learning Robot Object Placement Skills10 Oct 2025
- 2025AnyPlace: Learning Generalized Object Placement for Robot Manipulation6 Feb 2025
- 2024Non-rigid Relative Placement through 3D Dense Diffusion25 Oct 2024
- 2024Implicit Contact Diffuser: Sequential Contact Reasoning with Latent Point Cloud Diffusion21 Oct 2024
- 2024MATCH POLICY: A Simple Pipeline from Point Cloud Registration to Manipulation Policies23 Sep 2024
- 2024Precise Pick-and-Place using Score-Based Diffusion Networks15 Sep 2024
- 2024DAP: Diffusion-based Affordance Prediction for Multi-modality Storage31 Aug 2024
- 20243D-MVP: 3D Multiview Pretraining for Robotic Manipulation26 Jun 2024
- 2024Imagination Policy: Using Generative Point Cloud Models for Learning Manipulation Policies17 Jun 2024
- 2024RVT-2: Learning Precise Manipulation from Few Demonstrations12 Jun 2024
- 2023Out of Sight, Still in Mind: Reasoning and Planning about Unobserved Objects with Video Tracking Enabled Memory Models26 Sep 2023