References & Citations
Computer Science > Computer Vision and Pattern Recognition
Title: Bag of Image Patch Embedding Behind the Success of Self-Supervised Learning
(Submitted on 17 Jun 2022 (v1), last revised 13 Jun 2023 (this version, v2))
Abstract: Self-supervised learning (SSL) has recently achieved tremendous empirical advancements in learning image representation. However, our understanding of the principle behind learning such a representation is still limited. This work shows that joint-embedding SSL approaches primarily learn a representation of image patches, which reflects their co-occurrence. Such a connection to co-occurrence modeling can be established formally, and it supplements the prevailing invariance perspective. We empirically show that learning a representation for fixed-scale patches and aggregating local patch representations as the image representation achieves similar or even better results than the baseline methods. We denote this process as BagSSL. Even with 32x32 patch representation, BagSSL achieves 62% top-1 linear probing accuracy on ImageNet. On the other hand, with a multi-scale pretrained model, we show that the whole image embedding is approximately the average of local patch embeddings. While the SSL representation is relatively invariant at the global scale, we show that locality is preserved when we zoom into local patch-level representation. Further, we show that patch representation aggregation can improve various SOTA baseline methods by a large margin. The patch representation is considerably easier to understand, and this work makes a step to demystify self-supervised representation learning.
Submission history
From: Yubei Chen [view email][v1] Fri, 17 Jun 2022 18:11:23 GMT (14236kb,D)
[v2] Tue, 13 Jun 2023 00:48:40 GMT (14424kb,D)
Link back to: arXiv, form interface, contact.