本論文では、Swin Transformerと呼ばれる新しいビジョントランスフォーマーを紹介する。トランスフォーマーを言語から視覚に適応させるための課題は、視覚的なエンティティのスケールに大きなバリエーションがあることや、テキスト中の単語と比較して画像中のピクセルの解像度が高いことなど、2つのドメインの違いから生じる。これらの違いに対処するために、表現がシフトした窓で計算される階層的なトランスフォーマーを提案する。シフトされたウィンドウのスキームは、自己言及の計算をオーバーラップしないローカルウィンドウに限定する一方で、ウィンドウ間の接続を可能にすることで、より大きな効率性をもたらす。この階層型アーキテクチャは、様々なスケールのモデルに対応できる柔軟性を持ち、画像サイズに対して線形の計算量を持ちます。Swin Transformerのこれらの特性により、画像分類(ImageNet-1Kで86.4のトップ1精度)や、物体検出(COCO test-devで\ボックスAP 58.7、マスクAP 51.1)、セマンティックセグメンテーション(ADE20K valでmIoU 53.5)などの緻密な予測タスクを含む、幅広いビジョンタスクに対応する。その性能は、COCOではボックスAPで+2.7、マスクAPで+2.6、ADE20Kでは+3.2mIoUと、これまでの最先端を大きく上回り、トランスフォーマーベースのモデルはビジョンのバックボーンとしての可能性を示している。このコードとモデルは、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 (86.4 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 code and models will be made publicly available at~https://github.com/microsoft/Swin-Transformer.