CrossViT: Cross-Attention Multi-Scale Vision Transformer for Image Classification

ICCV 2021  ·  Chun-Fu Chen, Quanfu Fan, Rameswar Panda ·

The recently developed vision transformer (ViT) has achieved promising results on image classification compared to convolutional neural networks. Inspired by this, in this paper, we study how to learn multi-scale feature representations in transformer models for image classification... To this end, we propose a dual-branch transformer to combine image patches (i.e., tokens in a transformer) of different sizes to produce stronger image features. Our approach processes small-patch and large-patch tokens with two separate branches of different computational complexity and these tokens are then fused purely by attention multiple times to complement each other. Furthermore, to reduce computation, we develop a simple yet effective token fusion module based on cross attention, which uses a single token for each branch as a query to exchange information with other branches. Our proposed cross-attention only requires linear time for both computational and memory complexity instead of quadratic time otherwise. Extensive experiments demonstrate that our approach performs better than or on par with several concurrent works on vision transformer, in addition to efficient CNN models. For example, on the ImageNet1K dataset, with some architectural changes, our approach outperforms the recent DeiT by a large margin of 2\% with a small to moderate increase in FLOPs and model parameters. Our source codes and models are available at \url{https://github.com/IBM/CrossViT}. read more

PDF Abstract ICCV 2021 PDF ICCV 2021 Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Image Classification ImageNet CrossViT-18 Top 1 Accuracy 82.5% # 198
Number of params 43.3M # 153
Image Classification ImageNet CrossViT-15 Top 1 Accuracy 81.5% # 250
Number of params 27.4M # 189
Image Classification ImageNet CrossViT-15+ Top 1 Accuracy 82.3% # 211
Number of params 28.2M # 186
Image Classification ImageNet CrossViT-18+ Top 1 Accuracy 82.8% # 183
Number of params 44.3M # 152
Hardware Burden None # 1
Operations per network pass None # 1

Methods