A successful paradigm in representation learning is to perform self-supervised pretraining using tasks based on mini-batch statistics (e.g., SimCLR, VICReg, SwAV, MSN). We show that in the formulation of all these methods is an overlooked prior to learn features that enable uniform clustering of the data. While this prior has led to remarkably semantic representations when pretraining on class-balanced data, such as ImageNet, we demonstrate that it can hamper performance when pretraining on class-imbalanced data. By moving away from conventional uniformity priors and instead preferring power-law distributed feature clusters, we show that one can improve the quality of the learned representations on real-world class-imbalanced datasets. To demonstrate this, we develop an extension of the Masked Siamese Networks (MSN) method to support the use of arbitrary features priors.
@article{arxiv.2210.07277,
title = {The Hidden Uniform Cluster Prior in Self-Supervised Learning},
author = {Mahmoud Assran and Randall Balestriero and Quentin Duval and Florian Bordes and Ishan Misra and Piotr Bojanowski and Pascal Vincent and Michael Rabbat and Nicolas Ballas},
journal= {arXiv preprint arXiv:2210.07277},
year = {2022}
}