Multi-scale features are essential for various vision tasks, such as classification, detection, and segmentation. Although Vision Transformers (ViTs) show remarkable success in capturing global features within an image, how to leverage multi-scale features in Transformers is not well explored. This paper proposes a scale-aware vision Transformer called ScopeViT that efficiently captures multi-granularity representations. Two novel attention with lightweight computation are introduced: Multi-Scale Self-Attention (MSSA) and Global-Scale Dilated Attention (GSDA). MSSA embeds visual tokens with different receptive fields into distinct attention heads, allowing the model to perceive various scales across the network. GSDA enhances model understanding of the global context through token-dilation operation, which reduces the number of tokens involved in attention computations. This dual attention method enables ScopeViT to "see" various scales throughout the entire network and effectively learn inter-object relationships, reducing heavy quadratic computational complexity. Extensive experiments demonstrate that ScopeViT achieves competitive complexity/accuracy trade-offs compared to existing networks across a wide range of visual tasks. On the ImageNet-1K dataset, ScopeViT achieves a top-1 accuracy of 81.1%, using only 7.4 million parameters and 2.0 GFLOPs. Our approach outperforms Swin (ViT-based) by 1.9% accuracy while saving 42% of the parameters, outperforms MobileViTv2 (Hybrid-based) with a 0.7% accuracy gain while using 50% of the computations, and also beats ConvNeXtv2 (ConvNet-based) by 0.8% with fewer parameters.
| Model | Params (M) | GFLOPs | Resolution | Top-1 (%) | Reference |
|---|---|---|---|---|---|
| Swin-2G | 12.8 | 2.0 | 224 | 79.2 | ICCV'2021 |
| DeiT-S | 22.1 | 4.6 | 224 | 79.8 | ICML'2021 |
| PVT-Small | 24.5 | 3.8 | 224 | 79.8 | ICCV'2021 |
| P2T-T | 11.6 | 1.8 | 224 | 79.8 | TPAMI'2022 |
| Shunted-T | 11.5 | 2.1 | 224 | 79.8 | CVPR'2022 |
| EViT-S(Keep rate=0.6) | 22.1 | 3.5 | 256 | 80.0 | ICLR'2022 |
| ConvNeXt V2-P | 9.1 | 1.4 | 224 | 80.3 | arXiv'2023 |
| MobileViTv2-1.5 | 10.6 | 4.0 | 256 | 80.4 | arXiv'2022 |
| CoaT Mini | 10.0 | 6.8 | 224 | 80.8 | ICCV'2021 |
| MPViT-XS | 10.5 | 2.9 | 224 | 80.9 | CVPR'2022 |
| CrossViT-S | 26.7 | 5.6 | 224 | 81.0 | ICCV'2021 |
| ScopeViT-T | 7.4 | 2.0 | 256 | 81.1 |
This project is released under the Apache 2.0 license.
Our implementation is mainly based on the following codebases. We gratefully thank the authors for their wonderful works.
- pytorch-image-models (timm): PyTorch image models, scripts, pretrained weights.
- PoolFormer: Official PyTorch implementation of MetaFormer.
- ConvNeXt: Official PyTorch implementation of ConvNeXt.
- OpenMixup: Open-source toolbox for visual representation learning.
- MMDetection: OpenMMLab Detection Toolbox and Benchmark.
- MMSegmentation: OpenMMLab Semantic Segmentation Toolbox and Benchmark.
- MMPose: OpenMMLab Pose Estimation Toolbox and Benchmark.
- MMHuman3D: OpenMMLab 3D Human Parametric Model Toolbox and Benchmark.