An Empirical Study on Clustering Pretrained Embeddings: Is Deep Strictly Better?
Not on the map. Why?
Abstract (arXiv)
Recent research in clustering face embeddings has found that unsupervised, shallow, heuristic-based methods -- including k-means and hierarchical agglomerative clustering -- underperform supervised, deep, inductive methods. While the reported improvements are indeed impressive, experiments are mostly limited to face datasets, where the clustered embeddings are highly discriminative or well-separated by class (Recall@1 above 90% and often nearing ceiling), and the experimental methodology seemingly favors the deep methods. We conduct a large-scale empirical study of 17 clustering methods across three datasets and obtain several robust findings. Notably, deep methods are surprisingly fragile for embeddings with more uncertainty, where they match or even perform worse than shallow, heuristic-based methods. When embeddings are highly discriminative, deep methods do outperform the baselines, consistent with past results, but the margin between methods is much smaller than previously reported. We believe our benchmarks broaden the scope of supervised clustering methods beyond the face domain and can serve as a foundation on which these methods could be improved. To enable reproducibility, we include all necessary details in the appendices, and plan to release the code.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 28 of its 54 references that are in the map. First posted on arXiv 9 Nov 2022.
Builds on
28 of 54 references.
References · 28
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2017Prototypical Networks for Few-shot Learningcited by 3,001
- 2015FaceNet: A Unified Embedding for Face Recognition and Clusteringcited by 2,833
- 2018CosFace: Large Margin Cosine Loss for Deep Face Recognitioncited by 968
- 2017SphereFace: Deep Hypersphere Embedding for Face Recognitioncited by 733
- 2016MS-Celeb-1M: A Dataset and Benchmark for Large-Scale Face Recognitioncited by 621
- 2015Deep Metric Learning via Lifted Structured Feature Embeddingcited by 449
- 2020Rethinking Few-Shot Image Classification: a Good Embedding Is All You Need?cited by 435
- 2020A Metric Learning Reality Checkcited by 317
- 2018Additive Margin Softmax for Face Verificationcited by 275
- 2017NormFace: L2 Hypersphere Embedding for Face Verificationcited by 237
16 more
- 2015The MegaFace Benchmark: 1 Million Faces for Recognition at Scalecited by 199
- 2020A unifying mutual information view of metric learning: cross-entropy vs. pairwise lossescited by 100
- 2017L2-constrained Softmax Loss for Discriminative Face Verificationcited by 96
- 2019Linkage Based Face Clustering via Graph Convolution Networkcited by 55
- 2019Learning to Cluster Faces on an Affinity Graphcited by 50
- 2020Learning to Cluster Faces via Confidence and Connectivity Estimationcited by 42
- 2021von Mises-Fisher Loss: An Exploration of Embedding Geometries for Supervised Learningcited by 42
- 2016Clustering Millions of Faces by Identitycited by 26
- 2016A Survey of Inductive Biases for Factorial Representation-Learningcited by 23
- 2021Structure-Aware Face Clustering on a Large-Scale Graph with bf107 Nodescited by 21
- 2018Consensus-Driven Propagation in Massive Unlabeled Data for Face Recognitioncited by 20
- 2017Face Clustering: Representation and Pairwise Constraintscited by 14
- 2018Crystal Loss and Quality Pooling for Unconstrained Face Verification and Recognitioncited by 13
- 2017Merge or Not? Learning to Group Faces via Imitation Learningcited by 5
- 2021Learning to Cluster Faces via Transformercited by 5
- 2016Small-Variance Nonparametric Clustering on the Hyperspherecited by 3
Built on by
Citing papers per year: 2025: 1.
1 paper.