arXiv reaDer
自己蒸留自己監視表現学習
Self-Distilled Self-Supervised Representation Learning
自己監視学習の最先端のフレームワークは、最近、トランスベースのモデルを完全に利用すると、従来のCNNモデルと比較してパフォーマンスが向上する可能性があることを示しています。画像の2つのビューの相互情報量を最大化するように努めており、既存の作品は最終的な表現に対照的な損失を適用します。監視されたレジームでの自己蒸留に動機付けられて、中間表現が対照的な損失を介して最終層から学習できるようにすることで、これをさらに活用します。自己蒸留により、中間層はインスタンスの識別に適しているため、早期に終了したサブネットワークのパフォーマンスは、完全なネットワークのパフォーマンスからそれほど低下しません。これにより、最終レイヤーでも口実タスクが簡単になり、表現が向上します。私たちの方法であるSelf-DistilledSelf-SupervisedLearning(SDSSL)は、さまざまなタスクとデータセットでViTを使用して、競合ベースライン(SimCLR、BYOL、およびMoCo v3)よりも優れています。線形評価とk-NNプロトコルでは、SDSSLは最終層だけでなく、ほとんどの下位層でも優れたパフォーマンスをもたらします。さらに、正と負の配置は、表現がより効果的に形成される方法を説明するために使用されます。コードが利用可能になります。
State-of-the-art frameworks in self-supervised learning have recently shown that fully utilizing transformer-based models can lead to performance boost compared to conventional CNN models. Striving to maximize the mutual information of two views of an image, existing works apply a contrastive loss to the final representations. Motivated by self-distillation in the supervised regime, we further exploit this by allowing the intermediate representations to learn from the final layer via the contrastive loss. Through self-distillation, the intermediate layers are better suited for instance discrimination, making the performance of an early-exited sub-network not much degraded from that of the full network. This renders the pretext task easier also for the final layer, lead to better representations. Our method, Self-Distilled Self-Supervised Learning (SDSSL), outperforms competitive baselines (SimCLR, BYOL and MoCo v3) using ViT on various tasks and datasets. In the linear evaluation and k-NN protocol, SDSSL not only leads to superior performance in the final layers, but also in most of the lower layers. Furthermore, positive and negative alignments are used to explain how representations are formed more effectively. Code will be available.
updated: Tue Jun 21 2022 11:00:16 GMT+0000 (UTC)
published: Thu Nov 25 2021 07:52:36 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アソシエイト