arXiv reaDer
Swin Transformerによる自己教師付き学習
Self-Supervised Learning with Swin Transformers
コンピュータビジョンの分野では、CNNからトランスフォーマーへのモデリングの移行が進んでいる。本論文では、視覚トランスフォーマーをバックボーン・アーキテクチャとするMoBYと呼ばれる自己教師付き学習アプローチを紹介する。このアプローチには基本的に新しい発明はなく、MoCo v2とBYOLを組み合わせ、ImageNet-1Kの線形評価で合理的に高い精度を達成するように調整されている。DeiT-SとSwin-Tを用いて、300エポックの学習により、それぞれ72.8%と75.0%のトップ1精度を達成している。この性能は、DeiTをバックボーンとして採用しているMoCo v3やDINOなどの最近の研究よりも、より軽いトリックしか使わず、わずかに優れている。さらに重要なことは、汎用的なSwin Transformerをバックボーンとすることで、物体検出やセマンティック セグメンテーションなどの下流のタスクでも学習した表現を評価することができることである。対照的に、ViT/DeiTを用いた最近のいくつかのアプローチでは、ViT/DeiTがこれらの高密度予測タスクに対応していないため、ImageNet-1Kでの線形評価結果しか報告されていない。今回の結果が、トランスフォーマーアーキテクチャ用に設計された自己教師付き学習手法のより包括的な評価につながることを期待している。我々のコードとモデルは、 https://github.com/SwinTransformer/Transformer-SSL で公開されており、今後も継続的に充実させていく。
We are witnessing a modeling shift from CNN to Transformers in computer vision. In this work, we present a self-supervised learning approach called MoBY, with Vision Transformers as its backbone architecture. The approach basically has no new inventions, which is combined from MoCo v2 and BYOL and tuned to achieve reasonably high accuracy on ImageNet-1K linear evaluation: 72.8% and 75.0% top-1 accuracy using DeiT-S and Swin-T, respectively, by 300-epoch training. The performance is slightly better than recent works of MoCo v3 and DINO which adopt DeiT as the backbone, but with much lighter tricks. More importantly, the general-purpose Swin Transformer backbone enables us to also evaluate the learnt representations on downstream tasks such as object detection and semantic segmentation, in contrast to a few recent approaches built on ViT/DeiT which only report linear evaluation results on ImageNet-1K due to ViT/DeiT not tamed for these dense prediction tasks. We hope our results can facilitate more comprehensive evaluation of self-supervised learning methods designed for Transformer architectures. Our code and models are available at https://github.com/SwinTransformer/Transformer-SSL, which will be continually enriched.
updated: Tue May 11 2021 17:28:00 GMT+0000 (UTC)
published: Mon May 10 2021 17:59:45 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アソシエイト