Going deeper with Image Transformers

Code

• github.com/rwightman/pytorch-image-models
  OfficialIn paperpytorch★ 36,538
• github.com/facebookresearch/deit
  OfficialIn paperpytorch★ 4,327
• github.com/martinsbruveris/tensorflow-image-models
  tf★ 291
• github.com/DarshanDeshpande/jax-models
  jax★ 161
• github.com/ttt496/vit-pytorch
  pytorch★ 12
• github.com/sayakpaul/cait-tf
  tf★ 12
• github.com/IMvision12/keras-vision-models
  pytorch★ 8
• github.com/jaketae/res-mlp
  pytorch★ 3
• github.com/holdfire/CLS
  pytorch★ 3
• github.com/conceptofmind/CaiT-Flax
  jax★ 3

Abstract

Transformers have been recently adapted for large scale image classification, achieving high scores shaking up the long supremacy of convolutional neural networks. However the optimization of image transformers has been little studied so far. In this work, we build and optimize deeper transformer networks for image classification. In particular, we investigate the interplay of architecture and optimization of such dedicated transformers. We make two transformers architecture changes that significantly improve the accuracy of deep transformers. This leads us to produce models whose performance does not saturate early with more depth, for instance we obtain 86.5% top-1 accuracy on Imagenet when training with no external data, we thus attain the current SOTA with less FLOPs and parameters. Moreover, our best model establishes the new state of the art on Imagenet with Reassessed labels and Imagenet-V2 / match frequency, in the setting with no additional training data. We share our code and models.

Tasks

Benchmark Results

Reproductions