arXiv reaDer
自己教師あり表現学習のためのホワイトニング
Whitening for Self-Supervised Representation Learning
現在の自己教師あり表現学習方法のほとんどは、対照的な損失とインスタンス識別タスクに基づいており、同じ画像インスタンスの拡張バージョン(「ポジティブ」)が他の画像から抽出されたインスタンス(「ネガティブ」)と対比されます。学習を効果的にするためには、多くのネガティブをポジティブペアと比較する必要があります。これは計算量が多くなります。本論文では、潜在空間特徴の白色化に基づく自己教師あり表現学習のための異なる方向と新しい損失関数を提案した。ホワイトニング操作は、バッチサンプルに「散乱」効果をもたらします。これにより、ネガの使用が補正され、すべてのサンプル表現が1点に崩壊する縮退解が回避されます。ホワイトニングMSE(W-MSE)の損失は、追加の運動量ネットワークを必要とせず、概念的に単純です。さらに、ネガは必要ないため、同じ画像インスタンスから複数のポジティブペアを抽出できます。 W-MSEは、一般的でより複雑な自己監視方式に関して競争力があることを経験的に示しています。メソッドとすべての実験のソースコードは、https://github.com/htdt/self-supervisedで入手できます。
Most of the current self-supervised representation learning methods are based on the contrastive loss and the instance-discrimination task, where augmented versions of the same image instance ("positives") are contrasted with instances extracted from other images ("negatives"). For the learning to be effective, many negatives should be compared with a positive pair, which is computationally demanding. In this paper, we propose a different direction and a new loss function for self-supervised representation learning which is based on the whitening of the latent-space features. The whitening operation has a "scattering" effect on the batch samples, which compensates the use of negatives, avoiding degenerate solutions where all the sample representations collapse to a single point. Our Whitening MSE (W-MSE) loss does not require additional momentum networks and it is conceptually simple. Moreover, since negatives are not needed, we can extract multiple positive pairs from the same image instance. We empirically show that W-MSE is competitive with respect to popular, more complex self-supervised methods. The source code of the method and of all the experiments is available at https://github.com/htdt/self-supervised.
updated: Mon Feb 08 2021 11:05:58 GMT+0000 (UTC)
published: Mon Jul 13 2020 12:33:25 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アソシエイト