I0T: Embedding Standardization Method Towards Zero Modality Gap

Code

• github.com/xfactlab/i0t
  OfficialIn paperpytorch★ 12

Abstract

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, I0T_post that reduces the modality gap approximately to zero and (2) a trainable method, I0T_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, I0T_post can serve as an alternative explainable automatic evaluation metric of widely used CLIPScore (CLIP-S).

Tasks

Reproductions