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

Code

• github.com/IBM/CrossViT
  OfficialIn paperpytorch★ 414
• github.com/rwightman/pytorch-image-models
  pytorch★ 36,538
• github.com/rishikksh20/CrossViT-pytorch
  pytorch★ 208
• github.com/SforAiDl/vformer
  pytorch★ 164
• github.com/kit-mrt/masked-fusion-360
  pytorch★ 18
• github.com/ttt496/vit-pytorch
  pytorch★ 12
• github.com/kit-mrt/road-barlow-twins
  pytorch★ 11
• github.com/kit-mrt/red-motion
  pytorch★ 7
• github.com/ahmedelmahy/myownvit
  pytorch★ 2
• github.com/2023-MindSpore-1/ms-code-145
  mindspore★ 0

Abstract

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 https://github.com/IBM/CrossViT.

Tasks

Benchmark Results

Reproductions