English

Clustering Properties of Self-Supervised Learning

Machine Learning 2025-05-13 v2 Artificial Intelligence

Abstract

Self-supervised learning (SSL) methods via joint embedding architectures have proven remarkably effective at capturing semantically rich representations with strong clustering properties, magically in the absence of label supervision. Despite this, few of them have explored leveraging these untapped properties to improve themselves. In this paper, we provide an evidence through various metrics that the encoder's output encodingencoding exhibits superior and more stable clustering properties compared to other components. Building on this insight, we propose a novel positive-feedback SSL method, termed Representation Self-Assignment (ReSA), which leverages the model's clustering properties to promote learning in a self-guided manner. Extensive experiments on standard SSL benchmarks reveal that models pretrained with ReSA outperform other state-of-the-art SSL methods by a significant margin. Finally, we analyze how ReSA facilitates better clustering properties, demonstrating that it effectively enhances clustering performance at both fine-grained and coarse-grained levels, shaping representations that are inherently more structured and semantically meaningful.

Keywords

Cite

@article{arxiv.2501.18452,
  title  = {Clustering Properties of Self-Supervised Learning},
  author = {Xi Weng and Jianing An and Xudong Ma and Binhang Qi and Jie Luo and Xi Yang and Jin Song Dong and Lei Huang},
  journal= {arXiv preprint arXiv:2501.18452},
  year   = {2025}
}

Comments

Accepted at ICML 2025

R2 v1 2026-06-28T21:25:51.696Z