Unsupervised Keypoints from Pretrained Diffusion Models
Eric Hedlin, Gopal Sharma, Shweta Mahajan, Xingzhe He, Hossam Isack, Abhishek Kar, Helge Rhodin, Andrea Tagliasacchi, Kwang Moo Yi
Abstract
https://stablekeypoints.github.io/ Image dataset Randomly initialized tokens Optimized tokens Estimated keypoints Localize Figure 1. Teaser -we propose an unsupervised method to learn keypoints based on optimizing text embeddings of latent diffusion models [44]. Our method is motivated by the fact that random text tokens already respond roughly consistently to semantically similar regions. By promoting localization we obtain unsupervised keypoints that outperform the state-of-the-art.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers16
- Stable Diffusion Models Are Secretly Good at Visual In-Context LearningTrevine Oorloff, Vishwanath Sindagi, Wele Gedara Chaminda Bandara, Ali Shafahi et al.ICCV 2025 · 11 citations
- MotionShot: Adaptive Motion Transfer Across Arbitrary Objects for Text-to-Video GenerationYanchen Liu, Yanan Sun, Zhening Xing, Junyao Gao et al.ICCV 2025 · 5 citations
- Scene-Level Appearance Transfer with Semantic CorrespondencesLiyuan Zhu, Shengqu Cai, Shengyu Huang, Gordon Wetzstein et al.SIGGRAPH 2025 · 3 citations
- : Symmetry Understanding of 3D Shapes via Chirality DisentanglementWeikang Wang, Tobias Weißberg, Nafie El Amrani, Florian BernardICCV 2025 · 3 citations
- Pose Prior Learner: Unsupervised Categorical Prior Learning for Pose EstimationZiyu Wang, Shuangpeng Han, Mengmi ZhangICLR 2026 · 3 citations
Builds on23
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- Photorealistic Text-to-Image Diffusion Models with Deep Language UnderstandingChitwan Saharia, William Chan, Saurabh Saxena, Lala Li et al.NeurIPS 2022 · 8,965 citations
- Improved Denoising Diffusion Probabilistic ModelsAlexander Quinn Nichol, Prafulla DhariwalICML 2021 · 5,234 citations
- DiffusionDet: Diffusion Model for Object DetectionShoufa Chen, Peize Sun, Yibing Song, Ping LuoICCV 2023 · 715 citations
- Label-Efficient Semantic Segmentation with Diffusion ModelsDmitry Baranchuk, Andrey Voynov, Ivan Rubachev, Valentin Khrulkov et al.ICLR 2022 · 700 citations
Related papers
- Unsupervised Discovery of Facial Landmarks and Head PoseSatyajit Tourani, Siddharth Tourani, Arif Mahmood, Muhammad Haris KhanCVPR 2025
- Unsupervised Semantic Correspondence Using Stable DiffusionEric Hedlin, Gopal Sharma, Shweta Mahajan, Hossam Isack et al.NeurIPS 2023 · 152 citations
- NoiseCLR: A Contrastive Learning Approach for Unsupervised Discovery of Interpretable Directions in Diffusion ModelsYusuf Dalva, Pinar YanardagCVPR 2024
- SD4Match: Learning to Prompt Stable Diffusion Model for Semantic MatchingXinghui Li, Jingyi Lu, Kai Han, Victor Adrian PrisacariuCVPR 2024
- Repurposing Stable Diffusion Attention for Training-Free Unsupervised Interactive SegmentationMarkus Karmann, Onay UrfaliogluCVPR 2025
