arXiv:2410.09156cs.LGstat.ML2024-10ICLR被引 3

提出新方法提升自监督表征学习的生成能力。

On Discriminative Probabilistic Modeling for Self-Supervised Representation Learning

论文配图:On Discriminative Probabilistic Modeling for Self-Supervised Representation Learning
图 1 · 摘自论文原文
  • 用重要性采样重构对比损失,统一建模框架。
  • 在CC3M和CC12M上优于主流基线模型。
  • 适合研究自监督学习与概率建模交叉方向者。

本文研究连续域上的判别式概率建模,用于多模态自监督表示学习中的数据预测任务。为解决锚点数据对应配分函数积分难以计算的问题,引入多重重要性采样(MIS)实现鲁棒的蒙特卡洛积分,可还原InfoNCE对比损失作为特例。在该概率建模框架下,通过泛化误差分析揭示现有InfoNCE方法的局限性,并提出通过降低蒙特卡洛积分误差来改进模型的方法。为此,我们设计一种基于凸优化的非参数方法,近似MIS所需的条件概率密度之和,得到新的对比目标。同时开发高效算法求解该目标。在图像-文本对比预训练任务上,于CC3M与CC12M数据集上进行实验,结果表明所提方法整体性能更优。代码已开源:https://github.com/bokun-wang/NUCLR。

原文摘要 · Abstract (English)

We study the discriminative probabilistic modeling on a continuous domain for the data prediction task of (multimodal) self-supervised representation learning. To address the challenge of computing the integral in the partition function for each anchor data, we leverage the multiple importance sampling (MIS) technique for robust Monte Carlo integration, which can recover InfoNCE-based contrastive loss as a special case. Within this probabilistic modeling framework, we conduct generalization error analysis to reveal the limitation of current InfoNCE-based contrastive loss for self-supervised representation learning and derive insights for developing better approaches by reducing the error of Monte Carlo integration. To this end, we propose a novel non-parametric method for approximating the sum of conditional probability densities required by MIS through convex optimization, yielding a new contrastive objective for self-supervised representation learning. Moreover, we design an efficient algorithm for solving the proposed objective. We empirically compare our algorithm to representative baselines on the contrastive image-language pretraining task. Experimental results on the CC3M and CC12M datasets demonstrate the superior overall performance of our algorithm. Our code is available at https://github.com/bokun-wang/NUCLR.

自监督对比学习概率建模优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。