Unsupervised Learning of Object Keypoints for Perception and Control
Not on the map. Why?
Abstract (arXiv)
The study of object representations in computer vision has primarily focused on developing representations that are useful for image classification, object detection, or semantic segmentation as downstream tasks. In this work we aim to learn object representations that are useful for control and reinforcement learning (RL). To this end, we introduce Transporter, a neural network architecture for discovering concise geometric object representations in terms of keypoints or image-space coordinates. Our method learns from raw video frames in a fully unsupervised manner, by transporting learnt image features between video frames using a keypoint bottleneck. The discovered keypoints track objects and object parts across long time-horizons more accurately than recent similar methods. Furthermore, consistent long-term tracking enables two notable results in control domains -- (1) using the keypoint co-ordinates and corresponding image features as inputs enables highly sample-efficient reinforcement learning; (2) learning to explore by controlling keypoint locations drastically reduces the search space, enabling deep exploration (leading to states unreachable through random action exploration) without any extrinsic rewards.
Key facts. Cited by 109 papers in the map (data through 7 Oct 2026). First posted on arXiv 19 Jun 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 28; 2022: 27; 2023: 26; 2024: 14; 2025: 10; 2026: 3.
109 papers.
Sort:
By area:
- 2021Deep Reinforcement Learning at the Edge of the Statistical Precipicecited by 781
- 2021CLIPort: What and Where Pathways for Robotic Manipulationcited by 617
- 2022Socratic Models: Composing Zero-Shot Multimodal Reasoning with Languagecited by 513
- 2024ReKep: Spatio-Temporal Reasoning of Relational Keypoint Constraints for Robotic Manipulationcited by 305
- 2021Towards Causal Representation Learningcited by 207
- 2021PiCIE: Unsupervised Semantic Segmentation using Invariance and Equivariance in Clusteringcited by 163
- 2022Human-to-Robot Imitation in the Wildcited by 157
- 2022VIOLA: Imitation Learning for Vision-Based Manipulation with Object Proposal Priorscited by 135
- 20213D Neural Scene Representations for Visuomotor Controlcited by 110
- 2021Learning Temporally Causal Latent Processes from General Temporal Datacited by 96
- 2021Learning by Watching: Physical Imitation of Manipulation Skills from Human Videoscited by 87
- 2021CoNeRF: Controllable Neural Radiance Fieldscited by 65
38 more
- 2022Learning Multi-Object Dynamics with Compositional Neural Radiance Fieldscited by 63
- 2022Equivariant Descriptor Fields: SE(3)-Equivariant Energy-Based Models for End-to-End Visual Robotic Manipulation Learningcited by 54
- 2020Model-Based Inverse Reinforcement Learning from Visual Demonstrationscited by 52
- 2022Visual Reinforcement Learning with Self-Supervised 3D Representationscited by 45
- 2022Reinforcement Learning with Neural Radiance Fieldscited by 43
- 2023Crossway Diffusion: Improving Diffusion-based Visuomotor Policy via Self-supervised Learningcited by 37
- 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
- 2021Unsupervised Visual Attention and Invariance for Reinforcement Learningcited by 30
- 2021Unsupervised Learning of Visual 3D Keypoints for Controlcited by 30
- 2021Generalization Through Hand-Eye Coordination: An Action Space for Learning Spatially-Invariant Visuomotor Controlcited by 24
- 2022Does Self-supervised Learning Really Improve Reinforcement Learning from Pixels?cited by 24
- 2022Is an Object-Centric Video Representation Beneficial for Transfer?cited by 24
- 2025UAD: Unsupervised Affordance Distillation for Generalization in Robotic Manipulationcited by 23
- 2023ClothesNet: An Information-Rich 3D Garment Model Repository with Simulated Clothes Environmentcited by 22
- 2021Structured World Belief for Reinforcement Learning in POMDPcited by 21
- 2022Visuomotor Control in Multi-Object Scenes Using Object-Aware Representationscited by 21
- 2022Self-supervised video pretraining yields robust and more human-aligned visual representationscited by 20
- 20233D Implicit Transporter for Temporally Consistent Keypoint Discoverycited by 20
- 2021GANSeg: Learning to Segment by Unsupervised Hierarchical Image Generationcited by 19
- 2021COMPOSER: Compositional Reasoning of Group Activity in Videos with Keypoint-Only Modalitycited by 19
- 2022Unsupervised Image Representation Learning with Deep Latent Particlescited by 19
- 2023DDLP: Unsupervised Object-Centric Video Prediction with Deep Dynamic Latent Particlescited by 18
- 2021Learning Latent Graph Dynamics for Visual Manipulation of Deformable Objectscited by 17
- 2022AutoLink: Self-supervised Learning of Human Skeletons and Object Outlines by Linking Keypointscited by 17
- 2023SNeRL: Semantic-aware Neural Radiance Fields for Reinforcement Learningcited by 16
- 2023Contrastive Training of Complex-Valued Autoencoders for Object Discoverycited by 16
- 2023Model-Based Control with Sparse Neural Dynamicscited by 16
- 2024Composing Pre-Trained Object-Centric Representations for Robotics From "What" and "Where" Foundation Modelscited by 16
- 2022On the Feasibility of Cross-Task Transfer with Model-Based Reinforcement Learningcited by 15
- 2021Pushing it out of the Way: Interactive Visual Navigationcited by 14
- 2022Filtered-CoPhy: Unsupervised Learning of Counterfactual Physics in Pixel Spacecited by 14
- 2021Reasoning-Modulated Representationscited by 13
- 2023Detect Everything with Few Examplescited by 13
- 2023The Treachery of Images: Bayesian Scene Keypoints for Deep Policy Learning in Robotic Manipulationcited by 12
- 2021A Bayesian Treatment of Real-to-Sim for Deformable Object Manipulationcited by 11
- 2023Object-Centric Learning with Slot Mixture Modulecited by 11
- 2024Reinforcement Learning with Generalizable Gaussian Splattingcited by 11
50 most cited of 109 All 109 (JSON)
- 2026ParticleSplat: Self-supervised Object-centric Latent Particle Splatting16 Sep 2026
- 2026IMPASTO: Integrating Model-Based Planning with Learned Dynamics Models for Robotic Oil Painting Reproduction31 Mar 2026
- 2026Latent Particle World Models: Self-supervised Object-centric Stochastic Dynamics Modeling4 Mar 2026
- 2025Posterior-driven Heuristic Support Adaptation in a Probabilistic Treatment of Real2Sim2Real for Vision-Driven Deformable Linear Object Manipulation30 Oct 2025
- 2025VESSA: Video-based objEct-centric Self-Supervised Adaptation for Visual Foundation Models23 Oct 2025
- 2025CLASP: General-Purpose Clothes Manipulation with Semantic Keypoints26 Jul 2025
- 2025Unsupervised Monocular 3D Keypoint Discovery from Multi-View Diffusion Priors16 Jul 2025
- 2025Combining Pre-Trained Models for Enhanced Feature Representation in Reinforcement Learning9 Jul 2025
- 2025UAD: Unsupervised Affordance Distillation for Generalization in Robotic Manipulation10 Jun 2025
- 2025Revealing Human Attention Patterns from Gameplay Analysis for Reinforcement Learning15 Apr 2025
- 2025Are We Done with Object-Centric Learning?9 Apr 2025
- 2025Structure-Aware Correspondence Learning for Relative Pose Estimation24 Mar 2025
38 more
- 2025A Distributional Treatment of Real2Sim2Real for Object-Centric Agent Adaptation in Vision-Driven Deformable Linear Object Manipulation25 Feb 2025
- 2024Symbolic Disentangled Representations for Images25 Dec 2024
- 2024Learning from 10 Demos: Generalisable and Sample-Efficient Policy Learning with Oriented Affordance Frames15 Oct 2024
- 2024ReKep: Spatio-Temporal Reasoning of Relational Keypoint Constraints for Robotic Manipulation3 Sep 2024
- 2024General-purpose Clothes Manipulation with Semantic Keypoints15 Aug 2024
- 2024Unsupervised Part Discovery via Dual Representation Alignment15 Aug 2024
- 2024Deep Generative Models in Robotics: A Survey on Learning from Multimodal Demonstrations8 Aug 2024
- 2024Robotic Arm Manipulation with Inverse Reinforcement Learning & TD-MPC17 Jul 2024
- 2024Cycle-Correspondence Loss: Learning Dense View-Invariant Visual Features from Unlabeled and Unordered RGB Images18 Jun 2024
- 2024Composing Pre-Trained Object-Centric Representations for Robotics From "What" and "Where" Foundation Models20 Apr 2024
- 2024Pose-Guided Self-Training with Two-Stage Clustering for Unsupervised Landmark Discovery24 Mar 2024
- 2024Reinforcement Learning with Generalizable Gaussian Splatting18 Mar 2024
- 2024Learning Physical Dynamics for Object-centric Visual Prediction15 Mar 2024
- 2024To deform or not: treatment-aware longitudinal registration for breast DCE-MRI during neoadjuvant chemotherapy via unsupervised keypoints detection17 Jan 2024
- 2024Unsupervised Salient Patch Selection for Data-Efficient Reinforcement Learning10 Jan 2024
- 2023Causal State Distillation for Explainable Reinforcement Learning30 Dec 2023
- 2023Model-Based Control with Sparse Neural Dynamics20 Dec 2023
- 2023Tracking Object Positions in Reinforcement Learning: A Metric for Keypoint Detection (extended version)1 Dec 2023
- 2023Generalizable Imitation Learning Through Pre-Trained Representations15 Nov 2023
- 2023Object-Centric Learning with Slot Mixture Module8 Nov 2023
- 2023A Review and Roadmap of Deep Causal Model from Different Causal Structures and Representations2 Nov 2023
- 2023Learning visual-based deformable object rearrangement with local graph neural networks16 Oct 2023
- 2023SlotGNN: Unsupervised Discovery of Multi-Object Representations and Visual Dynamics6 Oct 2023
- 2023H-InDex: Visual Reinforcement Learning with Hand-Informed Representations for Dexterous Manipulation2 Oct 2023
- 2023Detect Everything with Few Examples22 Sep 2023
- 2023ForceSight: Text-Guided Mobile Manipulation with Visual-Force Goals21 Sep 2023
- 2023CLIPUNetr: Assisting Human-robot Interface for Uncalibrated Visual Servoing Control with CLIP-driven Referring Expression Segmentation17 Sep 2023
- 2023A Reinforcement Learning Approach for Robotic Unloading from Visual Observations12 Sep 2023
- 2023Learning Geometric Representations of Objects via Interaction11 Sep 2023
- 20233D Implicit Transporter for Temporally Consistent Keypoint Discovery10 Sep 2023
- 2023Self-supervised Scene Text Segmentation with Object-centric Layered Representations Augmented by Text Regions25 Aug 2023
- 2023ClothesNet: An Information-Rich 3D Garment Model Repository with Simulated Clothes Environment19 Aug 2023
- 2023Optical Flow boosts Unsupervised Localization and Segmentation25 Jul 2023
- 2023Crossway Diffusion: Improving Diffusion-based Visuomotor Policy via Self-supervised Learning4 Jul 2023
- 2023DDLP: Unsupervised Object-Centric Video Prediction with Deep Dynamic Latent Particles9 Jun 2023
- 2023Contrastive Training of Complex-Valued Autoencoders for Object Discovery24 May 2023
- 2023The Treachery of Images: Bayesian Scene Keypoints for Deep Policy Learning in Robotic Manipulation8 May 2023
- 2023Neural Constraint Satisfaction: Hierarchical Abstraction for Combinatorial Generalization in Object Rearrangement20 Mar 2023
50 newest of 109 All 109 (JSON)
- 2022VIOLA: Imitation Learning for Vision-Based Manipulation with Object Proposal Priorscited by 135
- 2020Model-Based Inverse Reinforcement Learning from Visual Demonstrationscited by 52
- 2024Deep Generative Models in Robotics: A Survey on Learning from Multimodal Demonstrationscited by 35
- 2021Generalization Through Hand-Eye Coordination: An Action Space for Learning Spatially-Invariant Visuomotor Controlcited by 24
- 2024Learning from 10 Demos: Generalisable and Sample-Efficient Policy Learning with Oriented Affordance Framescited by 9
- 2023Generalizable Imitation Learning Through Pre-Trained Representationscited by 3
- 2024Robotic Arm Manipulation with Inverse Reinforcement Learning & TD-MPCcited by 0
- 2024Learning from 10 Demos: Generalisable and Sample-Efficient Policy Learning with Oriented Affordance Frames15 Oct 2024
- 2024Deep Generative Models in Robotics: A Survey on Learning from Multimodal Demonstrations8 Aug 2024
- 2024Robotic Arm Manipulation with Inverse Reinforcement Learning & TD-MPC17 Jul 2024
- 2023Generalizable Imitation Learning Through Pre-Trained Representations15 Nov 2023
- 2022VIOLA: Imitation Learning for Vision-Based Manipulation with Object Proposal Priors20 Oct 2022
- 2021Generalization Through Hand-Eye Coordination: An Action Space for Learning Spatially-Invariant Visuomotor Control28 Feb 2021
- 2020Model-Based Inverse Reinforcement Learning from Visual Demonstrations18 Oct 2020
- 2025Unsupervised Monocular 3D Keypoint Discovery from Multi-View Diffusion Priors16 Jul 2025
- 2024Pose-Guided Self-Training with Two-Stage Clustering for Unsupervised Landmark Discovery24 Mar 2024
- 2023Crossway Diffusion: Improving Diffusion-based Visuomotor Policy via Self-supervised Learning4 Jul 2023
- 2021Deep Reinforcement Learning at the Edge of the Statistical Precipicecited by 781
- 2021CLIPort: What and Where Pathways for Robotic Manipulationcited by 617
- 2021Towards Causal Representation Learningcited by 207
- 2021PiCIE: Unsupervised Semantic Segmentation using Invariance and Equivariance in Clusteringcited by 163
- 2022Human-to-Robot Imitation in the Wildcited by 157
- 20213D Neural Scene Representations for Visuomotor Controlcited by 110
- 2021Learning Temporally Causal Latent Processes from General Temporal Datacited by 96
- 2021Learning by Watching: Physical Imitation of Manipulation Skills from Human Videoscited by 87
- 2021CoNeRF: Controllable Neural Radiance Fieldscited by 65
- 2022Learning Multi-Object Dynamics with Compositional Neural Radiance Fieldscited by 63
- 2022Equivariant Descriptor Fields: SE(3)-Equivariant Energy-Based Models for End-to-End Visual Robotic Manipulation Learningcited by 54
- 2022Visual Reinforcement Learning with Self-Supervised 3D Representationscited by 45
12 most cited of 94 All 94 (JSON)
- 2026ParticleSplat: Self-supervised Object-centric Latent Particle Splatting16 Sep 2026
- 2026IMPASTO: Integrating Model-Based Planning with Learned Dynamics Models for Robotic Oil Painting Reproduction31 Mar 2026
- 2025Posterior-driven Heuristic Support Adaptation in a Probabilistic Treatment of Real2Sim2Real for Vision-Driven Deformable Linear Object Manipulation30 Oct 2025
- 2025VESSA: Video-based objEct-centric Self-Supervised Adaptation for Visual Foundation Models23 Oct 2025
- 2025CLASP: General-Purpose Clothes Manipulation with Semantic Keypoints26 Jul 2025
- 2025Combining Pre-Trained Models for Enhanced Feature Representation in Reinforcement Learning9 Jul 2025
- 2025UAD: Unsupervised Affordance Distillation for Generalization in Robotic Manipulation10 Jun 2025
- 2025Revealing Human Attention Patterns from Gameplay Analysis for Reinforcement Learning15 Apr 2025
- 2025Are We Done with Object-Centric Learning?9 Apr 2025
- 2025Structure-Aware Correspondence Learning for Relative Pose Estimation24 Mar 2025
- 2025A Distributional Treatment of Real2Sim2Real for Object-Centric Agent Adaptation in Vision-Driven Deformable Linear Object Manipulation25 Feb 2025
- 2024Symbolic Disentangled Representations for Images25 Dec 2024
12 newest of 94 All 94 (JSON)