arXiv reaDer
同変対照学習
Equivariant Contrastive Learning
最先端の自己監視学習(SSL)では、事前トレーニングは、人間の知識から規定された意味のある変換の下で不変になるように促すことにより、意味的に優れた表現を生成します。実際、不変のプロパティは、同変と呼ばれるより広いクラスの些細なインスタンスです。これは、入力の変換方法に従って表現が変換されるプロパティとして直感的に理解できます。ここでは、不変性のみを使用するのではなく、他の変換への不変性を維持しながら、一部の変換への自明でない同変を促進する事前トレーニングを使用して、表現のセマンティック品質を向上できることを示します。具体的には、一般的なSSLメソッドを、Equivariant Self-Supervised Learning(E-SSL)という名前のより一般的なフレームワークに拡張します。 E-SSLでは、単純な追加の事前トレーニング目標が、入力に適用される変換を予測することによって同変を促進します。 E-SSLの有効性は、いくつかの一般的なコンピュータービジョンベンチマークで経験的に実証されています。たとえば、ImageNetでSimCLRを72.5%の線形プローブ精度に改善します。さらに、コンピュータビジョンを超えたアプリケーションでのE-SSLの有用性を示します。特に、フォトニクス科学における回帰問題でのその有用性を示します。私たちのコード、データセット、および事前にトレーニングされたモデルは、https://github.com/rdangovs/esslで入手でき、E-SSLのさらなる調査に役立ちます。
In state-of-the-art self-supervised learning (SSL) pre-training produces semantically good representations by encouraging them to be invariant under meaningful transformations prescribed from human knowledge. In fact, the property of invariance is a trivial instance of a broader class called equivariance, which can be intuitively understood as the property that representations transform according to the way the inputs transform. Here, we show that rather than using only invariance, pre-training that encourages non-trivial equivariance to some transformations, while maintaining invariance to other transformations, can be used to improve the semantic quality of representations. Specifically, we extend popular SSL methods to a more general framework which we name Equivariant Self-Supervised Learning (E-SSL). In E-SSL, a simple additional pre-training objective encourages equivariance by predicting the transformations applied to the input. We demonstrate E-SSL's effectiveness empirically on several popular computer vision benchmarks, e.g. improving SimCLR to 72.5% linear probe accuracy on ImageNet. Furthermore, we demonstrate usefulness of E-SSL for applications beyond computer vision; in particular, we show its utility on regression problems in photonics science. Our code, datasets and pre-trained models are available at https://github.com/rdangovs/essl to aid further research in E-SSL.
updated: Tue Mar 15 2022 01:42:01 GMT+0000 (UTC)
published: Thu Oct 28 2021 17:21:33 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アソシエイト