Keypoints into the Future: Self-Supervised Correspondence in Model-Based Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Predictive models have been at the core of many robotic systems, from quadrotors to walking robots. However, it has been challenging to develop and apply such models to practical robotic manipulation due to high-dimensional sensory observations such as images. Previous approaches to learning models in the context of robotic manipulation have either learned whole image dynamics or used autoencoders to learn dynamics in a low-dimensional latent state. In this work, we introduce model-based prediction with self-supervised visual correspondence learning, and show that not only is this indeed possible, but demonstrate that these types of predictive models show compelling performance improvements over alternative methods for vision-based RL with autoencoder-type vision training. Through simulation experiments, we demonstrate that our models provide better generalization precision, particularly in 3D scenes, scenes involving occlusion, and in category-generalization. Additionally, we validate that our method effectively transfers to the real world through hardware experiments. Videos and supplementary materials available at https://sites.google.com/view/keypointsintothefuture
Key facts. Cited by 69 papers in the map (data through 7 Oct 2026). First posted on arXiv 10 Sep 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 12; 2022: 19; 2023: 15; 2024: 16; 2025: 4; 2026: 3.
69 papers.
Sort:
By area:
- 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planningcited by 738
- 2024ReKep: Spatio-Temporal Reasoning of Relational Keypoint Constraints for Robotic Manipulationcited by 305
- 2023TAPIR: Tracking Any Point with per-frame Initialization and temporal Refinementcited by 281
- 2022TAP-Vid: A Benchmark for Tracking Any Point in a Videocited by 258
- 2023Any-point Trajectory Modeling for Policy Learningcited by 207
- 2022Masked World Models for Visual Controlcited by 169
- 2021The Surprising Effectiveness of Representation Learning for Visual Imitationcited by 149
- 20213D Neural Scene Representations for Visuomotor Controlcited by 110
- 2021Learning by Watching: Physical Imitation of Manipulation Skills from Human Videoscited by 87
- 2022NeRF-Supervision: Learning Dense Object Descriptors from Neural Radiance Fieldscited by 86
- 2024General Flow as Foundation Affordance for Scalable Robot Learningcited by 80
- 2022You Only Demonstrate Once: Category-Level Manipulation from Single Visual Demonstrationcited by 79
38 more
- 2024Local All-Pair Correspondence for Point Trackingcited by 77
- 2022RoboCraft: Learning to See, Simulate, and Shape Elasto-Plastic Objects with Graph Networkscited by 69
- 2022Learning Multi-Object Dynamics with Compositional Neural Radiance Fieldscited by 63
- 2023Learning Generalizable Manipulation Policies with Object-Centric 3D Representationscited by 60
- 2021KeypointDeformer: Unsupervised 3D Keypoint Discovery for Shape Controlcited by 50
- 2022TAX-Pose: Task-Specific Cross-Pose Estimation for Robot Manipulationcited by 47
- 2022Graph Inverse Reinforcement Learning from Diverse Videoscited by 45
- 2022Reinforcement Learning with Neural Radiance Fieldscited by 43
- 2024Dynamic 3D Gaussian Tracking for Graph-Based Neural Dynamics Modelingcited by 42
- 2023D3Fields: Dynamic 3D Descriptor Fields for Zero-Shot Generalizable Rearrangementcited by 35
- 2024Deep Generative Models in Robotics: A Survey on Learning from Multimodal Demonstrationscited by 35
- 2021Test-Time Personalization with a Transformer for Human Pose Estimationcited by 33
- 2023See to Touch: Learning Tactile Dexterity through Visual Incentivescited by 33
- 2021Unsupervised Learning of Visual 3D Keypoints for Controlcited by 30
- 2021Learning Models as Functionals of Signed-Distance Fields for Manipulation Planningcited by 29
- 2021VOILA: Visual-Observation-Only Imitation Learning for Autonomous Navigationcited by 27
- 2022K-VIL: Keypoints-based Visual Imitation Learningcited by 27
- 2022USEEK: Unsupervised SE(3)-Equivariant 3D Keypoints for Generalizable Manipulationcited by 27
- 2024GenDP: 3D Semantic Fields for Category-Level Generalizable Diffusion Policycited by 25
- 2021Generalization Through Hand-Eye Coordination: An Action Space for Learning Spatially-Invariant Visuomotor Controlcited by 24
- 2025UAD: Unsupervised Affordance Distillation for Generalization in Robotic Manipulationcited by 23
- 2023Dynamic-Resolution Model Learning for Object Pile Manipulationcited by 22
- 2022Visuomotor Control in Multi-Object Scenes Using Object-Aware Representationscited by 21
- 2024Deep Generative Models for Offline Policy Learning: Tutorial, Survey, and Perspectives on Future Directionscited by 20
- 2021Vision-driven Compliant Manipulation for Reliable, High-Precision Assembly Taskscited by 18
- 2024RoboPack: Learning Tactile-Informed Dynamics Models for Dense Packingcited by 18
- 2023Fighting Uncertainty with Gradients: Offline Reinforcement Learning via Diffusion Score Matchingcited by 17
- 2024Composable Part-Based Manipulationcited by 17
- 2023SNeRL: Semantic-aware Neural Radiance Fields for Reinforcement Learningcited by 16
- 2023One-shot Imitation Learning via Interaction Warpingcited by 16
- 2023Model-Based Control with Sparse Neural Dynamicscited by 16
- 2022Filtered-CoPhy: Unsupervised Learning of Counterfactual Physics in Pixel Spacecited by 14
- 2023Few-Shot In-Context Imitation Learning via Implicit Graph Alignmentcited by 14
- 20233D-IntPhys: Towards More Generalized 3D-grounded Visual Intuitive Physics under Challenging Scenescited by 12
- 2024Reinforcement Learning with Generalizable Gaussian Splattingcited by 11
- 2024Click to Grasp: Zero-Shot Precise Manipulation via Visual Diffusion Descriptorscited by 10
- 2022Where To Start? Transferring Simple Skills to Complex Environmentscited by 9
- 2023Multi-Object Manipulation via Object-Centric Neural Scattering Functionscited by 9
50 most cited of 69 All 69 (JSON)
- 2026GROW2: Grounding Which and Where for Robot Tool Use29 Jun 2026
- 2026Parallel Differentiable Reachability for Learning and Planning with Certified Neural Dynamics and Controllers25 May 2026
- 2026IMPASTO: Integrating Model-Based Planning with Learned Dynamics Models for Robotic Oil Painting Reproduction31 Mar 2026
- 2025Constraint-Preserving Data Generation for Visuomotor Policy Learning5 Aug 2025
- 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planning11 Jun 2025
- 2025UAD: Unsupervised Affordance Distillation for Generalization in Robotic Manipulation10 Jun 2025
- 2025FUNCTO: Function-Centric One-Shot Imitation Learning for Tool Manipulation17 Feb 2025
- 2024The Surprising Ineffectiveness of Pre-Trained Visual Representations for Model-Based Reinforcement Learning15 Nov 2024
- 2024Dynamic 3D Gaussian Tracking for Graph-Based Neural Dynamics Modeling24 Oct 2024
- 2024GenDP: 3D Semantic Fields for Category-Level Generalizable Diffusion Policy23 Oct 2024
- 2024REPeat: A Real2Sim2Real Approach for Pre-acquisition of Soft Food Items in Robot-assisted Feeding13 Oct 2024
- 2024DEL: Discrete Element Learner for Learning 3D Particle Dynamics with Neural Rendering11 Oct 2024
38 more
- 2024ReKep: Spatio-Temporal Reasoning of Relational Keypoint Constraints for Robotic Manipulation3 Sep 2024
- 2024Deep Generative Models in Robotics: A Survey on Learning from Multimodal Demonstrations8 Aug 2024
- 2024Local All-Pair Correspondence for Point Tracking22 Jul 2024
- 2024RoboPack: Learning Tactile-Informed Dynamics Models for Dense Packing1 Jul 2024
- 2024Cycle-Correspondence Loss: Learning Dense View-Invariant Visual Features from Unlabeled and Unordered RGB Images18 Jun 2024
- 2024Composable Part-Based Manipulation9 May 2024
- 2024Click to Grasp: Zero-Shot Precise Manipulation via Visual Diffusion Descriptors21 Mar 2024
- 2024Reinforcement Learning with Generalizable Gaussian Splatting18 Mar 2024
- 2024Deep Generative Models for Offline Policy Learning: Tutorial, Survey, and Perspectives on Future Directions21 Feb 2024
- 2024General Flow as Foundation Affordance for Scalable Robot Learning21 Jan 2024
- 2024Unsupervised Salient Patch Selection for Data-Efficient Reinforcement Learning10 Jan 2024
- 2023Any-point Trajectory Modeling for Policy Learning28 Dec 2023
- 2023Model-Based Control with Sparse Neural Dynamics20 Dec 2023
- 2023Learning Generalizable Manipulation Policies with Object-Centric 3D Representations22 Oct 2023
- 2023Few-Shot In-Context Imitation Learning via Implicit Graph Alignment18 Oct 2023
- 2023D3Fields: Dynamic 3D Descriptor Fields for Zero-Shot Generalizable Rearrangement28 Sep 2023
- 2023Doduo: Learning Dense Visual Correspondence from Unsupervised Semantic-Aware Flow26 Sep 2023
- 2023See to Touch: Learning Tactile Dexterity through Visual Incentives21 Sep 2023
- 2023Learning Sequential Acquisition Policies for Robot-Assisted Feeding11 Sep 2023
- 2023Dynamic-Resolution Model Learning for Object Pile Manipulation29 Jun 2023
- 2023Fighting Uncertainty with Gradients: Offline Reinforcement Learning via Diffusion Score Matching24 Jun 2023
- 2023One-shot Imitation Learning via Interaction Warping21 Jun 2023
- 2023Multi-Object Manipulation via Object-Centric Neural Scattering Functions14 Jun 2023
- 2023TAPIR: Tracking Any Point with per-frame Initialization and temporal Refinement14 Jun 2023
- 20233D-IntPhys: Towards More Generalized 3D-grounded Visual Intuitive Physics under Challenging Scenes22 Apr 2023
- 2023SNeRL: Semantic-aware Neural Radiance Fields for Reinforcement Learning27 Jan 2023
- 2022Where To Start? Transferring Simple Skills to Complex Environments12 Dec 2022
- 2022TAX-Pose: Task-Specific Cross-Pose Estimation for Robot Manipulation17 Nov 2022
- 2022TAP-Vid: A Benchmark for Tracking Any Point in a Video7 Nov 2022
- 2022Self-supervised Pre-training for Semantic Segmentation in an Indoor Scene4 Oct 2022
- 2022USEEK: Unsupervised SE(3)-Equivariant 3D Keypoints for Generalizable Manipulation28 Sep 2022
- 2022K-VIL: Keypoints-based Visual Imitation Learning7 Sep 2022
- 2022Graph Inverse Reinforcement Learning from Diverse Videos28 Jul 2022
- 2022Masked World Models for Visual Control28 Jun 2022
- 2022Efficient and Robust Training of Dense Object Nets for Multi-Object Robot Manipulation24 Jun 2022
- 2022Reinforcement Learning with Neural Radiance Fields3 Jun 2022
- 2022Self-Supervised Learning of Multi-Object Keypoints for Robotic Manipulation17 May 2022
- 2022Visuomotor Control in Multi-Object Scenes Using Object-Aware Representations12 May 2022
50 newest of 69 All 69 (JSON)
- 2023Any-point Trajectory Modeling for Policy Learningcited by 207
- 2021The Surprising Effectiveness of Representation Learning for Visual Imitationcited by 149
- 2023Learning Generalizable Manipulation Policies with Object-Centric 3D Representationscited by 60
- 2022Graph Inverse Reinforcement Learning from Diverse Videoscited by 45
- 2024Deep Generative Models in Robotics: A Survey on Learning from Multimodal Demonstrationscited by 35
- 2021VOILA: Visual-Observation-Only Imitation Learning for Autonomous Navigationcited by 27
- 2022K-VIL: Keypoints-based Visual Imitation Learningcited by 27
- 2021Generalization Through Hand-Eye Coordination: An Action Space for Learning Spatially-Invariant Visuomotor Controlcited by 24
- 2024Deep Generative Models for Offline Policy Learning: Tutorial, Survey, and Perspectives on Future Directionscited by 20
- 2023Fighting Uncertainty with Gradients: Offline Reinforcement Learning via Diffusion Score Matchingcited by 17
- 2023One-shot Imitation Learning via Interaction Warpingcited by 16
- 2023Few-Shot In-Context Imitation Learning via Implicit Graph Alignmentcited by 14
12 most cited of 13 All 13 (JSON)
- 2025FUNCTO: Function-Centric One-Shot Imitation Learning for Tool Manipulation17 Feb 2025
- 2024Deep Generative Models in Robotics: A Survey on Learning from Multimodal Demonstrations8 Aug 2024
- 2024Deep Generative Models for Offline Policy Learning: Tutorial, Survey, and Perspectives on Future Directions21 Feb 2024
- 2023Any-point Trajectory Modeling for Policy Learning28 Dec 2023
- 2023Learning Generalizable Manipulation Policies with Object-Centric 3D Representations22 Oct 2023
- 2023Few-Shot In-Context Imitation Learning via Implicit Graph Alignment18 Oct 2023
- 2023Fighting Uncertainty with Gradients: Offline Reinforcement Learning via Diffusion Score Matching24 Jun 2023
- 2023One-shot Imitation Learning via Interaction Warping21 Jun 2023
- 2022K-VIL: Keypoints-based Visual Imitation Learning7 Sep 2022
- 2022Graph Inverse Reinforcement Learning from Diverse Videos28 Jul 2022
- 2021The Surprising Effectiveness of Representation Learning for Visual Imitation2 Dec 2021
- 2021VOILA: Visual-Observation-Only Imitation Learning for Autonomous Navigation19 May 2021
12 newest of 13 All 13 (JSON)
- 2024GenDP: 3D Semantic Fields for Category-Level Generalizable Diffusion Policycited by 25
- 2024Composable Part-Based Manipulationcited by 17
- 2024GenDP: 3D Semantic Fields for Category-Level Generalizable Diffusion Policy23 Oct 2024
- 2024Composable Part-Based Manipulation9 May 2024
- 2026GROW2: Grounding Which and Where for Robot Tool Usecited by 0
- 2026GROW2: Grounding Which and Where for Robot Tool Use29 Jun 2026
- 2023TAPIR: Tracking Any Point with per-frame Initialization and temporal Refinementcited by 281
- 2022TAP-Vid: A Benchmark for Tracking Any Point in a Videocited by 258
- 20213D Neural Scene Representations for Visuomotor Controlcited by 110
- 2021Learning by Watching: Physical Imitation of Manipulation Skills from Human Videoscited by 87
- 2022NeRF-Supervision: Learning Dense Object Descriptors from Neural Radiance Fieldscited by 86
- 2024General Flow as Foundation Affordance for Scalable Robot Learningcited by 80
- 2022You Only Demonstrate Once: Category-Level Manipulation from Single Visual Demonstrationcited by 79
- 2024Local All-Pair Correspondence for Point Trackingcited by 77
- 2022Learning Multi-Object Dynamics with Compositional Neural Radiance Fieldscited by 63
- 2021KeypointDeformer: Unsupervised 3D Keypoint Discovery for Shape Controlcited by 50
- 2022TAX-Pose: Task-Specific Cross-Pose Estimation for Robot Manipulationcited by 47
- 2022Reinforcement Learning with Neural Radiance Fieldscited by 43
12 most cited of 49 All 49 (JSON)
- 2026Parallel Differentiable Reachability for Learning and Planning with Certified Neural Dynamics and Controllers25 May 2026
- 2026IMPASTO: Integrating Model-Based Planning with Learned Dynamics Models for Robotic Oil Painting Reproduction31 Mar 2026
- 2025Constraint-Preserving Data Generation for Visuomotor Policy Learning5 Aug 2025
- 2025UAD: Unsupervised Affordance Distillation for Generalization in Robotic Manipulation10 Jun 2025
- 2024The Surprising Ineffectiveness of Pre-Trained Visual Representations for Model-Based Reinforcement Learning15 Nov 2024
- 2024Dynamic 3D Gaussian Tracking for Graph-Based Neural Dynamics Modeling24 Oct 2024
- 2024REPeat: A Real2Sim2Real Approach for Pre-acquisition of Soft Food Items in Robot-assisted Feeding13 Oct 2024
- 2024DEL: Discrete Element Learner for Learning 3D Particle Dynamics with Neural Rendering11 Oct 2024
- 2024Local All-Pair Correspondence for Point Tracking22 Jul 2024
- 2024RoboPack: Learning Tactile-Informed Dynamics Models for Dense Packing1 Jul 2024
- 2024Cycle-Correspondence Loss: Learning Dense View-Invariant Visual Features from Unlabeled and Unordered RGB Images18 Jun 2024
- 2024Click to Grasp: Zero-Shot Precise Manipulation via Visual Diffusion Descriptors21 Mar 2024
12 newest of 49 All 49 (JSON)