arXiv reaDer
ISD:反復類似性蒸留による自己教師あり学習
ISD: Self-Supervised Learning by Iterative Similarity Distillation
最近、対照学習は、他のランダム画像(負のペア)と比較して画像(正のペア)の2つの拡張を近づけることを主なアイデアとする自己教師あり学習で素晴らしい結果を達成しました。すべてのランダム画像が等しいわけではないと主張します。したがって、正と負のペアを2値で区別するのではなく、ネガティブ画像にソフトな類似性を使用する自己教師あり学習アルゴリズムを導入します。クエリ画像といくつかのランダム画像の類似性をキャプチャし、その知識を生徒に転送することで、ゆっくりと進化する教師モデルを生徒モデルに繰り返し抽出します。私たちの方法は、最近の対照的な学習方法と比較して制約が少ないため、より良い機能を学習できると主張します。具体的には、ランダムに選択されたネガティブセットには、クエリイメージに意味的に類似した多くのサンプルが含まれる可能性があるため、このメソッドは、既存の対照学習メソッドよりも不均衡でラベルのないデータをより適切に処理する必要があります。この場合、私たちの方法はそれらを非常に類似しているとラベル付けしますが、標準的な対照的な方法はそれらを負のペアとしてラベル付けします。私たちの方法は、最先端のモデルに匹敵する結果を達成します。また、ラベルのないデータが不均衡な設定で、この方法のパフォーマンスが向上することも示しています。コードはhttps://github.com/UMBCvision/ISDから入手できます。
Recently, contrastive learning has achieved great results in self-supervised learning, where the main idea is to push two augmentations of an image (positive pairs) closer compared to other random images (negative pairs). We argue that not all random images are equal. Hence, we introduce a self supervised learning algorithm where we use a soft similarity for the negative images rather than a binary distinction between positive and negative pairs. We iteratively distill a slowly evolving teacher model to the student model by capturing the similarity of a query image to some random images and transferring that knowledge to the student. We argue that our method is less constrained compared to recent contrastive learning methods, so it can learn better features. Specifically, our method should handle unbalanced and unlabeled data better than existing contrastive learning methods, because the randomly chosen negative set might include many samples that are semantically similar to the query image. In this case, our method labels them as highly similar while standard contrastive methods label them as negative pairs. Our method achieves comparable results to the state-of-the-art models. We also show that our method performs better in the settings where the unlabeled data is unbalanced. Our code is available here: https://github.com/UMBCvision/ISD.
updated: Mon May 10 2021 14:20:59 GMT+0000 (UTC)
published: Wed Dec 16 2020 20:50:17 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アソシエイト