このホワイトペーパーでは、Swin Transformerと呼ばれる新しいビジョンTransformerを紹介します。これは、コンピュータービジョンの汎用バックボーンとして機能します。 Transformerを言語から視覚に適応させる際の課題は、視覚エンティティのスケールの大きな変動や、テキスト内の単語と比較した画像内のピクセルの高解像度など、2つのドメイン間の違いから生じます。これらの違いに対処するために、シフトされたウィンドウで表現が計算される階層型トランスフォーマーを提案します。シフトされたウィンドウ方式は、自己注意の計算を重複しないローカルウィンドウに制限すると同時に、ウィンドウ間の接続を可能にすることで、効率を高めます。この階層アーキテクチャには、さまざまなスケールでモデル化する柔軟性があり、画像サイズに関して線形計算の複雑さがあります。 Swin Transformerのこれらの品質により、画像分類(ImageNet-1Kで87.3トップ1の精度)やオブジェクト検出(COCOテストで58.7ボックスAPおよび51.1マスクAP)などの高密度予測タスクを含む幅広いビジョンタスクと互換性があります。 dev)およびセマンティックセグメンテーション(ADE20Kvalで53.5mIoU)。そのパフォーマンスは、COCOでは+2.7ボックスAPと+2.6マスクAP、ADE20Kでは+3.2 mIoUの大幅なマージンで以前の最先端技術を上回り、ビジョンバックボーンとしてのTransformerベースのモデルの可能性を示しています。階層設計とシフトウィンドウアプローチも、すべてのMLPアーキテクチャに有益であることが証明されています。コードとモデルは、〜https://github.com/microsoft/Swin-Transformerで公開されています。
This paper presents a new vision Transformer, called Swin Transformer, that capably serves as a general-purpose backbone for computer vision. Challenges in adapting Transformer from language to vision arise from differences between the two domains, such as large variations in the scale of visual entities and the high resolution of pixels in images compared to words in text. To address these differences, we propose a hierarchical Transformer whose representation is computed with Shifted windows. The shifted windowing scheme brings greater efficiency by limiting self-attention computation to non-overlapping local windows while also allowing for cross-window connection. This hierarchical architecture has the flexibility to model at various scales and has linear computational complexity with respect to image size. These qualities of Swin Transformer make it compatible with a broad range of vision tasks, including image classification (87.3 top-1 accuracy on ImageNet-1K) and dense prediction tasks such as object detection (58.7 box AP and 51.1 mask AP on COCO test-dev) and semantic segmentation (53.5 mIoU on ADE20K val). Its performance surpasses the previous state-of-the-art by a large margin of +2.7 box AP and +2.6 mask AP on COCO, and +3.2 mIoU on ADE20K, demonstrating the potential of Transformer-based models as vision backbones. The hierarchical design and the shifted window approach also prove beneficial for all-MLP architectures. The code and models are publicly available at~https://github.com/microsoft/Swin-Transformer.