中文

基于梯度范数谱界的对比学习:你需要多样性

计算与语言 2025-10-08 v1

摘要

我们推导非渐近谱带,界定 InfoNCE 梯度范数的平方通过对齐、温度和批次谱,恢复 1/τ21/\tau^{2} 定律,并在合成数据和 ImageNet 上紧密跟踪批均值梯度。使用有效秩 ReffR_{\mathrm{eff}} 作为各向异性代理,设计谱感知批选择,包括快速贪心构建器。在 ImageNet-100 上,Greedy-64 将 top-1 达到 67.5% 所需时间缩短 15%(相较随机模式下为 24%,相较 Pool--P3 下为 24%),在等效准确率下实现此目标;CIFAR-10 显示类似收益。在批内白化化中,促进各向同性,使 50 步梯度方差降低 1.37×1.37\times,匹配我们的理论上界。

关键词

引用

@article{arxiv.2510.05767,
  title  = {Diversity Is All You Need for Contrastive Learning: Spectral Bounds on Gradient Magnitudes},
  author = {Peter Ochieng},
  journal= {arXiv preprint arXiv:2510.05767},
  year   = {2025}
}