arXiv reaDer
ツインクラス分布を推定することによる自己監視学習
Self-Supervised Learning by Estimating Twin Class Distributions
大規模なラベルなしデータセットをエンドツーエンドの方法で分類することにより、新しい自己監視表現学習方法であるTWISTを紹介します。ソフトマックス演算で終了するシャムネットワークを使用して、2つの拡張画像のツインクラス分布を生成します。監督なしで、一貫性を保つためにさまざまな拡張のクラス分布を強制します。その間、クラスの分布を正規化して、シャープで多様なものにします。具体的には、各サンプルの分布のエントロピーを最小化して各サンプルのクラス予測を断定的にし、平均分布のエントロピーを最大化してさまざまなサンプルの予測を多様化します。このようにして、TWISTは、非対称ネットワーク、停止勾配操作、運動量エンコーダーなどの特定の設計がなくても、簡単なソリューションを自然に回避できます。クラスタリングと学習を交互に行うクラスタリングベースの方法とは異なり、私たちの方法は、統一された損失関数によって導かれる単一の学習プロセスです。その結果、TWISTは、教師なし分類、線形分類、半教師あり学習、転移学習、および検出やセグメンテーションなどのいくつかの高密度予測タスクを含む、幅広いタスクで最先端の方法よりも優れています。
We present TWIST, a novel self-supervised representation learning method by classifying large-scale unlabeled datasets in an end-to-end way. We employ a siamese network terminated by a softmax operation to produce twin class distributions of two augmented images. Without supervision, we enforce the class distributions of different augmentations to be consistent. In the meantime, we regularize the class distributions to make them sharp and diverse. Specifically, we minimize the entropy of the distribution for each sample to make the class prediction for each sample assertive and maximize the entropy of the mean distribution to make the predictions of different samples diverse. In this way, TWIST can naturally avoid the trivial solutions without specific designs such as asymmetric network, stop-gradient operation, or momentum encoder. Different from the clustering-based methods which alternate between clustering and learning, our method is a single learning process guided by a unified loss function. As a result, TWIST outperforms state-of-the-art methods on a wide range of tasks, including unsupervised classification, linear classification, semi-supervised learning, transfer learning, and some dense prediction tasks such as detection and segmentation.
updated: Thu Oct 14 2021 14:39:39 GMT+0000 (UTC)
published: Thu Oct 14 2021 14:39:39 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アソシエイト