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

对比学习视觉-语言模型的测试时分布归一化

Test-Time Distribution Normalization for Contrastively Learned Vision-language Models

University of California, Berkeley(加州大学伯克利分校);Cornell University(康奈尔大学);University of Central Florida(中佛罗里达大学)

Yifei Zhou, Juntao Ren, Fengyu Li, Ramin Zabih, Ser-Nam Lim

AI总结:

本文指出CLIP点积推理仅为零阶近似导致信息丢失,提出无需重训的分布归一化方法,用测试批次均值近似负样本以对齐InfoNCE优化目标,在多种下游任务上显著优于点积。

摘要:

视觉-语言对比学习领域的进展使得许多下游应用能够通过简单计算图像与文本表示之间的点积来高效、准确地执行。最近提出的最具代表性的方法之一CLIP因其有效性而获得了广泛采用。CLIP使用InfoNCE损失进行训练,该损失同时考虑正样本和负样本,以帮助学习更鲁棒的表示空间。本文揭示了取点积这一常见下游实践仅是优化目标的零阶近似,导致在测试时发生信息丢失。直观地说,既然模型已基于InfoNCE损失进行优化,测试时流程也应与之对齐。问题在于如何在推理期间以计算高效的方式检索任何类似于负样本的信息。为此,我们提出分布归一化(DN),其中我们近似一批测试样本的平均表示,并使用该均值来表示InfoNCE损失中类似于负样本的内容。DN无需重新训练或微调,可在推理期间轻松应用。在广泛多样的下游任务上的大量实验表明,DN在其他现有测试时增强方法之上相比点积具有明显优势。

英文摘要:

Advances in the field of vision-language contrastive learning have made it possible for many downstream applications to be carried out efficiently and accurately by simply taking the dot product between image and text representations. One of the most representative approaches proposed recently known as CLIP has garnered widespread adoption due to its effectiveness. CLIP is trained with an InfoNCE loss that takes into account both positive and negative samples to help learn a much more robust representation space. This paper reveals that the common downstream practice of taking a dot product is only a zeroth-order approximation of the optimization goal, resulting in a loss of information during test-time. Intuitively, since the model has been optimized based on the InfoNCE loss, test-time procedures should also be in alignment. The question lies in how one can retrieve any semblance of negative samples information during inference in a computationally efficient way. To this end, we propose Distribution Normalization (DN), where we approximate the mean representation of a batch of test samples and use such a mean to represent what would be analogous to negative samples in the InfoNCE loss. DN requires no retraining or fine-tuning and can be effortlessly applied during inference. Extensive experiments on a wide variety of downstream tasks exhibit a clear advantage of DN over the dot product on top of other existing test-time augmentation methods.

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

↑