Context Autoencoder for Self-Supervised Representation Learning
Not on the map. Why?
Abstract (arXiv)
We present a novel masked image modeling (MIM) approach, context autoencoder (CAE), for self-supervised representation pretraining. We pretrain an encoder by making predictions in the encoded representation space. The pretraining tasks include two tasks: masked representation prediction - predict the representations for the masked patches, and masked patch reconstruction - reconstruct the masked patches. The network is an encoder-regressor-decoder architecture: the encoder takes the visible patches as input; the regressor predicts the representations of the masked patches, which are expected to be aligned with the representations computed from the encoder, using the representations of visible patches and the positions of visible and masked patches; the decoder reconstructs the masked patches from the predicted encoded representations. The CAE design encourages the separation of learning the encoder (representation) from completing the pertaining tasks: masked representation prediction and masked patch reconstruction tasks, and making predictions in the encoded representation space empirically shows the benefit to representation learning. We demonstrate the effectiveness of our CAE through superior transfer performance in downstream tasks: semantic segmentation, object detection and instance segmentation, and classification. The code will be available at https://github.com/Atten4Vis/CAE.
Key facts. Cited by 297 papers in the map (data through 7 Oct 2026). Builds on 60 of its 114 references that are in the map. First posted on arXiv 7 Feb 2022.
Builds on
60 of 114 references.
References · 60
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 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
- 2017Mask R-CNNcited by 7,259
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
38 more
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
- 2021Zero-Shot Text-to-Image Generationcited by 4,276
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,229
- 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,875
- 2020Exploring Simple Siamese Representation Learningcited by 2,853
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,845
- 2021BEiT: BERT Pre-Training of Image Transformerscited by 2,360
- 2018Unsupervised Feature Learning via Non-Parametric Instance-level Discriminationcited by 1,838
- 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,818
- 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,610
- 2016Context Encoders: Feature Learning by Inpaintingcited by 1,424
- 2016Unsupervised Learning of Visual Representations by Solving Jigsaw Puzzlescited by 1,379
- 2019MMDetection: Open MMLab Detection Toolbox and Benchmarkcited by 1,379
- 2018Unified Perceptual Parsing for Scene Understandingcited by 1,265
- 2021Swin Transformer V2: Scaling Up Capacity and Resolutioncited by 1,215
- 2021SimMIM: A Simple Framework for Masked Image Modelingcited by 1,147
- 2016Colorful Image Colorizationcited by 1,113
- 2015Unsupervised Visual Representation Learning by Context Predictioncited by 1,062
- 2018Deep Clustering for Unsupervised Learning of Visual Featurescited by 1,020
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,013
- 2016Deep Networks with Stochastic Depthcited by 973
- 2021iBOT: Image BERT Pre-Training with Online Tokenizercited by 891
- 2020What Makes for Good Views for Contrastive Learning?cited by 843
- 2015Unsupervised Deep Embedding for Clustering Analysiscited by 763
- 2019Data-Efficient Image Recognition with Contrastive Predictive Codingcited by 672
- 2020Prototypical Contrastive Learning of Unsupervised Representationscited by 636
- 2021Masked Feature Prediction for Self-Supervised Visual Pre-Trainingcited by 573
- 2017Large Batch Training of Convolutional Networkscited by 529
- 2019Self-labelling via simultaneous clustering and representation learningcited by 525
- 2020Dense Contrastive Learning for Self-Supervised Visual Pre-Trainingcited by 512
- 2019Cascade R-CNN: High Quality Object Detection and Instance Segmentationcited by 428
- 2014Discriminative Unsupervised Feature Learning with Exemplar Convolutional Neural Networkscited by 299
- 2020Propagate Yourself: Exploring Pixel-Level Consistency for Unsupervised Visual Representation Learningcited by 297
- 2020Whitening for Self-Supervised Representation Learningcited by 278
- 2016Joint Unsupervised Learning of Deep Representations and Image Clusterscited by 205
- 2019Local Aggregation for Unsupervised Learning of Visual Embeddingscited by 204
50 most cited of 60 All 60 (JSON)
Built on by
Citing papers per year: 2022: 73; 2023: 98; 2024: 68; 2025: 38; 2026: 20.
297 papers.
Sort:
By area:
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 982
- 2022EVA: Exploring the Limits of Masked Visual Representation Learning at Scalecited by 685
- 2024Revisiting Feature Prediction for Learning Visual Representations from Videocited by 490
- 2022BEiT v2: Masked Image Modeling with Vector-Quantized Visual Tokenizerscited by 298
- 2023Multimodal Foundation Models: From Specialists to General-Purpose Assistantscited by 247
- 2022MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesiscited by 222
- 2023A Survey on Self-supervised Learning: Algorithms, Applications, and Future Trendscited by 205
- 2023Unmasked Teacher: Towards Training-Efficient Video Foundation Modelscited by 199
- 2022MaskCLIP: Masked Self-Distillation Advances Contrastive Language-Image Pretrainingcited by 187
- 2023EfficientSAM: Leveraged Masked Image Pretraining for Efficient Segment Anythingcited by 187
- 2023Forgery-aware Adaptive Transformer for Generalizable Synthetic Image Detectioncited by 180
- 2022Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Modelscited by 172
38 more
- 2022CroCo: Self-Supervised Pre-training for 3D Vision Tasks by Cross-View Completioncited by 158
- 2022Masked Discrimination for Self-Supervised Learning on Point Cloudscited by 155
- 2022ELEVATER: A Benchmark and Toolkit for Evaluating Language-Augmented Visual Modelscited by 141
- 2022Contrastive Masked Autoencoders are Stronger Vision Learnerscited by 133
- 2023Transformer-Based Visual Segmentation: A Surveycited by 124
- 2022ConvMAE: Masked Convolution Meets Masked Autoencoderscited by 105
- 2022Group DETR: Fast DETR Training with Group-Wise One-to-Many Assignmentcited by 96
- 2022MVP: Multimodality-guided Visual Pre-trainingcited by 92
- 2023Designing BERT for Convolutional Networks: Sparse and Hierarchical Masked Modelingcited by 92
- 2022Siamese Image Modeling for Self-Supervised Vision Representation Learningcited by 78
- 2022E-NeRV: Expedite Neural Video Representation with Disentangled Spatial-Temporal Contextcited by 76
- 2022Masked Image Modeling Advances 3D Medical Image Analysiscited by 71
- 2024Learning and Leveraging World Models in Visual Representation Learningcited by 71
- 2022Masked Frequency Modeling for Self-Supervised Visual Pre-Trainingcited by 70
- 2022MILAN: Masked Image Pretraining on Language Assisted Representationcited by 69
- 2022Bootstrapped Masked Autoencoders for Vision BERT Pretrainingcited by 60
- 2022A Survey on Masked Autoencoder for Self-supervised Learning in Vision and Beyondcited by 58
- 2022Uniform Masking: Enabling MAE Pre-training for Pyramid-based Vision Transformers with Localitycited by 57
- 2022Occupancy-MAE: Self-supervised Pre-training Large-scale LiDAR Point Clouds with Masked Occupancy Autoencoderscited by 57
- 2023Hard Patches Mining for Masked Image Modelingcited by 56
- 2022SdAE: Self-distillated Masked Autoencodercited by 52
- 2022Understanding Masked Image Modeling via Learning Occlusion Invariant Featurecited by 50
- 2023PLIP: Language-Image Pre-training for Person Representation Learningcited by 49
- 2022MixMAE: Mixed and Masked Autoencoder for Efficient Pretraining of Hierarchical Vision Transformerscited by 46
- 2023Masked Image Modeling with Local Multi-Scale Reconstructioncited by 43
- 2023Exploring Effective Factors for Improving Visual In-Context Learningcited by 43
- 2023StrucTexTv2: Masked Visual-Textual Prediction for Document Image Pre-trainingcited by 39
- 2023Know Your Self-supervised Learning: A Survey on Image-based Generative and Discriminative Trainingcited by 39
- 2023Understanding Masked Autoencoders via Hierarchical Latent Variable Modelscited by 39
- 2023GeoMAE: Masked Geometric Target Prediction for Self-supervised Point Cloud Pre-Trainingcited by 38
- 2022A Unified View of Masked Image Modelingcited by 36
- 2024LW-DETR: A Transformer Replacement to YOLO for Real-Time Detectioncited by 36
- 2022CLIP Itself is a Strong Fine-tuner: Achieving 85.7% and 88.0% Top-1 Accuracy with ViT-B and ViT-L on ImageNetcited by 35
- 2022MaskOCR: Text Recognition with Masked Encoder-Decoder Pretrainingcited by 34
- 2022ViTKD: Practical Guidelines for ViT feature knowledge distillationcited by 34
- 2022The Devil is in the Frequency: Geminated Gestalt Autoencoder for Self-Supervised Visual Pre-Trainingcited by 33
- 2022Masked Lip-Sync Prediction by Audio-Visual Contextual Exploitation in Transformerscited by 33
- 2023HAP: Structure-Aware Masked Image Modeling for Human-Centric Perceptioncited by 33
50 most cited of 297 All 297 (JSON)
- 2026Latent evolving World Action Model23 Sep 2026
- 2026ViTAMINS: An Empirical Study of Training Self-Supervised Vision Transformers with Synthetic Hard Negatives1 Sep 2026
- 2026Bar-JEPA: Extracting Values from Bar Chart with Joint-Embedding Predictive Architecture6 Aug 2026
- 2026SiamJEPA: On the Role of Siamese Student Encoders in JEPA4 Jul 2026
- 2026From SRA to Self-Flow: Data Augmentation or Self-Supervision?2 Jul 2026
- 2026ExPLoRe: Expert Patch-Level Loss Routing for Multi-Objective Masked Image Modeling30 Jun 2026
- 2026Who Needs Labels? Adapting Vision Foundation Models With the Metadata You Already Have3 Jun 2026
- 2026Where Do We (Not) Need Temporal Context in Low-Resource Video Task Adaptation?2 Jun 2026
- 2026Remembering by Reconstructing: Domain Incremental Learning With Test-Time Training on Video Streams29 May 2026
- 2026Learning from Semantic Dictionaries: Discriminative Codebook Contrastive Learning for Unified Visual Representation and Generation24 May 2026
- 2026Constrained latent state modeling: A unifying perspective on representation learning under competing constraints15 May 2026
- 2026Learning to Perceive "Where": Spatial Pretext Tasks for Robust Self-Supervised Learning11 May 2026
38 more
- 2026DINO-QPM: Adapting Visual Foundation Models for Globally Interpretable Image Classification8 Apr 2026
- 2026Maximizing Incremental Information Entropy for Contrastive Learning13 Mar 2026
- 2026From Semantics to Pixels: Coarse-to-Fine Masked Autoencoders for Hierarchical Visual Understanding10 Mar 2026
- 2026Mixed Magnification Aggregation for Generalizable Region-Level Representations in Computational Pathology25 Feb 2026
- 2026Xray-Visual Models: Scaling Vision models on Industry Scale Data18 Feb 2026
- 2026Robust Representation Learning in Masked Autoencoders3 Feb 2026
- 2026Aligned Stable Inpainting: Mitigating Unwanted Object Insertion and Preserving Color Consistency21 Jan 2026
- 2026Attentive multilayer fusion for vision transformers14 Jan 2026
- 2025Stochastic Siamese MAE Pretraining for Longitudinal Medical Images29 Dec 2025
- 2025Learning from Next-Frame Prediction: Autoregressive Video Modeling Encodes Effective Representations24 Dec 2025
- 2025Dual-Branch Center-Surrounding Contrast: Rethinking Contrastive Learning for 3D Point Clouds9 Dec 2025
- 2025Repulsor: Accelerating Generative Modeling with a Contrastive Memory Bank9 Dec 2025
- 2025Self-Supervised Multisensory Pretraining for Contact-Rich Robot Reinforcement Learning18 Nov 2025
- 2025Attentive Feature Aggregation or: How Policies Learn to Stop Worrying about Robustness and Attend to Task-Relevant Visual Cues13 Nov 2025
- 2025Exploring Structural Degradation in Dense Representations for Self-supervised Learning20 Oct 2025
- 2025SSL4RL: Revisiting Self-supervised Learning as Intrinsic Reward for Visual-Language Reasoning18 Oct 2025
- 2025Unmute the Patch Tokens: Rethinking Probing in Multi-Label Audio Classification29 Sep 2025
- 2025LCMF: Lightweight Cross-Modality Mambaformer for Embodied Robotics VQA23 Sep 2025
- 2025Domain Knowledge is Power: Leveraging Physiological Priors for Self Supervised Representation Learning in Electrocardiography9 Sep 2025
- 2025Anatomic Feature Fusion Model for Diagnosing Calcified Pulmonary Nodules on Chest X-Ray18 Aug 2025
- 2025TESPEC: Temporally-Enhanced Self-Supervised Pretraining for Event Cameras29 Jul 2025
- 2025Dataset Ownership Verification for Pre-trained Masked Models16 Jul 2025
- 2025Attention, Please! Revisiting Attentive Probing Through the Lens of Efficiency11 Jun 2025
- 2025Weakly-Supervised Affordance Grounding Guided by Part-Level Semantic Priors30 May 2025
- 2025A Unified and Scalable Membership Inference Method for Visual Self-supervised Encoder via Part-aware Capability15 May 2025
- 2025Total Variation-Based Image Decomposition and Denoising for Microscopy Images13 May 2025
- 2025CGTrack: Cascade Gating Network with Hierarchical Feature Aggregation for UAV Tracking9 May 2025
- 2025Automated Learning of Semantic Embedding Representations for Diffusion Models9 May 2025
- 2025SparseJEPA: Sparse Representation Learning of Joint Embedding Predictive Architectures22 Apr 2025
- 2025A Self-supervised Learning Method for Raman Spectroscopy based on Masked Autoencoders21 Apr 2025
- 2025Can Masked Autoencoders Also Listen to Birds?17 Apr 2025
- 2025Evolved Hierarchical Masking for Self-Supervised Learning12 Apr 2025
- 2025Learning Universal Features for Generalizable Image Forgery Localization10 Apr 2025
- 2025Uni4D: A Unified Self-Supervised Learning Framework for Point Cloud Videos7 Apr 2025
- 2025MergeVQ: A Unified Framework for Visual Generation and Representation with Disentangled Token Merging and Quantization1 Apr 2025
- 2025Conjuring Positive Pairs for Efficient Unification of Representation Learning and Image Synthesis19 Mar 2025
- 2025RoMA: Scaling up Mamba-based Foundation Models for Remote Sensing13 Mar 2025
- 2025Robustness Tokens: Towards Adversarial Robustness of Transformers13 Mar 2025
50 newest of 297 All 297 (JSON)
- 2023Forgery-aware Adaptive Transformer for Generalizable Synthetic Image Detectioncited by 180
- 2024Improving Diffusion-Based Image Synthesis with Context Predictioncited by 32
- 2023ADDP: Learning General Representations for Image Recognition and Generation with Alternating Denoising Diffusion Processcited by 19
- 2023Towards Enhanced Image Inpainting: Mitigating Unwanted Object Insertion and Preserving Color Consistencycited by 19
- 2024CTSyn: A Foundation Model for Cross Tabular Data Generationcited by 13
- 2024Denoising with a Joint-Embedding Predictive Architecturecited by 12
- 2024Residual Connections Harm Generative Representation Learningcited by 10
- 2024DiffSTR: Controlled Diffusion Models for Scene Text Removalcited by 2
- 2025Learning from Next-Frame Prediction: Autoregressive Video Modeling Encodes Effective Representationscited by 1
- 2025Automated Learning of Semantic Embedding Representations for Diffusion Modelscited by 0
- 2025Repulsor: Accelerating Generative Modeling with a Contrastive Memory Bankcited by 0
- 2026From SRA to Self-Flow: Data Augmentation or Self-Supervision?cited by 0
- 2026From SRA to Self-Flow: Data Augmentation or Self-Supervision?2 Jul 2026
- 2025Learning from Next-Frame Prediction: Autoregressive Video Modeling Encodes Effective Representations24 Dec 2025
- 2025Repulsor: Accelerating Generative Modeling with a Contrastive Memory Bank9 Dec 2025
- 2025Automated Learning of Semantic Embedding Representations for Diffusion Models9 May 2025
- 2024DiffSTR: Controlled Diffusion Models for Scene Text Removal29 Oct 2024
- 2024Denoising with a Joint-Embedding Predictive Architecture2 Oct 2024
- 2024CTSyn: A Foundation Model for Cross Tabular Data Generation7 Jun 2024
- 2024Residual Connections Harm Generative Representation Learning16 Apr 2024
- 2024Improving Diffusion-Based Image Synthesis with Context Prediction4 Jan 2024
- 2023Forgery-aware Adaptive Transformer for Generalizable Synthetic Image Detection27 Dec 2023
- 2023Towards Enhanced Image Inpainting: Mitigating Unwanted Object Insertion and Preserving Color Consistency8 Dec 2023
- 2023ADDP: Learning General Representations for Image Recognition and Generation with Alternating Denoising Diffusion Process8 Jun 2023
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 982
- 2024Revisiting Feature Prediction for Learning Visual Representations from Videocited by 490
- 2024Learning and Leveraging World Models in Visual Representation Learningcited by 71
- 2026Latent evolving World Action Modelcited by 0
- 2026Latent evolving World Action Model23 Sep 2026
- 2024Learning and Leveraging World Models in Visual Representation Learning1 Mar 2024
- 2024Revisiting Feature Prediction for Learning Visual Representations from Video15 Feb 2024
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecture19 Jan 2023
- 2022EVA: Exploring the Limits of Masked Visual Representation Learning at Scalecited by 685
- 2022BEiT v2: Masked Image Modeling with Vector-Quantized Visual Tokenizerscited by 298
- 2023Multimodal Foundation Models: From Specialists to General-Purpose Assistantscited by 247
- 2022MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesiscited by 222
- 2023A Survey on Self-supervised Learning: Algorithms, Applications, and Future Trendscited by 205
- 2023Unmasked Teacher: Towards Training-Efficient Video Foundation Modelscited by 199
- 2022MaskCLIP: Masked Self-Distillation Advances Contrastive Language-Image Pretrainingcited by 187
- 2023EfficientSAM: Leveraged Masked Image Pretraining for Efficient Segment Anythingcited by 187
- 2022Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Modelscited by 172
- 2022CroCo: Self-Supervised Pre-training for 3D Vision Tasks by Cross-View Completioncited by 158
- 2022Masked Discrimination for Self-Supervised Learning on Point Cloudscited by 155
- 2022ELEVATER: A Benchmark and Toolkit for Evaluating Language-Augmented Visual Modelscited by 141
12 most cited of 266 All 266 (JSON)
- 2026ViTAMINS: An Empirical Study of Training Self-Supervised Vision Transformers with Synthetic Hard Negatives1 Sep 2026
- 2026Bar-JEPA: Extracting Values from Bar Chart with Joint-Embedding Predictive Architecture6 Aug 2026
- 2026SiamJEPA: On the Role of Siamese Student Encoders in JEPA4 Jul 2026
- 2026Who Needs Labels? Adapting Vision Foundation Models With the Metadata You Already Have3 Jun 2026
- 2026Where Do We (Not) Need Temporal Context in Low-Resource Video Task Adaptation?2 Jun 2026
- 2026Remembering by Reconstructing: Domain Incremental Learning With Test-Time Training on Video Streams29 May 2026
- 2026Learning from Semantic Dictionaries: Discriminative Codebook Contrastive Learning for Unified Visual Representation and Generation24 May 2026
- 2026Constrained latent state modeling: A unifying perspective on representation learning under competing constraints15 May 2026
- 2026Learning to Perceive "Where": Spatial Pretext Tasks for Robust Self-Supervised Learning11 May 2026
- 2026DINO-QPM: Adapting Visual Foundation Models for Globally Interpretable Image Classification8 Apr 2026
- 2026Maximizing Incremental Information Entropy for Contrastive Learning13 Mar 2026
- 2026From Semantics to Pixels: Coarse-to-Fine Masked Autoencoders for Hierarchical Visual Understanding10 Mar 2026
12 newest of 266 All 266 (JSON)