One-shot Imitation Learning via Interaction Warping
On the map: Offline RL and imitation learning from data → Robot manipulation imitation learning
Abstract (arXiv)
Imitation learning of robot policies from few demonstrations is crucial in open-ended applications. We propose a new method, Interaction Warping, for learning SE(3) robotic manipulation policies from a single demonstration. We infer the 3D mesh of each object in the environment using shape warping, a technique for aligning point clouds across object instances. Then, we represent manipulation actions as keypoints on objects, which can be warped with the shape of the object. We show successful one-shot imitation learning on three simulated and real-world object re-arrangement tasks. We also demonstrate the ability of our method to predict object meshes and robot grasps in the wild.
Key facts. Cited by 16 papers in the map (data through 13 Oct 2026). Placed in Robot manipulation imitation learning, one of 13 directions in Offline RL and imitation learning from data. Builds on 33 of its 47 references that are in the map. First posted on arXiv 21 Jun 2023.
Builds on
33 of 47 references.
References · 33
- 2017Attention Is All You Needcited by 60,106
- 2014Adam: A Method for Stochastic Optimizationcited by 37,093
- 2023Segment Anythingcited by 8,195
- 2017Mask R-CNNcited by 7,266
- 2015ShapeNet: An Information-Rich 3D Model Repositorycited by 2,889
- 2021Masked-attention Mask Transformer for Universal Image Segmentationcited by 2,344
- 2018Dynamic Graph CNN for Learning on Point Cloudscited by 2,222
- 2022Detecting Twenty-thousand Classes using Image-level Supervisioncited by 613
- 2022Simple Open-Vocabulary Object Detection with Vision Transformerscited by 304
- 2021Neural Descriptor Fields: SE(3)-Equivariant Object Representations for Manipulationcited by 206
- 2019kPAM: KeyPoint Affordances for Category-Level Robotic Manipulationcited by 152
- 2018Dense Object Nets: Learning Dense Visual Object Descriptors By and For Robotic Manipulationcited by 140
21 more
- 2021A comprehensive survey on point cloud registrationcited by 84
- 2022You Only Demonstrate Once: Category-Level Manipulation from Single Visual Demonstrationcited by 79
- 2022SO(2)-Equivariant Reinforcement Learningcited by 79
- 2020Keypoints into the Future: Self-Supervised Correspondence in Model-Based Reinforcement Learningcited by 69
- 2019KETO: Learning Keypoint Representations for Tool Manipulationcited by 68
- 2018Explorations in Homeomorphic Variational Auto-Encodingcited by 65
- 2022Equivariant Descriptor Fields: SE(3)-Equivariant Energy-Based Models for End-to-End Visual Robotic Manipulation Learningcited by 55
- 2022Learning Symmetric Embeddings for Equivariant World Modelscited by 54
- 2022SE(3)-Equivariant Relational Rearrangement with Neural Descriptor Fieldscited by 51
- 2022TAX-Pose: Task-Specific Cross-Pose Estimation for Robot Manipulationcited by 47
- 2021kPAM 2.0: Feedback Control for Category-Level Robotic Manipulationcited by 38
- 2020S3K: Self-Supervised Semantic Keypoints for Robotic Manipulation via Multi-View Consistencycited by 28
- 2019kPAM-SC: Generalizable Manipulation Planning using KeyPoint Affordance and Shape Completioncited by 27
- 2023Local Neural Descriptor Fields: Locally Conditioned Object Representations for Manipulationcited by 21
- 2021GIFT: Generalizable Interaction-aware Functional Tool Affordances without Labelscited by 19
- 2018Transferring Category-based Functional Grasping Skills by Latent Space Non-Rigid Registrationcited by 8
- 2018Transferring Grasping Skills to Novel Instances by Latent Space Non-Rigid Registrationcited by 4
- 2018Supervised Autonomous Locomotion and Manipulation for Disaster Response with a Centaur-like Robotcited by 3
- 2018Transferring Grasp Configurations using Active Learning and Local Replanningcited by 2
- 2019Autonomous Bimanual Functional Regrasping of Novel Object Class Instancescited by 2
- 2018Learning Postural Synergies for Categorical Grasping through Shape Space Registrationcited by 1
Built on by
Citing papers per year: 2023: 1; 2024: 6; 2025: 6; 2026: 3.
16 papers.
Sort:
By area:
- 2024Sparse Diffusion Policy: A Sparse, Reusable, and Flexible Policy for Robot Learningcited by 56
- 2023Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulationcited by 51
- 2024One-Shot Imitation Learning with Invariance Matching for Robotic Manipulationcited by 18
- 2024Keypoint Abstraction using Large Models for Object-Relative Imitation Learningcited by 18
- 2025MimicFunc: Imitating Tool Manipulation from a Single Human Video via Functional Correspondencecited by 11
- 2025FUNCTO: Function-Centric One-Shot Imitation Learning for Tool Manipulationcited by 8
- 2025One-Shot Dual-Arm Imitation Learningcited by 7
- 2024One-Shot Manipulation Strategy Learning by Making Contact Analogiescited by 6
- 2025Elastic Motion Policy: An Adaptive Dynamical System for Robust and Efficient One-Shot Imitation Learningcited by 4
- 2024Visual Imitation Learning of Task-Oriented Object Grasping and Rearrangementcited by 3
- 2024One-shot Video Imitation via Parameterized Symbolic Abstraction Graphscited by 3
- 2026RAPID: Robot Agentic Programming from Demonstrationscited by 1
4 more
- 2025Multi-Group Equivariant Augmentation for Reinforcement Learning in Robot Manipulationcited by 0
- 2025Annotation-Free One-Shot Imitation Learning for Multi-Step Manipulation Taskscited by 0
- 2026RA-VLA: Retrieval-Augmented VLA for Test-Time Adaptationcited by 0
- 2026MemCorr-DP: Counterfactual Correspondence Conditioning for a Diffusion Policy Guided by a Referencecited by 0
- 2026RAPID: Robot Agentic Programming from Demonstrations24 Sep 2026
- 2026MemCorr-DP: Counterfactual Correspondence Conditioning for a Diffusion Policy Guided by a Reference6 Sep 2026
- 2026RA-VLA: Retrieval-Augmented VLA for Test-Time Adaptation26 Aug 2026
- 2025Annotation-Free One-Shot Imitation Learning for Multi-Step Manipulation Tasks29 Sep 2025
- 2025MimicFunc: Imitating Tool Manipulation from a Single Human Video via Functional Correspondence19 Aug 2025
- 2025Multi-Group Equivariant Augmentation for Reinforcement Learning in Robot Manipulation15 Aug 2025
- 2025Elastic Motion Policy: An Adaptive Dynamical System for Robust and Efficient One-Shot Imitation Learning11 Mar 2025
- 2025One-Shot Dual-Arm Imitation Learning10 Mar 2025
- 2025FUNCTO: Function-Centric One-Shot Imitation Learning for Tool Manipulation17 Feb 2025
- 2024One-Shot Manipulation Strategy Learning by Making Contact Analogies14 Nov 2024
- 2024Keypoint Abstraction using Large Models for Object-Relative Imitation Learning30 Oct 2024
- 2024One-shot Video Imitation via Parameterized Symbolic Abstraction Graphs22 Aug 2024
4 more
- 2024Sparse Diffusion Policy: A Sparse, Reusable, and Flexible Policy for Robot Learning1 Jul 2024
- 2024One-Shot Imitation Learning with Invariance Matching for Robotic Manipulation21 May 2024
- 2024Visual Imitation Learning of Task-Oriented Object Grasping and Rearrangement20 Mar 2024
- 2023Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulation6 Sep 2023
- 2024One-Shot Imitation Learning with Invariance Matching for Robotic Manipulationcited by 18
- 2024Keypoint Abstraction using Large Models for Object-Relative Imitation Learningcited by 18
- 2025FUNCTO: Function-Centric One-Shot Imitation Learning for Tool Manipulationcited by 8
- 2025One-Shot Dual-Arm Imitation Learningcited by 7
- 2025Elastic Motion Policy: An Adaptive Dynamical System for Robust and Efficient One-Shot Imitation Learningcited by 4
- 2024Visual Imitation Learning of Task-Oriented Object Grasping and Rearrangementcited by 3
- 2025Annotation-Free One-Shot Imitation Learning for Multi-Step Manipulation Taskscited by 0
- 2025Annotation-Free One-Shot Imitation Learning for Multi-Step Manipulation Tasks29 Sep 2025
- 2025Elastic Motion Policy: An Adaptive Dynamical System for Robust and Efficient One-Shot Imitation Learning11 Mar 2025
- 2025One-Shot Dual-Arm Imitation Learning10 Mar 2025
- 2025FUNCTO: Function-Centric One-Shot Imitation Learning for Tool Manipulation17 Feb 2025
- 2024Keypoint Abstraction using Large Models for Object-Relative Imitation Learning30 Oct 2024
- 2024One-Shot Imitation Learning with Invariance Matching for Robotic Manipulation21 May 2024
- 2024Visual Imitation Learning of Task-Oriented Object Grasping and Rearrangement20 Mar 2024
- 2024Sparse Diffusion Policy: A Sparse, Reusable, and Flexible Policy for Robot Learningcited by 56
- 2023Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulationcited by 51
- 2026MemCorr-DP: Counterfactual Correspondence Conditioning for a Diffusion Policy Guided by a Referencecited by 0
- 2026MemCorr-DP: Counterfactual Correspondence Conditioning for a Diffusion Policy Guided by a Reference6 Sep 2026
- 2024Sparse Diffusion Policy: A Sparse, Reusable, and Flexible Policy for Robot Learning1 Jul 2024
- 2023Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulation6 Sep 2023
- 2026RAPID: Robot Agentic Programming from Demonstrationscited by 1
- 2026RAPID: Robot Agentic Programming from Demonstrations24 Sep 2026
- 2026RA-VLA: Retrieval-Augmented VLA for Test-Time Adaptationcited by 0
- 2026RA-VLA: Retrieval-Augmented VLA for Test-Time Adaptation26 Aug 2026
Similar, not cited
Close in wording, no citation either way
- 2021Coarse-to-Fine Imitation Learning: Robot Manipulation from a Single Demonstration
- 2026Improving Imitation Learning Efficiency for Manipulation through Geometric Prior Pretraining
- 2026Instant-Fold: In-Context Imitation Learning for Deformable Object Manipulation
- 2025PRISM: Pointcloud Reintegrated Inference via Segmentation and Cross-attention for Manipulation
- 2026RE4: Transformation-aware Imitation of Object Interactions Using Manipulation Modes
- 2026WARP-RM: A Warp-Augmented Relative Progress Reward Model for Data Curation
- 2025Canonical Policy: Learning Canonical 3D Representation for SE(3)-Equivariant Policy
- 2025ManiLong-Shot: Interaction-Aware One-Shot Imitation Learning for Long-Horizon Manipulation