Swin Transformer: Hierarchical Vision Transformer using Shifted Windows

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 \textbf{S}hifted \textbf{win}dows. 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~\url{https://github.com/microsoft/Swin-Transformer}.

PDF Abstract ICCV 2021 PDF ICCV 2021 Abstract
Task Dataset Model Metric Name Metric Value Global Rank Uses Extra
Training Data
Result Benchmark
Semantic Segmentation ADE20K Swin-L (UperNet, ImageNet-22k pretrain) Validation mIoU 53.50 # 74
Test Score 62.8 # 1
Semantic Segmentation ADE20K Swin-B (UperNet, ImageNet-1k pretrain) Validation mIoU 49.7 # 119
Semantic Segmentation ADE20K val Swin-B (UperNet, ImageNet-1k pretrain) mIoU 49.7 # 50
Semantic Segmentation ADE20K val Swin-L (UperNet, ImageNet-22k pretrain) mIoU 53.5 # 36
Object Detection COCO minival Swin-L (HTC++, single scale) box AP 57.1 # 39
Instance Segmentation COCO minival Swin-L (HTC++, multi scale) mask AP 50.4 # 22
Object Detection COCO minival Swin-L (HTC++, multi scale) box AP 58 # 35
Instance Segmentation COCO minival Swin-L (HTC++, single scale) mask AP 49.5 # 25
Instance Segmentation COCO test-dev Swin-L (HTC++, single scale) mask AP 50.2 # 20
Instance Segmentation COCO test-dev Swin-L (HTC++, multi scale) mask AP 51.1 # 18
Object Detection COCO test-dev Swin-L (HTC++, multi scale) box mAP 58.7 # 31
Object Detection COCO test-dev Swin-L (HTC++, single scale) box mAP 57.7 # 33
Semantic Segmentation FoodSeg103 Swin-Transformer (Swin-Small) mIoU 41.6 # 4
Image Classification ImageNet Swin-L Top 1 Accuracy 87.3% # 99
Number of params 197M # 892
GFLOPs 103.9 # 450
Image Classification ImageNet Swin-B Top 1 Accuracy 86.4% # 143
Number of params 88M # 827
GFLOPs 47 # 417
Image Classification ImageNet Swin-T Top 1 Accuracy 81.3% # 590
Number of params 29M # 636
GFLOPs 4.5 # 211
Thermal Image Segmentation MFN Dataset SwinT mIOU 49.0 # 33
Instance Segmentation Occluded COCO Swin-B + Cascade Mask R-CNN Mean Recall 62.90 # 2
Instance Segmentation Occluded COCO Swin-T + Mask R-CNN Mean Recall 58.81 # 6
Instance Segmentation Occluded COCO Swin-S + Mask R-CNN Mean Recall 61.14 # 5
Image Classification OmniBenchmark SwinTransformer Average Top-1 Accuracy 46.4 # 2
Instance Segmentation Separated COCO Swin-S + Mask R-CNN Mean Recall 33.67 # 5
Instance Segmentation Separated COCO Swin-B + Cascade Mask R-CNN Mean Recall 36.31 # 2
Instance Segmentation Separated COCO Swin-T + Mask R-CNN Mean Recall 31.94 # 6

Methods