Diversity Is All You Need for Contrastive Learning: Spectral Bounds on Gradient Magnitudes
Not on the map. Why?
Abstract (arXiv)
We derive non-asymptotic spectral bands that bound the squared InfoNCE gradient norm via alignment, temperature, and batch spectrum, recovering the (1/τ2) law and closely tracking batch-mean gradients on synthetic data and ImageNet. Using effective rank (Reff) as an anisotropy proxy, we design spectrum-aware batch selection, including a fast greedy builder. On ImageNet-100, Greedy-64 cuts time-to-67.5% top-1 by 15% vs. random (24% vs. Pool--P3) at equal accuracy; CIFAR-10 shows similar gains. In-batch whitening promotes isotropy and reduces 50-step gradient variance by (1.37×), matching our theoretical upper bound.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 18 of its 28 references that are in the map. First posted on arXiv 7 Oct 2025.
Builds on
18 of 28 references.
References · 18
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,750
- 2020Exploring Simple Siamese Representation Learningcited by 2,866
- 2020Supervised Contrastive Learningcited by 2,829
- 2020Improved Baselines with Momentum Contrastive Learningcited by 2,287
- 2020Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hyperspherecited by 1,405
- 2020Contrastive Learning with Hard Negative Samplescited by 628
- 2020Hard Negative Mixing for Contrastive Learningcited by 402
- 2015An Introduction to Matrix Concentration Inequalitiescited by 397
- 2020Debiased Contrastive Learningcited by 356
6 more
- 2017Sampling Matters in Deep Embedding Learningcited by 285
- 2020Whitening for Self-Supervised Representation Learningcited by 279
- 2021Contrastive Learning Inverts the Data Generating Processcited by 221
- 2021On Feature Decorrelation in Self-Supervised Learningcited by 160
- 2022RankMe: Assessing the downstream performance of pretrained self-supervised representations by their rankcited by 153
- 2017Smart Mining for Deep Metric Learningcited by 96
Built on by
Citing papers per year: 2026: 2.
2 papers.