自己教師あり表現学習方法のほとんどは、対照的な損失とインスタンス識別タスクに基づいており、同じ画像インスタンスの拡張バージョン(「ポジティブ」)が他の画像から抽出されたインスタンス(「ネガティブ」)と対比されます。学習を効果的にするためには、多くのネガティブをポジティブペアと比較する必要があります。これは計算量が多くなります。本論文では、潜在空間特徴の白色化に基づく自己教師あり表現学習のための異なる方向と新しい損失関数を提案した。ホワイトニング操作は、バッチサンプルに「散乱」効果をもたらします。これにより、ネガの使用が補正され、すべてのサンプル表現が1点に崩壊する縮退解が回避されます。 Whitening MSE(W-MSE)の損失は、特別なヒューリスティック(追加のネットワークなど)を必要とせず、概念的に単純です。ネガは必要ないので、同じ画像インスタンスから複数のポジティブペアを抽出できます。 W-MSEは、一般的でより複雑な自己監視方式に関して競争力があることを経験的に示しています。メソッドとすべての実験のソースコードは、https://github.com/htdt/self-supervisedで入手できます。
Most of the 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, a lot of 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 special heuristics (e.g. additional networks) and it is conceptually simple. 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 all the experiments is available at https://github.com/htdt/self-supervised.