将对比学习重新理解为分布对齐问题,提升表示学习效果
Your contrastive learning problem is secretly a distribution alignment problem
- 将对比学习与熵正则最优传输结合,构建新型损失函数
- 通过分布感知调整增强样本关系,提升泛化能力
- 适合研究自监督学习机制及需引入领域知识的场景
尽管对比学习在视觉和语言任务中取得成功,但其理论基础与表征构建机制仍不清晰。本文揭示了广泛使用的噪声对比估计损失与熵正则最优传输下的分布对齐之间的联系,由此提出一类新的损失函数及多步迭代变体。该方法利用潜在表示的分布信息,实现更精细的样本间关系调控。理论分析与实验证明,所提方法在广义对比对齐上具有优势。通过将对比学习重构为对齐问题,并借助最优传输优化工具,可设计非平衡损失以应对噪声视图,或通过调整对齐约束定制表示空间。本工作为自监督学习模型提供了新视角,并支持更灵活地融入领域知识。
原文摘要 · Abstract (English)
Despite the success of contrastive learning (CL) in vision and language, its theoretical foundations and mechanisms for building representations remain poorly understood. In this work, we build connections between noise contrastive estimation losses widely used in CL and distribution alignment with entropic optimal transport (OT). This connection allows us to develop a family of different losses and multistep iterative variants for existing CL methods. Intuitively, by using more information from the distribution of latents, our approach allows a more distribution-aware manipulation of the relationships within augmented sample sets. We provide theoretical insights and experimental evidence demonstrating the benefits of our approach for {\em generalized contrastive alignment}. Through this framework, it is possible to leverage tools in OT to build unbalanced losses to handle noisy views and customize the representation space by changing the constraints on alignment. By reframing contrastive learning as an alignment problem and leveraging existing optimization tools for OT, our work provides new insights and connections between different self-supervised learning models in addition to new tools that can be more easily adapted to incorporate domain knowledge into learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。