paper-with-me

Papers

Robust Representation Learning in Masked Autoencoders

2026-02-03 · Anika Shrivastava, Renu Rameshan, Samar Agnihotri arxiv

Masked Autoencoders (MAEs) achieve impressive performance in image classification tasks, yet the internal representations they learn remain less understood. This work started as an attempt to understand the strong downstream classification performance of MAE. In this process we discover that representations learned with the pretraining and fine-tuning, are quite robust -- demonstrating a good classification performance in the presence of degradations, such as blur and occlusions. Through layer-wise analysis of token embeddings, we show that pretrained MAE progressively constructs its latent space in a class-aware manner across network depth: embeddings from different classes lie in subspaces that become increasingly separable. We further observe that MAE exhibits early and persistent global attention across encoder layers, in contrast to standard Vision Transformers (ViTs). To quantify feature robustness, we introduce two sensitivity indicators: directional alignment between clean and perturbed embeddings, and head-wise retention of active features under degradations. These studies help establish the robust classification performance of MAEs.

📄 PDF Abstract BibTeX arXiv:2602.03531

Code (0)

등록된 구현이 없습니다.

Tasks

Representation LearningImage Classification

Similar Papers 제목 키워드 기반

Exploring Target Representations for Masked Autoencoders

2022-09-08 · Xingbin Liu, Jinghao Zhou, Tao Kong, Xianming Lin 외

Masked autoencoders have become popular training paradigms for self-supervised visual representation learning. These models randomly mask a portion of the input and reconstruct the masked portion according to the target …

Image ClassificationInstance SegmentationKnowledge Distillationobject-detection+5

Diffusion Models as Masked Autoencoders

2023-04-06 · ICCV 2023 1 · Chen Wei, Karttikeya Mangalam, Po-Yao Huang, Yanghao Li 외

There has been a longstanding belief that generation can facilitate a true understanding of visual data. In line with this, we revisit generatively pre-training visual representations in light of recent interest in denoi…

DenoisingImage Inpainting

Rethinking Patch Dependence for Masked Autoencoders

2024-01-25 · Letian Fu, Long Lian, Renhao Wang, Baifeng Shi 외

In this work, we re-examine inter-patch dependencies in the decoding mechanism of masked autoencoders (MAE). We decompose this decoding mechanism for masked patch reconstruction in MAE into self-attention and cross-atten…

DecoderInstance SegmentationRepresentation LearningSemantic Segmentation

Quantum Masked Autoencoders for Vision Learning

2025-11-21 · Emma Andrews, Prabhat Mishra arxiv

Classical autoencoders are widely used to learn features of input data. To improve the feature learning, classical masked autoencoders extend classical autoencoders to learn the features of the original input sample in t…

i-MAE: Are Latent Representations in Masked Autoencoders Linearly Separable?

2022-10-20 · Kevin Zhang, Zhiqiang Shen

Masked image modeling (MIM) has been recognized as a strong self-supervised pre-training approach in the vision domain. However, the mechanism and properties of the learned representations by such a scheme, as well as ho…

Image Reconstruction