Self-Supervised Correspondence in Visuomotor Policy Learning
On the map: Offline RL and imitation learning from data → Robot manipulation imitation learning
Abstract (arXiv)
In this paper we explore using self-supervised correspondence for improving the generalization performance and sample efficiency of visuomotor policy learning. Prior work has primarily used approaches such as autoencoding, pose-based losses, and end-to-end policy optimization in order to train the visual portion of visuomotor policies. We instead propose an approach using self-supervised dense visual correspondence training, and show this enables visuomotor policy learning with surprisingly high generalization performance with modest amounts of data: using imitation learning, we demonstrate extensive hardware validation on challenging manipulation tasks with as few as 50 demonstrations. Our learned policies can generalize across classes of objects, react to deformable object configurations, and manipulate textureless symmetrical objects in a variety of backgrounds, all with closed-loop, real-time vision-based policies. Simulated imitation learning experiments suggest that correspondence training offers sample complexity and generalization benefits compared to autoencoding and end-to-end training.
Key facts. Cited by 106 papers in the map (data through 7 Oct 2026). Placed in Robot manipulation imitation learning, one of 13 directions in Offline RL and imitation learning from data. First posted on arXiv 16 Sep 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 2; 2021: 14; 2022: 19; 2023: 24; 2024: 20; 2025: 20; 2026: 7.
106 papers.
Sort:
By area:
- 2023Learning Fine-Grained Bimanual Manipulation with Low-Cost Hardwarecited by 1,490
- 2021What Matters in Learning from Offline Human Demonstrations for Robot Manipulationcited by 699
- 2021CLIPort: What and Where Pathways for Robotic Manipulationcited by 617
- 2022Socratic Models: Composing Zero-Shot Multimodal Reasoning with Languagecited by 513
- 2021Implicit Behavioral Cloningcited by 442
- 2022VIMA: General Robot Manipulation with Multimodal Promptscited by 403
- 2022Behavior Transformers: Cloning k modes with one stonecited by 299
- 2022TAP-Vid: A Benchmark for Tracking Any Point in a Videocited by 258
- 2024Diffusion Policy Policy Optimizationcited by 255
- 2022Real-World Robot Learning with Masked Visual Pre-trainingcited by 241
- 2023MimicPlay: Long-Horizon Imitation Learning by Watching Human Playcited by 214
- 2021How to Train Your Robot with Deep Reinforcement Learning; Lessons We've Learnedcited by 201
38 more
- 2024DexCap: Scalable and Portable Mocap Data Collection System for Dexterous Manipulationcited by 149
- 2022VIOLA: Imitation Learning for Vision-Based Manipulation with Object Proposal Priorscited by 135
- 2024Reconciling Reality through Simulation: A Real-to-Sim-to-Real Approach for Robust Manipulationcited by 135
- 2023Distilled Feature Fields Enable Few-Shot Language-Guided Manipulationcited by 130
- 2025Spatial Forcing: Implicit Spatial Representation Alignment for Vision-language-action Modelcited by 125
- 2021XIRL: Cross-embodiment Inverse Reinforcement Learningcited by 101
- 2021Coarse-to-Fine Imitation Learning: Robot Manipulation from a Single Demonstrationcited by 92
- 2022NeRF-Supervision: Learning Dense Object Descriptors from Neural Radiance Fieldscited by 86
- 2021S4RL: Surprisingly Simple Self-Supervision for Offline Reinforcement Learningcited by 85
- 2024Dreamitate: Real-World Visuomotor Policy Learning via Video Generationcited by 84
- 2024General Flow as Foundation Affordance for Scalable Robot Learningcited by 80
- 2022You Only Demonstrate Once: Category-Level Manipulation from Single Visual Demonstrationcited by 79
- 2021VAT-Mart: Learning Visual Action Trajectory Proposals for Manipulating 3D ARTiculated Objectscited by 78
- 2023XSkill: Cross Embodiment Skill Discoverycited by 74
- 2024Unpacking Failure Modes of Generative Policies: Runtime Monitoring of Consistency and Progresscited by 63
- 2024Equivariant Diffusion Policycited by 61
- 2023Learning Generalizable Manipulation Policies with Object-Centric 3D Representationscited by 60
- 2023Waypoint-Based Imitation Learning for Robotic Manipulationcited by 59
- 2023NeRF in the Palm of Your Hand: Corrective Augmentation for Robotics via Novel-View Synthesiscited by 56
- 2022SE(3)-Equivariant Relational Rearrangement with Neural Descriptor Fieldscited by 51
- 2023Robot Learning with Sensorimotor Pre-trainingcited by 51
- 2022Graph Inverse Reinforcement Learning from Diverse Videoscited by 45
- 2023Crossway Diffusion: Improving Diffusion-based Visuomotor Policy via Self-supervised Learningcited by 37
- 2024Diffusion Meets DAgger: Supercharging Eye-in-hand Imitation Learningcited by 34
- 2024ARCap: Collecting High-quality Human Demonstrations for Robot Learning with Augmented Reality Feedbackcited by 31
- 2023Shelving, Stacking, Hanging: Relational Pose Diffusion for Multi-modal Rearrangementcited by 29
- 2022K-VIL: Keypoints-based Visual Imitation Learningcited by 27
- 2022USEEK: Unsupervised SE(3)-Equivariant 3D Keypoints for Generalizable Manipulationcited by 27
- 2023CCIL: Continuity-based Data Augmentation for Corrective Imitation Learningcited by 26
- 2021Generalization Through Hand-Eye Coordination: An Action Space for Learning Spatially-Invariant Visuomotor Controlcited by 24
- 2021Learning Multi-Stage Tasks with One Demonstration via Self-Replaycited by 21
- 2022Visuomotor Control in Multi-Object Scenes Using Object-Aware Representationscited by 21
- 2024Don't Start from Scratch: Behavioral Refinement via Interpolant-based Policy Diffusioncited by 21
- 2025SKIL: Semantic Keypoint Imitation Learning for Generalizable Data-efficient Manipulationcited by 21
- 20233D Implicit Transporter for Temporally Consistent Keypoint Discoverycited by 20
- 2023Chain-of-Thought Predictive Controlcited by 18
- 2024Cross-Embodiment Robot Manipulation Skill Transfer using Latent Space Alignmentcited by 17
- 2023End-to-End (Instance)-Image Goal Navigation through Correspondence as an Emergent Phenomenoncited by 16
50 most cited of 106 All 106 (JSON)
- 2026Vision-Language Grounded Task-Context-Aware Imitation Learning for Robotic Disassembly15 Sep 2026
- 2026Dressing in Motion: A Human Motion-Aware Diffusion Policy for Robot-Assisted Dressing4 Sep 2026
- 2026FAR: Failure-Aware Retry for Test-Time Recovery and Continual Policy Improvement1 Jul 2026
- 2026Robustness of Robotic Manipulation: Foundations and Frontiers30 Jun 2026
- 2026Failing Forward: Adaptive Failure-Informed Learning for Vision-Language-Action Models8 May 2026
- 2026Referring-Aware Visuomotor Policy Learning for Closed-Loop Manipulation7 Apr 2026
- 2026PRISM: Performer RS-IMLE for Single-pass Multisensory Imitation Learning2 Feb 2026
- 2025ISS Policy : Scalable Diffusion Policy with Implicit Scene Supervision17 Dec 2025
- 2025Masked Generative Policy for Robotic Control9 Dec 2025
- 2025ACG: Action Coherence Guidance for Flow-based Vision-Language-Action models25 Oct 2025
- 2025Spatial Forcing: Implicit Spatial Representation Alignment for Vision-language-action Model14 Oct 2025
- 2025MSG: Multi-Stream Generative Policies for Sample-Efficient Robotic Manipulation29 Sep 2025
38 more
- 2025GLUE: Global-Local Unified Encoding for Imitation Learning via Key-Patch Tracking27 Sep 2025
- 2025Object-Centric Mobile Manipulation through SAM2-Guided Perception and Imitation Learning15 Jul 2025
- 2025VLM-TDP: VLM-guided Trajectory-conditioned Diffusion Policy for Robust Long-Horizon Manipulation6 Jul 2025
- 2025CodeDiffuser: Attention-Enhanced Diffusion Policy via VLM-Generated Code for Instruction Ambiguity19 Jun 2025
- 2025FreqPolicy: Efficient Flow-based Visuomotor Policy via Frequency Consistency10 Jun 2025
- 2025PartInstruct: Part-level Instruction Following for Fine-grained Robot Manipulation27 May 2025
- 2025Adaptive Wiping: Adaptive contact-rich manipulation through few-shot imitation learning with Force-Torque feedback and pre-trained object representations9 May 2025
- 2025The Unreasonable Effectiveness of Discrete-Time Gaussian Process Mixtures for Robot Policy Learning6 May 2025
- 2025SemAlign3D: Semantic Correspondence between RGB-Images through Aligning 3D Object-Class Representations28 Mar 2025
- 2025Denoising-based Contractive Imitation Learning20 Mar 2025
- 2025Adapt3R: Adaptive 3D Scene Representation for Domain Transfer in Imitation Learning6 Mar 2025
- 2025Robotic Compliant Object Prying Using Diffusion Policy Guided by Vision and Force Observations6 Mar 2025
- 2025IMLE Policy: Fast and Sample Efficient Visuomotor Policy Learning via Implicit Maximum Likelihood Estimation17 Feb 2025
- 2025Bridging the Sim2Real Gap: Vision Encoder Pre-Training for Visuomotor Policy Transfer26 Jan 2025
- 2025SKIL: Semantic Keypoint Imitation Learning for Generalizable Data-efficient Manipulation24 Jan 2025
- 2024ARCap: Collecting High-quality Human Demonstrations for Robot Learning with Augmented Reality Feedback11 Oct 2024
- 2024Unpacking Failure Modes of Generative Policies: Runtime Monitoring of Consistency and Progress6 Oct 2024
- 2024IMRL: Integrating Visual, Physical, Temporal, and Geometric Representations for Enhanced Food Acquisition18 Sep 2024
- 2024KiloBot: A Programming Language for Deploying Perception-Guided Industrial Manipulators at Scale5 Sep 2024
- 2024Diffusion Policy Policy Optimization1 Sep 2024
- 2024VITAL: Interactive Few-Shot Imitation Learning via Visual Human-in-the-Loop Corrections30 Jul 2024
- 2024Equivariant Diffusion Policy1 Jul 2024
- 2024Dreamitate: Real-World Visuomotor Policy Learning via Video Generation24 Jun 2024
- 2024Cycle-Correspondence Loss: Learning Dense View-Invariant Visual Features from Unlabeled and Unordered RGB Images18 Jun 2024
- 2024Cross-Embodiment Robot Manipulation Skill Transfer using Latent Space Alignment4 Jun 2024
- 2024Robotic Constrained Imitation Learning for the Peg Transfer Task in Fundamentals of Laparoscopic Surgery6 May 2024
- 2024Towards Improving Learning from Demonstration Algorithms via MCMC Methods3 May 2024
- 2024Click to Grasp: Zero-Shot Precise Manipulation via Visual Diffusion Descriptors21 Mar 2024
- 2024DexCap: Scalable and Portable Mocap Data Collection System for Dexterous Manipulation12 Mar 2024
- 2024Reconciling Reality through Simulation: A Real-to-Sim-to-Real Approach for Robust Manipulation6 Mar 2024
- 2024Bi-KVIL: Keypoints-based Visual Imitation Learning of Bimanual Manipulation Tasks5 Mar 2024
- 2024Diffusion Meets DAgger: Supercharging Eye-in-hand Imitation Learning27 Feb 2024
- 2024Don't Start from Scratch: Behavioral Refinement via Interpolant-based Policy Diffusion25 Feb 2024
- 2024LeTO: Learning Constrained Visuomotor Policy with Differentiable Trajectory Optimization30 Jan 2024
- 2024General Flow as Foundation Affordance for Scalable Robot Learning21 Jan 2024
- 2023Working Backwards: Learning to Place by Picking4 Dec 2023
- 2023Generalizable Imitation Learning Through Pre-Trained Representations15 Nov 2023
- 2023Learning Generalizable Manipulation Policies with Object-Centric 3D Representations22 Oct 2023
50 newest of 106 All 106 (JSON)
- 2023Learning Fine-Grained Bimanual Manipulation with Low-Cost Hardwarecited by 1,490
- 2021What Matters in Learning from Offline Human Demonstrations for Robot Manipulationcited by 699
- 2021Implicit Behavioral Cloningcited by 442
- 2022Behavior Transformers: Cloning k modes with one stonecited by 299
- 2023MimicPlay: Long-Horizon Imitation Learning by Watching Human Playcited by 214
- 2024DexCap: Scalable and Portable Mocap Data Collection System for Dexterous Manipulationcited by 149
- 2022VIOLA: Imitation Learning for Vision-Based Manipulation with Object Proposal Priorscited by 135
- 2024Reconciling Reality through Simulation: A Real-to-Sim-to-Real Approach for Robust Manipulationcited by 135
- 2021XIRL: Cross-embodiment Inverse Reinforcement Learningcited by 101
- 2021Coarse-to-Fine Imitation Learning: Robot Manipulation from a Single Demonstrationcited by 92
- 2021S4RL: Surprisingly Simple Self-Supervision for Offline Reinforcement Learningcited by 85
- 2023XSkill: Cross Embodiment Skill Discoverycited by 74
12 most cited of 40 All 40 (JSON)
- 2026Vision-Language Grounded Task-Context-Aware Imitation Learning for Robotic Disassembly15 Sep 2026
- 2025Masked Generative Policy for Robotic Control9 Dec 2025
- 2025GLUE: Global-Local Unified Encoding for Imitation Learning via Key-Patch Tracking27 Sep 2025
- 2025Object-Centric Mobile Manipulation through SAM2-Guided Perception and Imitation Learning15 Jul 2025
- 2025Adaptive Wiping: Adaptive contact-rich manipulation through few-shot imitation learning with Force-Torque feedback and pre-trained object representations9 May 2025
- 2025The Unreasonable Effectiveness of Discrete-Time Gaussian Process Mixtures for Robot Policy Learning6 May 2025
- 2025Denoising-based Contractive Imitation Learning20 Mar 2025
- 2025Adapt3R: Adaptive 3D Scene Representation for Domain Transfer in Imitation Learning6 Mar 2025
- 2025SKIL: Semantic Keypoint Imitation Learning for Generalizable Data-efficient Manipulation24 Jan 2025
- 2024ARCap: Collecting High-quality Human Demonstrations for Robot Learning with Augmented Reality Feedback11 Oct 2024
- 2024Unpacking Failure Modes of Generative Policies: Runtime Monitoring of Consistency and Progress6 Oct 2024
- 2024VITAL: Interactive Few-Shot Imitation Learning via Visual Human-in-the-Loop Corrections30 Jul 2024
12 newest of 40 All 40 (JSON)
- 2024Diffusion Policy Policy Optimizationcited by 255
- 2024Equivariant Diffusion Policycited by 61
- 2023Crossway Diffusion: Improving Diffusion-based Visuomotor Policy via Self-supervised Learningcited by 37
- 2024Don't Start from Scratch: Behavioral Refinement via Interpolant-based Policy Diffusioncited by 21
- 2025IMLE Policy: Fast and Sample Efficient Visuomotor Policy Learning via Implicit Maximum Likelihood Estimationcited by 13
- 2025Robotic Compliant Object Prying Using Diffusion Policy Guided by Vision and Force Observationscited by 9
- 2025CodeDiffuser: Attention-Enhanced Diffusion Policy via VLM-Generated Code for Instruction Ambiguitycited by 8
- 2025VLM-TDP: VLM-guided Trajectory-conditioned Diffusion Policy for Robust Long-Horizon Manipulationcited by 1
- 2025MSG: Multi-Stream Generative Policies for Sample-Efficient Robotic Manipulationcited by 1
- 2026PRISM: Performer RS-IMLE for Single-pass Multisensory Imitation Learningcited by 1
- 2025ISS Policy : Scalable Diffusion Policy with Implicit Scene Supervisioncited by 0
- 2026Dressing in Motion: A Human Motion-Aware Diffusion Policy for Robot-Assisted Dressingcited by 0
- 2026Dressing in Motion: A Human Motion-Aware Diffusion Policy for Robot-Assisted Dressing4 Sep 2026
- 2026PRISM: Performer RS-IMLE for Single-pass Multisensory Imitation Learning2 Feb 2026
- 2025ISS Policy : Scalable Diffusion Policy with Implicit Scene Supervision17 Dec 2025
- 2025MSG: Multi-Stream Generative Policies for Sample-Efficient Robotic Manipulation29 Sep 2025
- 2025VLM-TDP: VLM-guided Trajectory-conditioned Diffusion Policy for Robust Long-Horizon Manipulation6 Jul 2025
- 2025CodeDiffuser: Attention-Enhanced Diffusion Policy via VLM-Generated Code for Instruction Ambiguity19 Jun 2025
- 2025Robotic Compliant Object Prying Using Diffusion Policy Guided by Vision and Force Observations6 Mar 2025
- 2025IMLE Policy: Fast and Sample Efficient Visuomotor Policy Learning via Implicit Maximum Likelihood Estimation17 Feb 2025
- 2024Diffusion Policy Policy Optimization1 Sep 2024
- 2024Equivariant Diffusion Policy1 Jul 2024
- 2024Don't Start from Scratch: Behavioral Refinement via Interpolant-based Policy Diffusion25 Feb 2024
- 2023Crossway Diffusion: Improving Diffusion-based Visuomotor Policy via Self-supervised Learning4 Jul 2023
- 2022Socratic Models: Composing Zero-Shot Multimodal Reasoning with Languagecited by 513
- 2023Chain-of-Thought Predictive Controlcited by 18
- 2023Chain-of-Thought Predictive Control3 Apr 2023
- 2022Socratic Models: Composing Zero-Shot Multimodal Reasoning with Language1 Apr 2022
- 2021CLIPort: What and Where Pathways for Robotic Manipulationcited by 617
- 2022VIMA: General Robot Manipulation with Multimodal Promptscited by 403
- 2022TAP-Vid: A Benchmark for Tracking Any Point in a Videocited by 258
- 2022Real-World Robot Learning with Masked Visual Pre-trainingcited by 241
- 2021How to Train Your Robot with Deep Reinforcement Learning; Lessons We've Learnedcited by 201
- 2023Distilled Feature Fields Enable Few-Shot Language-Guided Manipulationcited by 130
- 2022NeRF-Supervision: Learning Dense Object Descriptors from Neural Radiance Fieldscited by 86
- 2024Dreamitate: Real-World Visuomotor Policy Learning via Video Generationcited by 84
- 2024General Flow as Foundation Affordance for Scalable Robot Learningcited by 80
- 2022You Only Demonstrate Once: Category-Level Manipulation from Single Visual Demonstrationcited by 79
- 2021VAT-Mart: Learning Visual Action Trajectory Proposals for Manipulating 3D ARTiculated Objectscited by 78
- 2022SE(3)-Equivariant Relational Rearrangement with Neural Descriptor Fieldscited by 51
12 most cited of 49 All 49 (JSON)
- 2026FAR: Failure-Aware Retry for Test-Time Recovery and Continual Policy Improvement1 Jul 2026
- 2026Robustness of Robotic Manipulation: Foundations and Frontiers30 Jun 2026
- 2026Referring-Aware Visuomotor Policy Learning for Closed-Loop Manipulation7 Apr 2026
- 2025FreqPolicy: Efficient Flow-based Visuomotor Policy via Frequency Consistency10 Jun 2025
- 2025PartInstruct: Part-level Instruction Following for Fine-grained Robot Manipulation27 May 2025
- 2025SemAlign3D: Semantic Correspondence between RGB-Images through Aligning 3D Object-Class Representations28 Mar 2025
- 2025Bridging the Sim2Real Gap: Vision Encoder Pre-Training for Visuomotor Policy Transfer26 Jan 2025
- 2024IMRL: Integrating Visual, Physical, Temporal, and Geometric Representations for Enhanced Food Acquisition18 Sep 2024
- 2024KiloBot: A Programming Language for Deploying Perception-Guided Industrial Manipulators at Scale5 Sep 2024
- 2024Dreamitate: Real-World Visuomotor Policy Learning via Video Generation24 Jun 2024
- 2024Cycle-Correspondence Loss: Learning Dense View-Invariant Visual Features from Unlabeled and Unordered RGB Images18 Jun 2024
- 2024Cross-Embodiment Robot Manipulation Skill Transfer using Latent Space Alignment4 Jun 2024
12 newest of 49 All 49 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2025Correspondence-Oriented Imitation Learning: Flexible Visuomotor Control with 3D Conditioning
- 2018Reinforcement and Imitation Learning for Diverse Visuomotor Skills
- 2025DINOv3-Diffusion Policy: Self-Supervised Large Visual Model for Visuomotor Diffusion Policy Learning
- 2025A Study on Enhancing the Generalization Ability of Visuomotor Policies via Data Augmentation ↗
- 2025SViP: Sequencing Bimanual Visuomotor Policies with Object-Centric Motion Primitives
- 2024DynaMo: In-Domain Dynamics Pretraining for Visuo-Motor Control
- 2026AdaTempo: Learning Shared Relative Tempo from Demonstrations for Faster Robot Manipulation
- 2023SpawnNet: Learning Generalizable Visuomotor Skills from Pre-trained Networks