arXiv reaDer
対照学習の良い見方には何が必要ですか?
What Makes for Good Views for Contrastive Learning?
データの複数のビュー間の対照的な学習は、最近、自己教師あり表現学習の分野で最先端のパフォーマンスを達成しました。その成功にもかかわらず、さまざまなビューの選択の影響はあまり研究されていません。このホワイトペーパーでは、理論的および経験的分析を使用して、ビュー選択の重要性をよりよく理解し、タスク関連情報をそのまま維持しながら、ビュー間の相互情報量(MI)を削減する必要があると主張します。この仮説を検証するために、MIの削減を目的として、効果的なビューを学習する教師なしおよび半教師ありフレームワークを考案します。また、MIを削減する方法としてデータの拡張を検討し、データの拡張を増やすと実際にMIが減少し、ダウンストリームの分類精度が向上することを示します。副産物として、ImageNet分類の教師なし事前トレーニングで新しい最先端の精度を実現します(ResNet-50で73%のトップ1線形読み出し)。さらに、モデルをPASCAL VOCオブジェクト検出とCOCOインスタンスセグメンテーションに転送すると、教師あり事前トレーニングよりも一貫して優れたパフォーマンスを発揮します。コード:http://github.com/HobbitLong/PyContrast
Contrastive learning between multiple views of the data has recently achieved state of the art performance in the field of self-supervised representation learning. Despite its success, the influence of different view choices has been less studied. In this paper, we use theoretical and empirical analysis to better understand the importance of view selection, and argue that we should reduce the mutual information (MI) between views while keeping task-relevant information intact. To verify this hypothesis, we devise unsupervised and semi-supervised frameworks that learn effective views by aiming to reduce their MI. We also consider data augmentation as a way to reduce MI, and show that increasing data augmentation indeed leads to decreasing MI and improves downstream classification accuracy. As a by-product, we achieve a new state-of-the-art accuracy on unsupervised pre-training for ImageNet classification (73% top-1 linear readout with a ResNet-50). In addition, transferring our models to PASCAL VOC object detection and COCO instance segmentation consistently outperforms supervised pre-training. Code:http://github.com/HobbitLong/PyContrast
updated: Fri Dec 18 2020 10:01:34 GMT+0000 (UTC)
published: Wed May 20 2020 17:59:57 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アソシエイト