Vision Transformer

A Vision Transformer applies Transformer blocks to image or grid patches. It replaces convolutional locality with tokenization plus attention over patches.

Patch embedding is:

where is image patch , is the patch projection, and is a positional embedding.

For an image and patch size , the number of tokens is:

Self-attention then scales as:

so smaller patches preserve more detail but increase attention cost. A class token or pooled patch representation is commonly used for image-level prediction:

Inductive Bias

Compared with CNNs, ViTs encode less hard locality and more global token mixing. This can be useful with large pretraining, but weak locality can be data-hungry. Hybrid stems, window attention, hierarchical patch merging, or strong augmentation often reintroduce local bias.

Why It Matters

  • Shows that Transformers can process non-text token sequences.
  • Useful as a reference for patch, voxel, contact-map, or grid tokenization.
  • Often requires more data or stronger pretraining than CNNs when locality is not hard-coded.

Architecture Decomposition

A ViT block can be read as one instance of a broader MetaFormer-style pattern:

For ViT, the token mixer is self-attention. Other vision backbones replace that slot with pooling, convolution, window attention, or token-axis MLPs. The canonical paper note for this scaffold view is MetaFormer.

BackboneToken Mixer
ViTglobal self-attention
Swin Transformershifted-window attention
MLP-Mixertoken-axis MLP
PoolFormerpooling
ConvNeXtdepthwise convolution

Checks

  • What is the patch size, and what spatial detail is lost?
  • Are positional encodings absolute, relative, or learned?
  • Is locality recovered through hybrid CNN stems, window attention, or data scale?
  • Is the paper changing attention itself, or changing the broader block scaffold and training recipe?