Matrix Information Theory for Self-Supervised Learning
Not on the map. Why?
Abstract (arXiv)
The maximum entropy encoding framework provides a unified perspective for many non-contrastive learning methods like SimSiam, Barlow Twins, and MEC. Inspired by this framework, we introduce Matrix-SSL, a novel approach that leverages matrix information theory to interpret the maximum entropy encoding loss as matrix uniformity loss. Furthermore, Matrix-SSL enhances the maximum entropy encoding method by seamlessly incorporating matrix alignment loss, directly aligning covariance matrices in different branches. Experimental results reveal that Matrix-SSL outperforms state-of-the-art methods on the ImageNet dataset under linear evaluation settings and on MS-COCO for transfer learning tasks. Specifically, when performing transfer learning tasks on MS-COCO, our method outperforms previous SOTA methods such as MoCo v2 and BYOL up to 3.3% with only 400 epochs compared to 800 epochs pre-training. We also try to introduce representation learning into the language modeling regime by fine-tuning a 7B model using matrix cross-entropy loss, with a margin of 3.1% on the GSM8K dataset over the standard cross-entropy loss. Code available at https://github.com/yifanzhang-pro/Matrix-SSL.
Key facts. Cited by 21 papers in the map (data through 14 Oct 2026). Builds on 69 of its 94 references that are in the map. First posted on arXiv 27 May 2023.
Builds on
69 of 94 references.
References · 69
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 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
38 more
- 2020Improved Baselines with Momentum Contrastive Learningcited by 2,287
- 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
- 2019Contrastive Multiview Codingcited by 1,301
- 2022CoCa: Contrastive Captioners are Image-Text Foundation Modelscited by 1,274
- 2018Learning deep representations by mutual information estimation and maximizationcited by 1,273
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,057
- 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
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 719
- 2019Data-Efficient Image Recognition with Contrastive Predictive Codingcited by 673
- 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 659
- 2020Contrastive Learning with Hard Negative Samplescited by 628
- 2019A Theoretical Analysis of Contrastive Unsupervised Representation Learningcited by 530
- 2022Image as a Foreign Language: BEiT Pretraining for All Vision and Vision-Language Taskscited by 511
- 2021Provable Guarantees for Self-Supervised Deep Learning with Spectral Contrastive Losscited by 313
- 2020Towards Understanding Ensemble, Knowledge Distillation and Self-Distillation in Deep Learningcited by 297
- 2021Understanding self-supervised Learning Dynamics without Contrastive Pairscited by 256
- 2019Unsupervised Embedding Learning via Invariant and Spreading Instance Featurecited by 232
- 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 227
- 2021Contrastive Learning Inverts the Data Generating Processcited by 221
- 2018An efficient framework for learning sentence representationscited by 198
- 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
- 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
- 2021On the Role of Neural Collapse in Transfer Learningcited by 100
- 2021Towards the Generalization of Contrastive Self-Supervised Learningcited by 99
- 2022Understanding Contrastive Learning Requires Incorporating Inductive Biasescited by 96
- 2022Are All Losses Created Equal: A Neural Collapse Perspectivecited by 78
- 2021Self-Supervised Learning with Kernel Dependence Maximizationcited by 77
- 2022Connect, Not Collapse: Explaining Contrastive Learning for Unsupervised Domain Adaptationcited by 67
- 2020Contrastive estimation reveals topic posterior information to linear modelscited by 54
50 most cited of 69 All 69 (JSON)
Built on by
Citing papers per year: 2023: 5; 2024: 8; 2025: 4; 2026: 4.
21 papers.
Sort:
- 2025LeJEPA: Provable and Scalable Self-Supervised Learning Without the Heuristicscited by 254
- 2024Diff-eRank: A Novel Rank-Based Metric for Evaluating Large Language Modelscited by 35
- 2023Information Flow in Self-Supervised Learningcited by 16
- 2024Provable Contrastive Continual Learningcited by 13
- 2023Understanding Grokking Through A Robustness Viewpointcited by 12
- 2023OTMatch: Improving Semi-Supervised Learning with Optimal Transportcited by 6
- 2024The Information of Large Language Model Geometrycited by 5
- 2024Scaling Image Tokenizers with Grouped Spherical Quantizationcited by 4
- 2025AdaDim: Dimensionality Adaptation for SSL Representational Dynamicscited by 4
- 2024Unveiling the Dynamics of Information Interplay in Supervised Learningcited by 3
- 2023FroSSL: Frobenius Norm Minimization for Efficient Multiview Self-Supervised Learningcited by 2
- 2024Exploiting Supervised Poison Vulnerability to Strengthen Self-Supervised Defensecited by 1
9 more
- 2024Exploring Information-Theoretic Metrics Associated with Neural Collapse in Supervised Trainingcited by 1
- 2025Countering Multi-modal Representation Collapse through Rank-targeted Fusioncited by 1
- 2026Vendi Novelty Scores for Out-of-Distribution Detectioncited by 1
- 2023Deep Tensor Networkcited by 0
- 2024Uncovering RL Integration in SSL Loss: Objective-Specific Implications for Data-Efficient RLcited by 0
- 2025Information-Theoretic Perspectives on Optimizerscited by 0
- 2026Maximizing Incremental Information Entropy for Contrastive Learningcited by 0
- 2026Neural Collapse by Design: Learning Class Prototypes on the Hyperspherecited by 0
- 2026Efficient, Validation-Free Intrinsic Quality Estimation for Large-Scale Face Recognition Datasetscited by 0
- 2026Efficient, Validation-Free Intrinsic Quality Estimation for Large-Scale Face Recognition Datasets28 May 2026
- 2026Neural Collapse by Design: Learning Class Prototypes on the Hypersphere19 May 2026
- 2026Maximizing Incremental Information Entropy for Contrastive Learning13 Mar 2026
- 2026Vendi Novelty Scores for Out-of-Distribution Detection10 Feb 2026
- 2025LeJEPA: Provable and Scalable Self-Supervised Learning Without the Heuristics11 Nov 2025
- 2025Countering Multi-modal Representation Collapse through Rank-targeted Fusion9 Nov 2025
- 2025AdaDim: Dimensionality Adaptation for SSL Representational Dynamics18 May 2025
- 2025Information-Theoretic Perspectives on Optimizers28 Feb 2025
- 2024Scaling Image Tokenizers with Grouped Spherical Quantization3 Dec 2024
- 2024Uncovering RL Integration in SSL Loss: Objective-Specific Implications for Data-Efficient RL22 Oct 2024
- 2024Exploring Information-Theoretic Metrics Associated with Neural Collapse in Supervised Training25 Sep 2024
- 2024Exploiting Supervised Poison Vulnerability to Strengthen Self-Supervised Defense13 Sep 2024
9 more
- 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
- 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
- 2023FroSSL: Frobenius Norm Minimization for Efficient Multiview Self-Supervised Learning4 Oct 2023
- 2023Information Flow in Self-Supervised Learning29 Sep 2023
- 2025LeJEPA: Provable and Scalable Self-Supervised Learning Without the Heuristicscited by 254
- 2024Diff-eRank: A Novel Rank-Based Metric for Evaluating Large Language Modelscited by 35
- 2023Information Flow in Self-Supervised Learningcited by 16
- 2024Provable Contrastive Continual Learningcited by 13
- 2023Understanding Grokking Through A Robustness Viewpointcited by 12
- 2023OTMatch: Improving Semi-Supervised Learning with Optimal Transportcited by 6
- 2024The Information of Large Language Model Geometrycited by 5
- 2024Scaling Image Tokenizers with Grouped Spherical Quantizationcited by 4
- 2025AdaDim: Dimensionality Adaptation for SSL Representational Dynamicscited by 4
- 2024Unveiling the Dynamics of Information Interplay in Supervised Learningcited by 3
- 2023FroSSL: Frobenius Norm Minimization for Efficient Multiview Self-Supervised Learningcited by 2
- 2024Exploiting Supervised Poison Vulnerability to Strengthen Self-Supervised Defensecited by 1
12 most cited of 21 All 21 (JSON)
- 2026Efficient, Validation-Free Intrinsic Quality Estimation for Large-Scale Face Recognition Datasets28 May 2026
- 2026Neural Collapse by Design: Learning Class Prototypes on the Hypersphere19 May 2026
- 2026Maximizing Incremental Information Entropy for Contrastive Learning13 Mar 2026
- 2026Vendi Novelty Scores for Out-of-Distribution Detection10 Feb 2026
- 2025LeJEPA: Provable and Scalable Self-Supervised Learning Without the Heuristics11 Nov 2025
- 2025Countering Multi-modal Representation Collapse through Rank-targeted Fusion9 Nov 2025
- 2025AdaDim: Dimensionality Adaptation for SSL Representational Dynamics18 May 2025
- 2025Information-Theoretic Perspectives on Optimizers28 Feb 2025
- 2024Scaling Image Tokenizers with Grouped Spherical Quantization3 Dec 2024
- 2024Uncovering RL Integration in SSL Loss: Objective-Specific Implications for Data-Efficient RL22 Oct 2024
- 2024Exploring Information-Theoretic Metrics Associated with Neural Collapse in Supervised Training25 Sep 2024
- 2024Exploiting Supervised Poison Vulnerability to Strengthen Self-Supervised Defense13 Sep 2024
12 newest of 21 All 21 (JSON)