Source-linked AI summary

A Comprehensive Survey of Transformers for Computer Vision

Sonain Jamil, Md. Jalil Piran, Oh-Jin Kwon

arXiv:2211.06004v1cs.CV

TL;DR

Tracking relationships among image tokens can be time- and memory-intensive. This survey classifies ViT applications across computer vision, reviews models and their trade-offs, and reports broad use across CV tasks while identifying implementation challenges.

  • Problem

    Calculating every relationship among image tokens in a normal image would be time-consuming and memory-intensive.

    Source: §I · p. 1
  • Method

    The survey classifies ViT applications across seven CV tasks, reviews available models and their pros and cons, and presents lessons and research directions.

    Source: §I · p. 3
  • Results

    The survey reports ViT use across classification, detection, segmentation, compression, super-resolution, denoising, and anomaly detection.

    Source: §V · p. 8
  • Takeaways & Limitations

    The survey identifies open research challenges and future directions for implementing ViTs.

    Source: §V · p. 8

    ViT performs better than CNN in the cited comparison, but its computational cost is much higher.

    Source: A. High computational cost · p. 8

Abstract

from arXiv · show

As a special type of transformer, Vision Transformers (ViTs) are used to various computer vision applications (CV), such as image recognition. There are several potential problems with convolutional neural networks (CNNs) that can be solved with ViTs. For image coding tasks like compression, super-resolution, segmentation, and denoising, different variants of the ViTs are used. The purpose of this survey is to present the first application of ViTs in CV. The survey is the first of its kind on ViTs for CVs to the best of our knowledge. In the first step, we classify different CV applications where ViTs are applicable. CV applications include image classification, object detection, image segmentation, image compression, image super-resolution, image denoising, and anomaly detection. Our next step is to review the state-of-the-art in each category and list the available models. Following that, we present a detailed analysis and comparison of each model and list its pros and cons. After that, we present our insights and lessons learned for each category. Moreover, we discuss several open research challenges and future research directions.

I. INTRODUCTION

Vision Transformers process images as patch tokens, allowing them to model global relationships, while their training can be more difficult than CNNs. This survey reviews ViTs across computer-vision applications, comparing models and identifying lessons and research directions.

I. INTRODUCTION

  • ViTs divide an image into patches, linearly embed them as tokens, encode them with a transformer, and use an MLP head to predict the image class.The encoder uses multi-head attention and normalization.Source: §I · p. 1
  • Earlier milestones include a pure transformer outperforming CNNs in image classification and a transformer backend added to ResNet to lower costs while enhancing accuracy.The cited work places these developments in 2021.Source: §I · p. 1
  • CNN inductive biases can help models learn image characteristics with less data, whereas ViTs capture broader global relationships but are harder to train.The survey also notes that CNN biases can make global connections harder to recognize.Source: §I · pp. 1–2
  • The survey describes approaches intended to train ViTs without massive vision datasets or precise labels, a motivation for their appeal.It frames near-autonomous learning as a possible way to infer problem characteristics without large datasets or precise labeling.Source: §I · p. 2
  • The survey classifies ViT use across seven CV tasks, reviews available models and their pros and cons, and presents category-specific lessons and future research directions.The tasks are image classification, object detection, segmentation, compression, super-resolution, denoising, and anomaly detection.Source: §I · p. 3
  • The survey situates its chapters from related work to CV applications, then open research challenges and future directions, and finally the conclusion.The organization is shown in Figure 3.Source: §I · p. 3

II. RELATED WORK

Prior ViT surveys covered selected theoretical foundations and vision tasks, but the cited work leaves gaps across image-processing applications. The survey motivates a broader synthesis of classification, detection, segmentation, compression, denoising, and super-resolution.

II. RELATED WORK

  • Survey [23] covers transformer theory, foundations, and memory efficiency, but omits computer-vision tasks; survey covers several vision tasks but not denoising or compression.The vision tasks in include classification, segmentation, super-resolution, and object detection.Source: §II · p. 3
  • Survey [26] describes transformer architectures for image segmentation, classification, and object detection, but omits super-resolution, denoising, and compression.Source: §II · p. 3
  • Survey covers low- and high-level vision, including colorization, super-resolution, generation, segmentation, and detection, but not image compression or classification.The passage also notes that [25] surveys NLP architectures without covering transformer applications to CV.Source: §II · p. 3
  • The survey identifies a need for insight across classification, detection, segmentation, compression, denoising, and super-resolution based on gaps in existing survey coverage.The passage cites Table II as the basis for this analysis.Source: §II · p. 3

III. APPLICATIONS OF VITS IN CV

The survey introduces modified classical ViTs for several computer-vision and image-processing applications. The supplied section material also identifies a table of acronyms.

III. APPLICATIONS OF VITS IN CV

  • Modified classical ViTs are used for object detection, segmentation, compression, super-resolution, denoising, and anomaly detection.Source: §III · p. 3
  • Table I is titled as a list of acronyms.Source: Table I · p. 4

A. ViTs for Image Classification

Classical ViTs encode image patches with a transformer encoder and MLP head, while the surveyed classification studies apply ViT variants across diverse datasets and domains. Reported results include high accuracies for remote sensing, medical, agricultural, and other specialized classification tasks.

A. ViTs for Image Classification

  • Image patches pass through a transformer encoder and MLP head to predict the class in the classical ViT pipeline.The encoder applies MLP, normalization, and multi-head attention to create embedded patches.Source: A. ViTs for Image Classification · p. 4
  • CrossViT variants report 77.1%, 82.3%, and 82.8% accuracy on ImageNet1K, and CrossViT-18 reports 99.11% accuracy on CIFAR10.CrossViT-15 and CrossViT-18 report 99.0% and 99.11% accuracy on CIFAR10, respectively; CrossViT-18 reports 91.36% accuracy on CIFAR100.Source: A. ViTs for Image Classification · p. 4
  • For remote-sensing classification, CTNet achieved 97.70% accuracy on AID and 95.49% on NWPU-RESISC45, while ES-GSNet reached 99.29% on UCM.DHViT reported 99.58%, 99.55%, and 96.40% accuracy on Trento, Houston 2013, and Houston 2018, respectively; ForestViT reported 94.28% on PUAS.Source: A. ViTs for Image Classification · p. 4
  • MIL-ViT achieved 97.9% accuracy on APTOS2019 and 95.9% on RFMiD2020 for fundus image classification.Source: A. ViTs for Image Classification · p. 4
  • LeViT reported 91.56% accuracy on Chinese asphalt pavement and 99.17% on German asphalt pavement, while a ViT achieved 98.3% on a malicious-drone dataset.Source: A. ViTs for Image Classification · pp. 4–5
  • SeedViT outperformed CNNs with 96.70% accuracy on a custom maize-seed dataset, while DOViT reported 90.32% and 92.78% accuracy on GAOs-1 and GAOs-2.For femur-fracture classification on real X-rays, the surveyed model achieved 83% accuracy, 77% precision, 76% recall, and 77% F1-score.Source: A. ViTs for Image Classification · p. 5

B. ViTs for Object Detection

This section surveys vanilla ViT approaches to object detection and examples spanning satellite-image manipulation and 3D detection.

C. ViTs for Image Segmentation

The section reviews ViT-based medical and referring-image segmentation models, reporting results across challenge datasets and metrics.

D. ViTs for Image Compression

The section describes Entroformer, which incorporates a transformer into a learned image-compression entropy module.

E. ViTs for Image Super Resolution

The section presents a spatio-temporal ViT for microscopic-image and video super-resolution, with results reported by video motion level.

F. ViTs for Image Denoising

The survey describes transformer-based approaches for image denoising and anomaly detection, then identifies broader challenges facing ViT implementation.

F. ViTs for Image Denoising

  • TED-Net uses transformers in both encoder and decoder for low-dose CT denoising, reporting SSIM 0.9144 and RMSE 8.7681 on the AAPM-Mayo clinic LDCT Grand Challenge dataset.Source: F. ViTs for Image Denoising · p. 7

G. ViTs for Anamoly Detection

  • VT-ADL reports mean PRO scores of 0.984 on MNIST, 0.807 on MVTec, and 0.89 on BTAD.Source: G. ViTs for Anamoly Detection · p. 7
  • AnoViT reports mean AUROC values of 92.4, 60.1, and 78 on MNIST, CIFAR, and MVTecAD, while TransAnomaly reports AUC values with and without sliding windows on Pred1, Pred2, and Avenue.TransAnomaly's AUC values without sliding windows are 84.0%, 96.10%, and 85.80%; with sliding windows they are 86.70%, 96.40%, and 87.00%, respectively.Source: G. ViTs for Anamoly Detection · p. 7

IV. OPEN RESEARCH CHALLENGES AND FUTURE DIRECTIONS

  • The survey identifies computational cost, large training datasets, neural architecture search, interpretability, and hardware efficiency as implementation challenges for ViTs.Source: §IV · p. 7

A. High computational cost

The survey highlights computational cost and training-data demands as challenges for ViTs, with performance relative to ResNet differing by training dataset size.

A. High computational cost

  • ViT-based models have millions of parameters and require high-computational-power computers for training.Source: A. High computational cost · p. 7
  • ViTs perform better than CNNs in the comparison described, but their computational cost is much higher, making cost reduction a major research challenge.Source: A. High computational cost · p. 8

B. Large training dataset

C. Neural architecture search (NAS)

The survey identifies underexplored architecture search, interpretability, and hardware efficiency as challenges and research directions for ViTs.

C. Neural architecture search (NAS)

D. Interpret-ability of the transformers

E. Hardware efficient desings

V. CONCLUSION

  • The survey presents future directions intended to give young researchers a new perspective on ViT research challenges.Source: §V · p. 8
Loading 2211.06004v1…