SwinTransformer

December 7, 2021 ยท View on GitHub


Catalogue

1. Overview

Swin Transformer a new vision Transformer, that capably serves as a general-purpose backbone for computer vision. It is 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. Paperใ€‚

2. Accuracy, FLOPs and Parameters

ModelsTop1Top5Reference
top1
Reference
top5
FLOPs
(G)
Params
(M)
SwinTransformer_tiny_patch4_window7_2240.80690.95340.8120.9554.528
SwinTransformer_small_patch4_window7_2240.82750.96130.8320.9628.750
SwinTransformer_base_patch4_window7_2240.83000.96260.8350.96515.488
SwinTransformer_base_patch4_window12_3840.84390.96930.8450.97047.188
SwinTransformer_base_patch4_window7_224[1]0.84870.97460.8520.97515.488
SwinTransformer_base_patch4_window12_384[1]0.86420.98070.8640.98047.188
SwinTransformer_large_patch4_window7_224[1]0.85960.97830.8630.97934.5197
SwinTransformer_large_patch4_window12_384[1]0.87190.98230.8730.982103.9197

[1]: Based on imagenet22k dataset pre-training, and then in imagenet1k dataset transfer learning.

Note: The difference of precision with reference from the difference of data preprocessing.