arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2303.15103cs.LGcs.AIcs.CV

对比学习是相似度图上的谱聚类

Contrastive Learning Is Spectral Clustering On Similarity Graph

Tsinghua University(清华大学);Shanghai Artificial Intelligence Laboratory(上海人工智能实验室);Shanghai Qizhi Institute(上海期智研究院)

Zhiquan Tan, Yifan Zhang, Jingqin Yang, Yang Yuan

AI总结:

本文证明基于 InfoNCE 损失的对比学习等价于相似度图上的谱聚类,并将该理论扩展到 CLIP 模型,进而提出 Kernel-InfoNCE 损失以提升视觉任务表现。

摘要:

对比学习是一种强大的自监督学习方法,但我们对它的工作机制及其有效原因的理论理解仍然有限。在本文中,我们证明了使用标准 InfoNCE 损失的对比学习等价于在相似度图上进行谱聚类。以这一等价关系为构建基础,我们将分析扩展到 CLIP 模型,并严格刻画了相似的多模态对象是如何被嵌入到一起的。受我们理论洞见的启发,我们提出了 Kernel-InfoNCE 损失,该损失融合了多种核函数,并在多个视觉数据集上优于标准高斯核。代码可在 https://github.com/yifanzhang-pro/Kernel-InfoNCE 获取。

英文摘要:

Contrastive learning is a powerful self-supervised learning method, but we have a limited theoretical understanding of how it works and why it works. In this paper, we prove that contrastive learning with the standard InfoNCE loss is equivalent to spectral clustering on the similarity graph. Using this equivalence as the building block, we extend our analysis to the CLIP model and rigorously characterize how similar multi-modal objects are embedded together. Motivated by our theoretical insights, we introduce the Kernel-InfoNCE loss, incorporating mixtures of kernel functions that outperform the standard Gaussian kernel on several vision datasets. The code is available at https://github.com/yifanzhang-pro/Kernel-InfoNCE.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑