arXiv reaDer
対照学習における拡張モジュールの再考:拡張ビューによる階層的拡張不変性の学習
Rethinking the Augmentation Module in Contrastive Learning: Learning Hierarchical Augmentation Invariance with Expanded Views
データ拡張モジュールは、対照的な学習で利用され、与えられたデータの例を2つのビューに変換します。これは、不可欠でかけがえのないものと見なされます。ただし、複数のデータ拡張の所定の構成には、2つの欠点があります。まず、拡張タイプを人為的に選択すると、モデルに特定の表現の不変性がもたらされます。これは、さまざまなダウンストリームタスクにさまざまな程度の正と負の影響を及ぼします。トレーニング中に各タイプの拡張を均等に処理すると、モデルはさまざまなダウンストリームタスクの最適でない表現を学習し、拡張タイプを事前に選択する柔軟性が制限されます。第2に、従来の対照的な学習方法で使用される強力なデータ拡張は、場合によっては不変性が大きすぎる可能性があり、一部のダウンストリームタスクに不可欠なきめ細かい情報が失われる可能性があります。この論文は、一般的な対照学習フレームワークのどこで何を対比するかを検討することにより、これら2つの問題を軽減する一般的な方法を提案します。最初に、バックボーンで表現の不変性を均等に学習するのではなく、各データ拡張の重要性に応じて、モデルのさまざまな深さでさまざまな拡張不変性を学習することを提案します。次に、強力なデータ拡張の誤解を招く影響を減らすために、拡張埋め込みを使用してコントラストコンテンツを拡張することを提案します。いくつかのベースライン方法に基づく実験は、分類、検出、およびセグメンテーションのダウンストリームタスクに関するさまざまなベンチマークのより良い表現を学習することを示しています。
A data augmentation module is utilized in contrastive learning to transform the given data example into two views, which is considered essential and irreplaceable. However, the predetermined composition of multiple data augmentations brings two drawbacks. First, the artificial choice of augmentation types brings specific representational invariances to the model, which have different degrees of positive and negative effects on different downstream tasks. Treating each type of augmentation equally during training makes the model learn non-optimal representations for various downstream tasks and limits the flexibility to choose augmentation types beforehand. Second, the strong data augmentations used in classic contrastive learning methods may bring too much invariance in some cases, and fine-grained information that is essential to some downstream tasks may be lost. This paper proposes a general method to alleviate these two problems by considering where and what to contrast in a general contrastive learning framework. We first propose to learn different augmentation invariances at different depths of the model according to the importance of each data augmentation instead of learning representational invariances evenly in the backbone. We then propose to expand the contrast content with augmentation embeddings to reduce the misleading effects of strong data augmentations. Experiments based on several baseline methods demonstrate that we learn better representations for various benchmarks on classification, detection, and segmentation downstream tasks.
updated: Mon Aug 22 2022 03:28:52 GMT+0000 (UTC)
published: Wed Jun 01 2022 04:30:46 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アソシエイト