arXiv reaDer
表現学習のための対照的な引力と対照的な反発
Contrastive Attraction and Contrastive Repulsion for Representation Learning
対比学習(CL)メソッドは、ラベルの監視なしでデータ表現を効果的に学習します。エンコーダーは、1対多のソフトマックスクロスエントロピー損失を介して、複数の負のサンプルに対して各正のサンプルを対比します。最近のCLメソッドは、ラベル付けされていない大量の画像データを活用することで、バランスの取れた画像クラスを備えた厳選されたデータセットであるImageNetで事前トレーニングしたときに有望な結果を達成しています。ただし、実際の画像で事前トレーニングを行うと、パフォーマンスが低下する傾向があります。このホワイトペーパーでは、CLのパフォーマンスをさらに向上させ、キュレートされていないデータセットに対する堅牢性を強化するために、クエリのポジティブ(ネガティブ)サンプルを、それらをプル(プッシュ)する強さを決定する前に対比する二重CL戦略を提案します。この戦略は、対照的な引力と対照的な反発(CACR)で実現されます。これにより、クエリは、より遠い正のサンプルを引き付けるだけでなく、より近い負のサンプルをはじくためにより大きな力を発揮します。理論的分析により、CACRは、一般にクエリとは独立してサンプリングされる正/負のサンプルの分布と、クエリが与えられた場合の真の条件付き分布との違いを考慮して、CLの動作を一般化することがわかります。データとその潜在的表現の両方で均一な事前分布を仮定する必要性を排除するのに役立つ、このユニークなポジティブ内引力とネガティブ内反発メカニズムが、データセットのキュレーションが少ない場合に特に有益であることを示します。多数の標準的な視覚タスクに関する大規模な大規模な実験により、CACRは、表現学習のベンチマークデータセットで既存のCLメソッドを一貫して上回っているだけでなく、不均衡な画像データセットでの事前トレーニングに一般化した場合に優れた堅牢性を示します。
Contrastive learning (CL) methods effectively learn data representations without label supervision, where the encoder contrasts each positive sample over multiple negative samples via a one-vs-many softmax cross-entropy loss. By leveraging large amounts of unlabeled image data, recent CL methods have achieved promising results when pre-trained on ImageNet, a well-curated data set with balanced image classes. However, they tend to yield worse performance when pre-trained on images in the wild. In this paper, to further improve the performance of CL and enhance its robustness on uncurated data sets, we propose a doubly CL strategy that contrasts the positive (negative) samples of a query within themselves before deciding how strongly to pull (push) them. We realize this strategy with contrastive attraction and contrastive repulsion (CACR), which makes the query not only exert a greater force to attract more distant positive samples but also do so to repel closer negative samples. Theoretical analysis reveals that CACR generalizes CL's behavior by taking into consideration the differences between the distributions of the positive/negative samples, which in general are sampled independently of the query, and their true conditional distributions given the query. We demonstrate this unique intra-positive attraction and intra-negative repulsion mechanism, which helps remove the need to assume uniform prior distributions on both the data and their latent representation, is particularly beneficial when data sets are less curated. Extensive large-scale experiments on a number of standard vision tasks show that CACR not only consistently outperforms existing CL methods on benchmark data sets in representation learning, but also shows better robustness when generalized to pre-training on imbalanced image data sets.
updated: Mon Jul 25 2022 18:04:24 GMT+0000 (UTC)
published: Sat May 08 2021 17:25:08 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アソシエイト