Modulate Your Spectrum in Self-Supervised Learning
Not on the map. Why?
Abstract (arXiv)
Whitening loss offers a theoretical guarantee against feature collapse in self-supervised learning (SSL) with joint embedding architectures. Typically, it involves a hard whitening approach, transforming the embedding and applying loss to the whitened output. In this work, we introduce Spectral Transformation (ST), a framework to modulate the spectrum of embedding and to seek for functions beyond whitening that can avoid dimensional collapse. We show that whitening is a special instance of ST by definition, and our empirical investigations unveil other ST instances capable of preventing collapse. Additionally, we propose a novel ST instance named IterNorm with trace loss (INTL). Theoretical analysis confirms INTL's efficacy in preventing collapse and modulating the spectrum of embedding toward equal-eigenvalues during optimization. Our experiments on ImageNet classification and COCO object detection demonstrate INTL's potential in learning superior representations. The code is available at https://github.com/winci-ai/INTL.
Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Builds on 44 of its 52 references that are in the map. First posted on arXiv 26 May 2023.
Builds on
44 of 52 references.
References · 44
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 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
32 more
- 2020Improved Baselines with Momentum Contrastive Learningcited by 2,287
- 2018Unsupervised Feature Learning via Non-Parametric Instance-level Discriminationcited by 1,845
- 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,838
- 2019Contrastive Multiview Codingcited by 1,301
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,057
- 2019Learning Representations by Maximizing Mutual Information Across Viewscited by 734
- 2019Data-Efficient Image Recognition with Contrastive Predictive Codingcited by 673
- 2020Prototypical Contrastive Learning of Unsupervised Representationscited by 640
- 2020A Survey on Contrastive Self-supervised Learningcited by 554
- 2019A Theoretical Analysis of Contrastive Unsupervised Representation Learningcited by 530
- 2021Understanding Dimensional Collapse in Contrastive Self-supervised Learningcited by 378
- 2021Graph Self-Supervised Learning: A Surveycited by 289
- 2020Whitening for Self-Supervised Representation Learningcited by 279
- 2021Understanding self-supervised Learning Dynamics without Contrastive Pairscited by 256
- 2019Unsupervised Embedding Learning via Invariant and Spreading Instance Featurecited by 232
- 2021On Feature Decorrelation in Self-Supervised Learningcited by 160
- 2021From Canonical Correlation Analysis to Self-supervised Graph Neural Networkscited by 141
- 2021Solo-learn: A Library of Self-supervised Methods for Visual Representation Learningcited by 116
- 2022On the duality between contrastive and non-contrastive self-supervised learningcited by 108
- 2021Contrastive Learning with Stronger Augmentationscited by 100
- 2020BYOL works even without batch statisticscited by 95
- 2019Iterative Normalization: Beyond Standardization towards Efficient Whiteningcited by 92
- 2018Decorrelated Batch Normalizationcited by 83
- 2022How Does SimSiam Avoid Collapse Without Negative Samples? A Unified Understanding with Self-supervised Contrastive Learningcited by 65
- 2021Self-supervised Video Transformercited by 64
- 2020Understanding Self-supervised Learning with Dual Deep Networkscited by 62
- 2021Exploring the Equivalence of Siamese Self-Supervised Learning via A Unified Gradient Frameworkcited by 49
- 2022Spectral Feature Augmentation for Graph Contrastive Learning and Beyondcited by 49
- 2018Whitening and Coloring batch transform for GANscited by 27
- 2022An Investigation into Whitening Loss for Self-supervised Learningcited by 21
- 2022Investigating Power laws in Deep Representation Learningcited by 17
- 2019Network Deconvolutioncited by 9
Built on by
Citing papers per year: 2024: 4; 2025: 4; 2026: 1.
9 papers.
- 2024How JEPA Avoids Noisy Features: The Implicit Bias of Deep Linear Self Distillation Networkscited by 31
- 2024Preventing Dimensional Collapse in Self-Supervised Learning via Orthogonality Regularizationcited by 9
- 2025Clustering Properties of Self-Supervised Learningcited by 8
- 2024Balanced Data, Imbalanced Spectra: Unveiling Class Disparities with Spectral Imbalancecited by 2
- 2025T-REGS: Minimum Spanning Tree Regularization for Self-Supervised Learningcited by 2
- 2025Provable Affine Identifiability of Nonlinear CCA under Latent Distributional Priorscited by 1
- 2025IBNorm: Information-Bottleneck Inspired Normalization for Representation Learningcited by 1
- 2024Uncovering RL Integration in SSL Loss: Objective-Specific Implications for Data-Efficient RLcited by 0
- 2026ITC-MoE: Importance-guided Token-aware Compression for MoE Diffusion Language Modelscited by 0
- 2026ITC-MoE: Importance-guided Token-aware Compression for MoE Diffusion Language Models1 Oct 2026
- 2025IBNorm: Information-Bottleneck Inspired Normalization for Representation Learning29 Oct 2025
- 2025T-REGS: Minimum Spanning Tree Regularization for Self-Supervised Learning27 Oct 2025
- 2025Provable Affine Identifiability of Nonlinear CCA under Latent Distributional Priors6 Oct 2025
- 2025Clustering Properties of Self-Supervised Learning30 Jan 2025
- 2024Preventing Dimensional Collapse in Self-Supervised Learning via Orthogonality Regularization1 Nov 2024
- 2024Uncovering RL Integration in SSL Loss: Objective-Specific Implications for Data-Efficient RL22 Oct 2024
- 2024How JEPA Avoids Noisy Features: The Implicit Bias of Deep Linear Self Distillation Networks3 Jul 2024
- 2024Balanced Data, Imbalanced Spectra: Unveiling Class Disparities with Spectral Imbalance18 Feb 2024