ColorMAE: Exploring data-independent masking strategies in Masked AutoEncoders
Not on the map. Why?
Abstract (arXiv)
Masked AutoEncoders (MAE) have emerged as a robust self-supervised framework, offering remarkable performance across a wide range of downstream tasks. To increase the difficulty of the pretext task and learn richer visual representations, existing works have focused on replacing standard random masking with more sophisticated strategies, such as adversarial-guided and teacher-guided masking. However, these strategies depend on the input data thus commonly increasing the model complexity and requiring additional calculations to generate the mask patterns. This raises the question: Can we enhance MAE performance beyond random masking without relying on input data or incurring additional computational costs? In this work, we introduce a simple yet effective data-independent method, termed ColorMAE, which generates different binary mask patterns by filtering random noise. Drawing inspiration from color noise in image processing, we explore four types of filters to yield mask patterns with different spatial and semantic priors. ColorMAE requires no additional learnable parameters or computational overhead in the network, yet it significantly enhances the learned representations. We provide a comprehensive empirical evaluation, demonstrating our strategy's superiority in downstream tasks compared to random masking. Notably, we report an improvement of 2.72 in mIoU in semantic segmentation tasks relative to baseline MAE implementations.
Key facts. Cited by 8 papers in the map (data through 7 Oct 2026). Builds on 39 of its 52 references that are in the map. First posted on arXiv 17 Jul 2024.
Builds on
39 of 52 references.
References · 39
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2014Microsoft COCO: Common Objects in Contextcited by 17,498
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2020Exploring Simple Siamese Representation Learningcited by 2,853
27 more
- 2021BEiT: BERT Pre-Training of Image Transformerscited by 2,360
- 2018Unsupervised Feature Learning via Non-Parametric Instance-level Discriminationcited by 1,838
- 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,610
- 2018Unified Perceptual Parsing for Scene Understandingcited by 1,265
- 2018Learning deep representations by mutual information estimation and maximizationcited by 1,265
- 2021SimMIM: A Simple Framework for Masked Image Modelingcited by 1,147
- 2021iBOT: Image BERT Pre-Training with Online Tokenizercited by 891
- 2019Learning Representations by Maximizing Mutual Information Across Viewscited by 733
- 2022Exploring Plain Vision Transformer Backbones for Object Detectioncited by 723
- 2022data2vec: A General Framework for Self-supervised Learning in Speech, Vision and Languagecited by 718
- 2021Masked Feature Prediction for Self-Supervised Visual Pre-Trainingcited by 573
- 2022Context Autoencoder for Self-Supervised Representation Learningcited by 297
- 2020Whitening for Self-Supervised Representation Learningcited by 278
- 2020Eigen-CAM: Class Activation Map using Principal Componentscited by 188
- 2020GridMask Data Augmentationcited by 123
- 2021Are Large-scale Datasets Necessary for Self-Supervised Pre-training?cited by 117
- 2022What to Hide from Your Students: Attention-Guided Masked Image Modelingcited by 117
- 2021MST: Masked Self-Supervised Transformer for Visual Representationcited by 114
- 2022SemMAE: Semantic-Guided Masking for Learning Masked Autoencoderscited by 113
- 2022How Mask Matters: Towards Theoretical Understandings of Masked Autoencoderscited by 86
- 2022Adversarial Masking for Self-Supervised Learningcited by 78
- 2022Masked Frequency Modeling for Self-Supervised Visual Pre-Trainingcited by 70
- 2022Bootstrapped Masked Autoencoders for Vision BERT Pretrainingcited by 60
- 2022Uniform Masking: Enabling MAE Pre-training for Pyramid-based Vision Transformers with Localitycited by 57
- 2023Hard Patches Mining for Masked Image Modelingcited by 56
- 2023Mixed Autoencoder for Self-supervised Visual Representation Learningcited by 38
- 2023DropPos: Pre-Training Vision Transformers by Reconstructing Dropped Positionscited by 19
Built on by
Citing papers per year: 2024: 1; 2025: 4; 2026: 3.
8 papers.
- 2024Masked Image Modeling: A Surveycited by 28
- 2026Self-Distillation of Hidden Layers for Self-Supervised Representation Learningcited by 3
- 2025VasoMIM: Vascular Anatomy-Aware Masked Image Modeling for Vessel Segmentationcited by 2
- 2025Structured-Noise Masked Modeling for Video, Audio and Beyondcited by 1
- 2025Conjuring Positive Pairs for Efficient Unification of Representation Learning and Image Synthesiscited by 0
- 2025CoMA: Complementary Masking and Hierarchical Dynamic Multi-Window Self-Attention in a Unified Pre-training Frameworkcited by 0
- 2026JEPA Learns What the Mask Leaves Unrecoverablecited by 0
- 2026Masked Swingers: Harnessing Data Augmentation to Advance Autoencoders for Self-Supervised Learningcited by 0
- 2026Masked Swingers: Harnessing Data Augmentation to Advance Autoencoders for Self-Supervised Learning29 Sep 2026
- 2026JEPA Learns What the Mask Leaves Unrecoverable26 Sep 2026
- 2026Self-Distillation of Hidden Layers for Self-Supervised Representation Learning16 Mar 2026
- 2025CoMA: Complementary Masking and Hierarchical Dynamic Multi-Window Self-Attention in a Unified Pre-training Framework8 Nov 2025
- 2025VasoMIM: Vascular Anatomy-Aware Masked Image Modeling for Vessel Segmentation14 Aug 2025
- 2025Structured-Noise Masked Modeling for Video, Audio and Beyond20 Mar 2025
- 2025Conjuring Positive Pairs for Efficient Unification of Representation Learning and Image Synthesis19 Mar 2025
- 2024Masked Image Modeling: A Survey13 Aug 2024