arXiv reaDer
ArCL: 増強ロバスト表現による対照学習の強化
ArCL: Enhancing Contrastive Learning with Augmentation-Robust Representations
自己教師あり学習 (SSL) は、モデルのトレーニングにラベル付けされていないデータを活用するパラダイムです。実証研究によると、SSL は、ダウンストリーム ディストリビューションとトレーニング ディストリビューションが異なるディストリビューション シフト シナリオで有望なパフォーマンスを達成できることが示されています。ただし、その転送可能性の理論的理解は限られています。この論文では、自己教師あり対照学習の転送可能性を分析するための理論的フレームワークを開発し、データ拡張の影響を調査します。私たちの結果は、対照学習の下流のパフォーマンスがデータ拡張の選択に大きく依存することを明らかにしています。さらに、対照学習はドメイン不変の機能を学習できず、その転送可能性が制限されることを示します。これらの理論的洞察に基づいて、Augmentation-robust Contrastive Learning (ArCL) と呼ばれる新しい方法を提案します。これは、ドメイン不変の特徴を学習することを保証し、既存の対照学習アルゴリズムと簡単に統合できます。いくつかのデータセットで実験を行い、ArCL が対照学習の伝達性を大幅に改善することを示します。
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.
updated: Thu Mar 02 2023 09:26:20 GMT+0000 (UTC)
published: Thu Mar 02 2023 09:26:20 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト