An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale

AI-generated keywords: Vision Transformer Convolutional Networks Self-Attention Image Recognition Alexey Dosovitskiy

AI-generated Key Points

The license of the paper does not allow us to build upon its content and the key points are generated using the paper metadata rather than the full article.

  • The Transformer architecture is widely used for natural language processing tasks and gaining popularity in computer vision.
  • A recent study by Alexey Dosovitskiy and his team shows that a pure transformer applied directly to sequences of image patches can perform very well on image classification tasks without the need for CNNs.
  • This approach is called Vision Transformer (ViT) and involves pre-training on large amounts of data and transferring it to multiple mid-sized or small image recognition benchmarks such as ImageNet, CIFAR-100, and VTAB.
  • ViT attains excellent results compared to state-of-the-art convolutional networks while requiring substantially fewer computational resources to train.
  • ViT's success is due to its ability to capture long-range dependencies between image patches through self-attention mechanisms, allowing the model to learn meaningful representations of images without being constrained by the local receptive fields of convolutions.
  • This study suggests that transformers have great potential for computer vision tasks and could potentially replace convolutional networks as the go-to architecture for image recognition at scale.
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Authors: Alexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn, Xiaohua Zhai, Thomas Unterthiner, Mostafa Dehghani, Matthias Minderer, Georg Heigold, Sylvain Gelly, Jakob Uszkoreit, Neil Houlsby

Fine-tuning code and pre-trained models are available at https://github.com/google-research/vision_transformer. ICLR camera-ready version with 2 small modifications: 1) Added a discussion of CLS vs GAP classifier in the appendix, 2) Fixed an error in exaFLOPs computation in Figure 5 and Table 6 (relative performance of models is basically not affected)

Abstract: While the Transformer architecture has become the de-facto standard for natural language processing tasks, its applications to computer vision remain limited. In vision, attention is either applied in conjunction with convolutional networks, or used to replace certain components of convolutional networks while keeping their overall structure in place. We show that this reliance on CNNs is not necessary and a pure transformer applied directly to sequences of image patches can perform very well on image classification tasks. When pre-trained on large amounts of data and transferred to multiple mid-sized or small image recognition benchmarks (ImageNet, CIFAR-100, VTAB, etc.), Vision Transformer (ViT) attains excellent results compared to state-of-the-art convolutional networks while requiring substantially fewer computational resources to train.

Submitted to arXiv on 22 Oct. 2020

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

The license of the paper does not allow us to build upon its content and the AI assistant only knows about the paper metadata rather than the full article.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2010.11929v2

This paper's license doesn't allow us to build upon its content and the summarizing process is here made with the paper's metadata rather than the article.

The Transformer architecture has become the standard for natural language processing tasks and its application to computer vision is gaining traction. A recent study by Alexey Dosovitskiy and his team shows that reliance on convolutional networks (CNNs) is not necessary. They demonstrate that a pure transformer applied directly to sequences of image patches can perform very well on image classification tasks without the need for CNNs. This approach, called Vision Transformer (ViT), involves pre-training on large amounts of data and transferring it to multiple mid-sized or small image recognition benchmarks such as ImageNet, CIFAR-100, and VTAB. ViT attains excellent results compared to state-of-the-art convolutional networks while requiring substantially fewer computational resources to train. The authors argue that ViT's success is due to its ability to capture long-range dependencies between image patches through self-attention mechanisms. This allows the model to learn meaningful representations of images without being constrained by the local receptive fields of convolutions. Overall, this study suggests that transformers have great potential for computer vision tasks and could potentially replace convolutional networks as the go-to architecture for image recognition at scale.
Created on 11 May. 2023

Assess the quality of the AI-generated content by voting

Score: 0

Why do we need votes?

Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.

The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.

The license of this specific paper does not allow us to build upon its content and the summarizing tools will be run using the paper metadata rather than the full article. However, it still does a good job, and you can also try our tools on papers with more open licenses.

Similar papers summarized with our AI tools

Navigate through even more similar papers through a

tree representation

Look for similar papers (in beta version)

By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while our database grows.

Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.