Conditional Predictive Sufficient Statistics for Visual Representation Learning
Not on the map. Why?
Abstract (arXiv)
A useful visual representation is a statistic of the observed past that retains the latent factors shared with the future and discards patch-private noise. We formalize this requirement as a conditional predictive sufficient statistic (CPSS). Under a shared-factor model of image patches, the mutual information between the past and the next patch equals the information the past carries about the shared factor, up to a remainder that the next patch itself fails to reveal. Predicting the next patch embedding with a cosine loss is maximum likelihood for a von Mises-Fisher model of that embedding's direction, and is therefore a tractable surrogate for the predictive information. The same population loss is also minimized by a constant embedding, so stop-gradient does not by itself select the sufficient statistic; it only blocks the symmetric gradient that implements the constant solution in one step. The regression target is a shallow embedding, which forces the network output back into that shallow range and leaves the sufficient statistic in intermediate blocks. Small causal Transformers on MNIST and CIFAR-10 are used as diagnostics, not as a leaderboard. On MNIST the future shift and the stop-gradient move probe accuracy by tens of points, and the CPSS readout peaks before the output. On CIFAR-10, with the same short budget and no augmentation, every objective lands near a linear classifier on pixels. What still matches the derivation is the geometry: the CPSS output is a worse readout than its best intermediate block, next-pixel regression does not pay that penalty, and removing the stop-gradient collapses the effective rank of the embedding even when the pretext loss looks perfect.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 62 of its 81 references that are in the map. First posted on arXiv 25 Sep 2026.
Builds on
62 of 81 references.
References · 62
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2020Denoising Diffusion Implicit Modelscited by 9,781
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
38 more
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,750
- 2017Neural Discrete Representation Learningcited by 4,712
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
- 2021Zero-Shot Text-to-Image Generationcited by 4,288
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,253
- 2020Exploring Simple Siamese Representation Learningcited by 2,866
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,857
- 2021BEiT: BERT Pre-Training of Image Transformerscited by 2,375
- 2018Unsupervised Feature Learning via Non-Parametric Instance-level Discriminationcited by 1,845
- 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,838
- 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,616
- 2018Unsupervised Representation Learning by Predicting Image Rotationscited by 1,591
- 2016Context Encoders: Feature Learning by Inpaintingcited by 1,428
- 2020Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hyperspherecited by 1,405
- 2016Unsupervised Learning of Visual Representations by Solving Jigsaw Puzzlescited by 1,382
- 2018Learning deep representations by mutual information estimation and maximizationcited by 1,273
- 2021SimMIM: A Simple Framework for Masked Image Modelingcited by 1,153
- 2016Colorful Image Colorizationcited by 1,114
- 2022MaskGIT: Masked Generative Image Transformercited by 1,099
- 2015Unsupervised Visual Representation Learning by Context Predictioncited by 1,066
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,057
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 1,039
- 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,023
- 2016Deep Variational Information Bottleneckcited by 930
- 2024Autoregressive Model Beats Diffusion: Llama for Scalable Image Generationcited by 777
- 2019Learning Representations by Maximizing Mutual Information Across Viewscited by 734
- 2022data2vec: A General Framework for Self-supervised Learning in Speech, Vision and Languagecited by 727
- 2022Autoregressive Image Generation using Residual Quantizationcited by 722
- 2019Data-Efficient Image Recognition with Contrastive Predictive Codingcited by 673
- 2024Autoregressive Image Generation without Vector Quantizationcited by 651
- 2024Revisiting Feature Prediction for Learning Visual Representations from Videocited by 531
- 2021Understanding Dimensional Collapse in Contrastive Self-supervised Learningcited by 378
- 2019On Mutual Information Maximization for Representation Learningcited by 316
50 most cited of 62 All 62 (JSON)
Built on by
Not cited yet.