ArCL: Enhancing Contrastive Learning with Augmentation-Robust Representations
Not on the map. Why?
Abstract (arXiv)
Self-Supervised Learning (SSL) is a paradigm that leverages unlabeled data for model training. Empirical studies show that SSL can achieve promising performance in distribution shift scenarios, where the downstream and training distributions differ. However, the theoretical understanding of its transferability remains limited. In this paper, we develop a theoretical framework to analyze the transferability of self-supervised contrastive learning, by investigating the impact of data augmentation on it. Our results reveal that the downstream performance of contrastive learning depends largely on the choice of data augmentation. Moreover, we show that contrastive learning fails to learn domain-invariant features, which limits its transferability. Based on these theoretical insights, we propose a novel method called Augmentation-robust Contrastive Learning (ArCL), which guarantees to learn domain-invariant features and can be easily integrated with existing contrastive learning algorithms. We conduct experiments on several datasets and show that ArCL significantly improves the transferability of contrastive learning.
Key facts. Cited by 10 papers in the map (data through 7 Oct 2026). Builds on 42 of its 57 references that are in the map. First posted on arXiv 2 Mar 2023.
Builds on
42 of 57 references.
References · 42
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2020Exploring Simple Siamese Representation Learningcited by 2,853
- 2013Describing Textures in the Wildcited by 2,110
- 2019Invariant Risk Minimizationcited by 1,832
- 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,818
- 2013Fine-Grained Visual Classification of Aircraftcited by 1,701
- 2019Contrastive Multiview Codingcited by 1,297
- 2018Learning deep representations by mutual information estimation and maximizationcited by 1,265
- 2021Generalizing to Unseen Domains: A Survey on Domain Generalizationcited by 762
- 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 737
30 more
- 2013Domain Generalization via Invariant Feature Representationcited by 622
- 2018Disentangling by Factorisingcited by 614
- 2015Causal inference using invariant prediction: identification and confidence intervalscited by 533
- 2019A Theoretical Analysis of Contrastive Unsupervised Representation Learningcited by 530
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 431
- 2020The Pitfalls of Simplicity Bias in Neural Networkscited by 349
- 2018Learning Models with Uniform Performance via Distributionally Robust Optimizationcited by 334
- 2019On Mutual Information Maximization for Representation Learningcited by 315
- 2021Self-Supervised Learning with Data Augmentations Provably Isolates Content from Stylecited by 308
- 2020Domain Generalization using Causal Matchingcited by 224
- 2020How Well Do Self-Supervised Models Transfer?cited by 213
- 2021Self-supervised Pretraining of Visual Features in the Wildcited by 197
- 2016Statistics of Robust Optimization: A Generalized Empirical Likelihood Approachcited by 182
- 2020Robust Pre-Training by Adversarial Contrastive Learningcited by 144
- 2020Adversarial Self-Supervised Contrastive Learningcited by 138
- 2021Self-supervised Learning is More Robust to Dataset Imbalancecited by 122
- 2021Generative Models as a Data Source for Multiview Representation Learningcited by 104
- 2021Towards the Generalization of Contrastive Self-Supervised Learningcited by 99
- 2020On Disentangled Representations Learned From Correlated Datacited by 94
- 2021Towards a Theoretical Framework of Out-of-Distribution Generalizationcited by 94
- 2019Stable Learning via Sample Reweightingcited by 70
- 2022Connect, Not Collapse: Explaining Contrastive Learning for Unsupervised Domain Adaptationcited by 66
- 2020Stable Prediction with Model Misspecification and Agnostic Distribution Shiftcited by 62
- 2020Learning Causal Semantic Representation for Out-of-Distribution Predictioncited by 60
- 2021Investigating the Role of Negatives in Contrastive Representation Learningcited by 54
- 2021Self-Supervised Learning Disentangled Group Representation as Featurecited by 47
- 2022Beyond Separability: Analyzing the Linear Transferability of Contrastive Representations to Related Subpopulationscited by 44
- 2022Your Contrastive Learning Is Secretly Doing Stochastic Neighbor Embeddingcited by 28
- 2021On the Surrogate Gap between Contrastive and Supervised Lossescited by 25
- 2020Enforcing Predictive Invariance across Structured Biomedical Domainscited by 23
Built on by
Citing papers per year: 2023: 6; 2024: 4.
10 papers.
- 2023Self-Supervised Learning for Time Series Analysis: Taxonomy, Progress, and Prospectscited by 97
- 2024GraphCLIP: Enhancing Transferability in Graph Foundation Models for Text-Attributed Graphscited by 58
- 2023Know Your Self-supervised Learning: A Survey on Image-based Generative and Discriminative Trainingcited by 39
- 2023Unsupervised Representation Learning for Time Series: A Reviewcited by 14
- 2024Adversarial Representation Engineering: A General Model Editing Framework for Large Language Modelscited by 12
- 2024Tri-Level Navigator: LLM-Empowered Tri-Level Learning for Time Series OOD Generalizationcited by 12
- 2023Task-Oriented Koopman-Based Control with Contrastive Encodercited by 8
- 2023MARIO: Model Agnostic Recipe for Improving OOD Generalization of Graph Contrastive Learningcited by 7
- 2024Look Ahead or Look Around? A Theoretical Comparison Between Autoregressive and Masked Pretrainingcited by 7
- 2023Towards Understanding the Mechanism of Contrastive Learning via Similarity Structure: A Theoretical Analysiscited by 6
- 2024GraphCLIP: Enhancing Transferability in Graph Foundation Models for Text-Attributed Graphs14 Oct 2024
- 2024Tri-Level Navigator: LLM-Empowered Tri-Level Learning for Time Series OOD Generalization9 Oct 2024
- 2024Look Ahead or Look Around? A Theoretical Comparison Between Autoregressive and Masked Pretraining1 Jul 2024
- 2024Adversarial Representation Engineering: A General Model Editing Framework for Large Language Models21 Apr 2024
- 2023Task-Oriented Koopman-Based Control with Contrastive Encoder28 Sep 2023
- 2023Unsupervised Representation Learning for Time Series: A Review3 Aug 2023
- 2023MARIO: Model Agnostic Recipe for Improving OOD Generalization of Graph Contrastive Learning24 Jul 2023
- 2023Self-Supervised Learning for Time Series Analysis: Taxonomy, Progress, and Prospects16 Jun 2023
- 2023Know Your Self-supervised Learning: A Survey on Image-based Generative and Discriminative Training23 May 2023
- 2023Towards Understanding the Mechanism of Contrastive Learning via Similarity Structure: A Theoretical Analysis1 Apr 2023