Information Flow in Self-Supervised Learning
Not on the map. Why?
Abstract (arXiv)
In this paper, we conduct a comprehensive analysis of two dual-branch (Siamese architecture) self-supervised learning approaches, namely Barlow Twins and spectral contrastive learning, through the lens of matrix mutual information. We prove that the loss functions of these methods implicitly optimize both matrix mutual information and matrix joint entropy. This insight prompts us to further explore the category of single-branch algorithms, specifically MAE and U-MAE, for which mutual information and joint entropy become the entropy. Building on this intuition, we introduce the Matrix Variational Masked Auto-Encoder (M-MAE), a novel method that leverages the matrix-based estimation of entropy as a regularizer and subsumes U-MAE as a special case. The empirical evaluations underscore the effectiveness of M-MAE compared with the state-of-the-art methods, including a 3.9% improvement in linear probing ViT-Base, and a 1% improvement in fine-tuning ViT-Large, both on ImageNet.
Key facts. Cited by 16 papers in the map (data through 14 Oct 2026). Builds on 63 of its 69 references that are in the map. First posted on arXiv 29 Sep 2023.
Builds on
63 of 69 references.
References · 63
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,750
- 2020Exploring Simple Siamese Representation Learningcited by 2,866
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,857
- 2021SimCSE: Simple Contrastive Learning of Sentence Embeddingscited by 2,173
- 2018Unsupervised Feature Learning via Non-Parametric Instance-level Discriminationcited by 1,845
- 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,838
- 2020Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hyperspherecited by 1,405
38 more
- 2019Contrastive Multiview Codingcited by 1,301
- 2018Learning deep representations by mutual information estimation and maximizationcited by 1,273
- 2021SimMIM: A Simple Framework for Masked Image Modelingcited by 1,153
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,057
- 2021iBOT: Image BERT Pre-Training with Online Tokenizercited by 903
- 2020What Makes for Good Views for Contrastive Learning?cited by 846
- 2019Self-Supervised Learning of Pretext-Invariant Representationscited by 808
- 2019Learning Representations by Maximizing Mutual Information Across Viewscited by 734
- 2019Data-Efficient Image Recognition with Contrastive Predictive Codingcited by 673
- 2020Contrastive Learning with Hard Negative Samplescited by 628
- 2019A Theoretical Analysis of Contrastive Unsupervised Representation Learningcited by 530
- 2016Tutorial on Variational Autoencoderscited by 409
- 2021Provable Guarantees for Self-Supervised Deep Learning with Spectral Contrastive Losscited by 313
- 2021Understanding self-supervised Learning Dynamics without Contrastive Pairscited by 256
- 2019Unsupervised Embedding Learning via Invariant and Spreading Instance Featurecited by 232
- 2022RankMe: Assessing the downstream performance of pretrained self-supervised representations by their rankcited by 153
- 2020Predicting What You Already Know Helps: Provable Self-Supervised Learningcited by 152
- 2022Contrastive and Non-Contrastive Self-Supervised Learning Recover Global and Local Spectral Embedding Methodscited by 149
- 2020Contrastive learning, multi-view redundancy, and linear modelscited by 146
- 2020Learning Diverse and Discriminative Representations via the Principle of Maximal Coding Rate Reductioncited by 142
- 2021Toward Understanding the Feature Learning Process of Self-supervised Contrastive Learningcited by 119
- 2022On the duality between contrastive and non-contrastive self-supervised learningcited by 108
- 2022Chaos is a Ladder: A New Theoretical Understanding of Contrastive Learning via Augmentation Overlapcited by 105
- 2021Towards the Generalization of Contrastive Self-Supervised Learningcited by 99
- 2022Understanding Contrastive Learning Requires Incorporating Inductive Biasescited by 96
- 2022How Mask Matters: Towards Theoretical Understandings of Masked Autoencoderscited by 86
- 2021Self-Supervised Learning with Kernel Dependence Maximizationcited by 77
- 2023To Compress or Not to Compress- Self-Supervised Learning and Information Theory: A Reviewcited by 72
- 2022A Survey on Masked Autoencoder for Self-supervised Learning in Vision and Beyondcited by 58
- 2020Contrastive estimation reveals topic posterior information to linear modelscited by 54
- 2021Exploring the Equivalence of Siamese Self-Supervised Learning via A Unified Gradient Frameworkcited by 49
- 2021Understanding Negative Samples in Instance Discriminative Self-supervised Representation Learningcited by 47
- 2022Beyond Separability: Analyzing the Linear Transferability of Contrastive Representations to Related Subpopulationscited by 44
- 2022A Theoretical Study of Inductive Biases in Contrastive Learningcited by 40
- 2023The SSL Interplay: Augmentations, Inductive Bias, and Generalizationcited by 40
- 2023Understanding Masked Autoencoders via Hierarchical Latent Variable Modelscited by 39
- 2022How to Understand Masked Autoencoderscited by 38
- 2022The Mechanism of Prediction Head in Non-contrastive Self-supervised Learningcited by 38
50 most cited of 63 All 63 (JSON)
Built on by
Citing papers per year: 2023: 3; 2024: 7; 2025: 5; 2026: 1.
16 papers.
Sort:
- 2024Diff-eRank: A Novel Rank-Based Metric for Evaluating Large Language Modelscited by 35
- 2024Provable Contrastive Continual Learningcited by 13
- 2023Understanding Grokking Through A Robustness Viewpointcited by 12
- 2025Projection Head is Secretly an Information Bottleneckcited by 11
- 2023OTMatch: Improving Semi-Supervised Learning with Optimal Transportcited by 6
- 2024The Information of Large Language Model Geometrycited by 5
- 2025AdaDim: Dimensionality Adaptation for SSL Representational Dynamicscited by 4
- 2024Unveiling the Dynamics of Information Interplay in Supervised Learningcited by 3
- 2024Learning to Embed Distributions via Maximum Kernel Entropycited by 3
- 2024Large Language Model Evaluation via Matrix Nuclear-Normcited by 2
- 2024Exploring Information-Theoretic Metrics Associated with Neural Collapse in Supervised Trainingcited by 1
- 2023Deep Tensor Networkcited by 0
4 more
- 2025Information-Theoretic Perspectives on Optimizerscited by 0
- 2025Self-Supervised Representation Learning as Mutual Information Maximizationcited by 0
- 2025Information Theory in Open-world Machine Learning Foundations, Frameworks, and Future Directioncited by 0
- 2026Information theoretic underpinning of self-supervised learning by clusteringcited by 0
- 2026Information theoretic underpinning of self-supervised learning by clustering12 May 2026
- 2025Information Theory in Open-world Machine Learning Foundations, Frameworks, and Future Direction17 Oct 2025
- 2025Self-Supervised Representation Learning as Mutual Information Maximization1 Oct 2025
- 2025AdaDim: Dimensionality Adaptation for SSL Representational Dynamics18 May 2025
- 2025Projection Head is Secretly an Information Bottleneck1 Mar 2025
- 2025Information-Theoretic Perspectives on Optimizers28 Feb 2025
- 2024Large Language Model Evaluation via Matrix Nuclear-Norm14 Oct 2024
- 2024Exploring Information-Theoretic Metrics Associated with Neural Collapse in Supervised Training25 Sep 2024
- 2024Learning to Embed Distributions via Maximum Kernel Entropy1 Aug 2024
- 2024Unveiling the Dynamics of Information Interplay in Supervised Learning6 Jun 2024
- 2024Provable Contrastive Continual Learning29 May 2024
- 2024The Information of Large Language Model Geometry1 Feb 2024
4 more
- 2024Diff-eRank: A Novel Rank-Based Metric for Evaluating Large Language Models30 Jan 2024
- 2023Deep Tensor Network18 Nov 2023
- 2023Understanding Grokking Through A Robustness Viewpoint11 Nov 2023
- 2023OTMatch: Improving Semi-Supervised Learning with Optimal Transport26 Oct 2023
- 2024Diff-eRank: A Novel Rank-Based Metric for Evaluating Large Language Modelscited by 35
- 2024Provable Contrastive Continual Learningcited by 13
- 2023Understanding Grokking Through A Robustness Viewpointcited by 12
- 2025Projection Head is Secretly an Information Bottleneckcited by 11
- 2023OTMatch: Improving Semi-Supervised Learning with Optimal Transportcited by 6
- 2024The Information of Large Language Model Geometrycited by 5
- 2025AdaDim: Dimensionality Adaptation for SSL Representational Dynamicscited by 4
- 2024Unveiling the Dynamics of Information Interplay in Supervised Learningcited by 3
- 2024Learning to Embed Distributions via Maximum Kernel Entropycited by 3
- 2024Large Language Model Evaluation via Matrix Nuclear-Normcited by 2
- 2024Exploring Information-Theoretic Metrics Associated with Neural Collapse in Supervised Trainingcited by 1
- 2023Deep Tensor Networkcited by 0
12 most cited of 16 All 16 (JSON)
- 2026Information theoretic underpinning of self-supervised learning by clustering12 May 2026
- 2025Information Theory in Open-world Machine Learning Foundations, Frameworks, and Future Direction17 Oct 2025
- 2025Self-Supervised Representation Learning as Mutual Information Maximization1 Oct 2025
- 2025AdaDim: Dimensionality Adaptation for SSL Representational Dynamics18 May 2025
- 2025Projection Head is Secretly an Information Bottleneck1 Mar 2025
- 2025Information-Theoretic Perspectives on Optimizers28 Feb 2025
- 2024Large Language Model Evaluation via Matrix Nuclear-Norm14 Oct 2024
- 2024Exploring Information-Theoretic Metrics Associated with Neural Collapse in Supervised Training25 Sep 2024
- 2024Learning to Embed Distributions via Maximum Kernel Entropy1 Aug 2024
- 2024Unveiling the Dynamics of Information Interplay in Supervised Learning6 Jun 2024
- 2024Provable Contrastive Continual Learning29 May 2024
- 2024The Information of Large Language Model Geometry1 Feb 2024
12 newest of 16 All 16 (JSON)