Machine learningDeep learningDeep Learning, Self-Supervised LearningAlgorithm

Masked Autoencoders

Also known as: MAE, Vision MAE

OriginatorKaiming HeYear2021Sources1Related methods15

Masked Autoencoders (MAE) is a self-supervised learning approach introduced by He et al. in 2021 that masks random patches of an image and trains a model to reconstruct the missing content. Adapting the masked language modeling paradigm from NLP to vision, MAE learns rich visual representations by solving a challenging reconstruction task without requiring labels.

Key highlights

  • Enables learning from unlabeled image data at scale, making it practical for domains lacking annotations
  • Learns transferable representations that improve performance on downstream tasks when fine-tuned with limited labels
  • Asymmetric encoder-decoder design makes pre-training efficient, requiring less computation than symmetric architectures
  • Achieves competitive performance with supervised pre-training on ImageNet while using no labels

Intuition

This section is available to Pro members. Upgrade to Pro

How it works

This section is available to Pro members. Upgrade to Pro

When to use it

Masked Autoencoders are ideal for self-supervised pre-training when labeled data is scarce or expensive to obtain. They work exceptionally well for learning visual representations that transfer to downstream tasks like classification, detection, and segmentation. MAE is preferred over supervised pre-training when privacy concerns limit data labeling. Use supervised methods when large labeled datasets are available, as they typically achieve higher downstream task performance with less pre-training.

Strengths & limitations

Strengths
  • Enables learning from unlabeled image data at scale, making it practical for domains lacking annotations
  • Learns transferable representations that improve performance on downstream tasks when fine-tuned with limited labels
  • Asymmetric encoder-decoder design makes pre-training efficient, requiring less computation than symmetric architectures
  • Achieves competitive performance with supervised pre-training on ImageNet while using no labels
Limitations
  • Reconstruction loss may not always align with downstream task objectives, requiring careful fine-tuning
  • Mask ratio and masking strategy are critical hyperparameters that require tuning for different domains
  • Pre-trained models may not transfer well to tasks with drastically different distributions from pre-training data

Common pitfalls

This section is available to Pro members. Upgrade to Pro

Applications

This section is available to Pro members. Upgrade to Pro

Frequently asked

How does masking in vision differ from masking in NLP?

In NLP, tokens represent discrete words. In vision, patches represent spatial regions containing diverse content. Vision masking typically uses higher mask ratios (60-75%) because images contain local structure; masking 75% patches still leaves sufficient visible context. NLP typically masks 15% of tokens. Images also require reconstruction of continuous pixel values, not discrete token prediction.

Why is the encoder-decoder asymmetric?

The encoder only processes visible patches, reducing computation. The decoder, which runs once per training sample, can be heavier without significantly affecting training efficiency. After pre-training, the decoder is discarded and only the encoder is used. This design achieves better efficiency-accuracy tradeoffs than symmetric architectures.

What is the right mask ratio?

For natural images, mask ratios of 60-75% work best. Higher masks (80%+) provide strong learning signals but may be impossible for the model. Lower masks (30-50%) are less challenging and learn less discriminative features. The optimal value depends on the image domain; medical images with sparse structures may need lower ratios.

Sources

  1. 1.
    He, K., Chen, X., Xie, S., Li, Y., Dollár, P., & Girshick, R. (2022). Masked autoencoders are scalable vision learners. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (pp. 16000-16009).

You have read it. What now?

Cite this page

ScholarGate. (2026, June 3). Masked Autoencoders. ScholarGate. https://scholargate.app/deep-learning/masked-autoencoders

Masked Autoencoders | ScholarGate