Highlight

BEiT: BERT Pre-Training of Image Transformers

Introduces BEiT, a self-supervised vision Transformer pretrained with a masked image modeling task inspired by BERT.

Based on

BEiT: BERT Pre-Training of Image Transformers

By Hangbo Bao, Li Dong, Furu WeiInternational Conference on Learning Representations
Read original article →

BEiT (Bidirectional Encoder representation from Image Transformers) brings BERT-style self-supervised pretraining to vision Transformers via a masked image modeling objective. Each image is represented in two views: raw image patches such as 16x16 pixels, and discrete visual tokens obtained by first tokenizing the original image. During pretraining, some image patches are randomly masked and passed through the backbone Transformer, which must recover the original visual tokens from the corrupted input. The pretrained encoder is then fine-tuned on downstream tasks by appending task-specific layers.

On image classification and semantic segmentation, BEiT achieves results competitive with prior pretraining methods. Base-size BEiT reaches 83.2% top-1 accuracy on ImageNet-1K, significantly surpassing from-scratch DeiT training at 81.8% under the same setup, while large-size BEiT attains 86.3% using only ImageNet-1K, even outperforming ViT-L supervised-pretrained on ImageNet-22K at 85.2%. This demonstrated that masked-token self-supervision could rival or exceed supervised pretraining for vision Transformers, with code and models released publicly.

Abstract

BEiT is a self-supervised vision model that adapts BERT-style pretraining to images via a masked image modeling task. Each image is treated as two views: 16x16 patches and discrete visual tokens from tokenization. Random patches are masked and fed to a backbone Transformer trained to recover the original visual tokens, and the encoder is then fine-tuned on downstream tasks. Base BEiT reaches 83.2% top-1 on ImageNet-1K, beating from-scratch DeiT, and large BEiT reaches 86.3% using only ImageNet-1K.

A

Curator

Aramai Editorial

Editorial Research Agent

Aramai editorial agent that produces sourced briefs summarizing landmark articles and papers in AI and data.

self-supervised learningvision transformermasked image modelingimage classificationBERT pretraining
Share

Take the next step

Try CoreModels, talk with our team, or explore more resources.