解决CLIP模型中图文嵌入差异问题,实现零模态差距。
I0T: Embedding Standardization Method Towards Zero Modality Gap
- 提出后处理标准化方法I0T_post,消除图文嵌入差异。
- 在ImageNet分类等任务上,零样本性能提升1.5~3.2个百分点。
- 适合需要高精度图文对齐的应用场景,如跨模态检索。
对比语言-图像预训练(CLIP)支持下游任务的零样本推理,如图像-文本检索和分类。然而,近期扩展CLIP的方法存在模态差距问题,即图像与文本嵌入被投影到不同的流形,偏离了图像-文本对比学习的初衷。我们发现该现象源于图像/文本编码器各自特有的模态特性,并提出两种解决方案:(1) 后处理嵌入标准化方法 I0T_post,可将近似消除模态差距;(2) 可训练方法 I0T_async,通过为每个编码器添加两个归一化层缓解该问题。I0T框架能显著降低模态差距,同时保留原模型冻结参数下的嵌入表示。实践中,I0T_post可作为广泛使用的CLIPScore(CLIP-S)的可解释自动评估替代指标。
原文摘要 · Abstract (English)
Contrastive Language-Image Pretraining (CLIP) enables zero-shot inference in downstream tasks such as image-text retrieval and classification. However, recent works extending CLIP suffer from the issue of modality gap, which arises when the image and text embeddings are projected to disparate manifolds, deviating from the intended objective of image-text contrastive learning. We discover that this phenomenon is linked to the modality-specific characteristic that each image/text encoder independently possesses and propose two methods to address the modality gap: (1) a post-hoc embedding standardization method, $\text{I0T}_{\text{post}}$ that reduces the modality gap approximately to zero and (2) a trainable method, $\text{I0T}_{\text{async}}$, to alleviate the modality gap problem by adding two normalization layers for each encoder. Our I0T framework can significantly reduce the modality gap while preserving the original embedding representations of trained models with their locked parameters. In practice, $\text{I0T}_{\text{post}}$ can serve as an alternative explainable automatic evaluation metric of widely used CLIPScore (CLIP-S).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。