mlsanityData to 14 Oct 2026

Conditional Predictive Sufficient Statistics for Visual Representation Learning

arXiv 2609.30647 ↗First posted 25 Sep 2026cs.CVcited by 0

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

  1. 2017Attention Is All You Needcited by 60,361
  2. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  3. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  4. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  5. 2020Language Models are Few-Shot Learnerscited by 32,419
  6. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  7. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  8. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
  9. 2020Denoising Diffusion Implicit Modelscited by 9,781
  10. 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
  11. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
  12. 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
38 more
  1. 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
  2. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
  3. 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
  4. 2022Scalable Diffusion Models with Transformerscited by 6,054
  5. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
  6. 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,750
  7. 2017Neural Discrete Representation Learningcited by 4,712
  8. 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
  9. 2021Zero-Shot Text-to-Image Generationcited by 4,288
  10. 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,253
  11. 2020Exploring Simple Siamese Representation Learningcited by 2,866
  12. 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,857
  13. 2021BEiT: BERT Pre-Training of Image Transformerscited by 2,375
  14. 2018Unsupervised Feature Learning via Non-Parametric Instance-level Discriminationcited by 1,845
  15. 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,838
  16. 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,616
  17. 2018Unsupervised Representation Learning by Predicting Image Rotationscited by 1,591
  18. 2016Context Encoders: Feature Learning by Inpaintingcited by 1,428
  19. 2020Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hyperspherecited by 1,405
  20. 2016Unsupervised Learning of Visual Representations by Solving Jigsaw Puzzlescited by 1,382
  21. 2018Learning deep representations by mutual information estimation and maximizationcited by 1,273
  22. 2021SimMIM: A Simple Framework for Masked Image Modelingcited by 1,153
  23. 2016Colorful Image Colorizationcited by 1,114
  24. 2022MaskGIT: Masked Generative Image Transformercited by 1,099
  25. 2015Unsupervised Visual Representation Learning by Context Predictioncited by 1,066
  26. 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,057
  27. 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 1,039
  28. 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,023
  29. 2016Deep Variational Information Bottleneckcited by 930
  30. 2024Autoregressive Model Beats Diffusion: Llama for Scalable Image Generationcited by 777
  31. 2019Learning Representations by Maximizing Mutual Information Across Viewscited by 734
  32. 2022data2vec: A General Framework for Self-supervised Learning in Speech, Vision and Languagecited by 727
  33. 2022Autoregressive Image Generation using Residual Quantizationcited by 722
  34. 2019Data-Efficient Image Recognition with Contrastive Predictive Codingcited by 673
  35. 2024Autoregressive Image Generation without Vector Quantizationcited by 651
  36. 2024Revisiting Feature Prediction for Learning Visual Representations from Videocited by 531
  37. 2021Understanding Dimensional Collapse in Contrastive Self-supervised Learningcited by 378
  38. 2019On Mutual Information Maximization for Representation Learningcited by 316

50 most cited of 62 All 62 (JSON)

Built on by

Not cited yet.