MATH · IN · MODELS

MAE per-class patch-token subspaces rotate apart with depth

measured in 1 paper

Shrivastava et al. take the pretrained MAE ViT-Base encoder and, per class, SVD the visible patch-token embeddings at each layer to get a k-dimensional class subspace [shrivastava-etal-2026-robust-representation-learning-mae] The smallest principal angle between per-class subspaces grows with depth, so class-specific subspaces rotate apart and become well-separated without labels, corroborated by t-SNE around layers 7-9 [shrivastava-etal-2026-robust-representation-learning-mae] This class structure is linked to MAE's persistent global attention (mean attention distance ~80-120px at all layers, unlike standard ViT's local-to-global progression) [shrivastava-etal-2026-robust-representation-learning-mae] Fine-tuned MAE stays above 80% accuracy under Gaussian blur and above 60% at 90% occlusion, with correctly-classified images showing higher clean-vs-perturbed cosine similarity; no causal intervention is performed [shrivastava-etal-2026-robust-representation-learning-mae]

Context

per-class SVD subspace of patch-token embeddings at each encoder layer, principal angles (theta_1) between class-specific subspaces as a separation metric, persistent global attention (mean attention distance ~80-120px at all depths) contrasted with standard ViT's local-to-global progression, directional robustness (cosine similarity) and head-wise active-feature retention under blur/occlusion

Papers

Robust Representation Learning in Masked Autoencoders — Shrivastava, Anika, Rameshan, Renu, Agnihotri, Samar2026 · arXiv:2602.03531