Weighted Point Set Embedding for Multimodal Contrastive Learning Toward Optimal Similarity Metric

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Uesaka, Toshimitsu, Suzuki, Taiji, Takida, Yuhta, Lai, Chieh-Hsin, Murata, Naoki, Mitsufuji, Yuki
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909518544240640
author Uesaka, Toshimitsu
Suzuki, Taiji
Takida, Yuhta
Lai, Chieh-Hsin
Murata, Naoki
Mitsufuji, Yuki
author_facet Uesaka, Toshimitsu
Suzuki, Taiji
Takida, Yuhta
Lai, Chieh-Hsin
Murata, Naoki
Mitsufuji, Yuki
contents In typical multimodal contrastive learning, such as CLIP, encoders produce one point in the latent representation space for each input. However, one-point representation has difficulty in capturing the relationship and the similarity structure of a huge amount of instances in the real world. For richer classes of the similarity, we propose the use of weighted point sets, namely, sets of pairs of weight and vector, as representations of instances. In this work, we theoretically show the benefit of our proposed method through a new understanding of the contrastive loss of CLIP, which we call symmetric InfoNCE. We clarify that the optimal similarity that minimizes symmetric InfoNCE is the pointwise mutual information, and show an upper bound of excess risk on downstream classification tasks of representations that achieve the optimal similarity. In addition, we show that our proposed similarity based on weighted point sets consistently achieves the optimal similarity. To verify the effectiveness of our proposed method, we demonstrate pretraining of text-image representation models and classification tasks on common benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2404_19228
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Weighted Point Set Embedding for Multimodal Contrastive Learning Toward Optimal Similarity Metric
Uesaka, Toshimitsu
Suzuki, Taiji
Takida, Yuhta
Lai, Chieh-Hsin
Murata, Naoki
Mitsufuji, Yuki
Machine Learning
In typical multimodal contrastive learning, such as CLIP, encoders produce one point in the latent representation space for each input. However, one-point representation has difficulty in capturing the relationship and the similarity structure of a huge amount of instances in the real world. For richer classes of the similarity, we propose the use of weighted point sets, namely, sets of pairs of weight and vector, as representations of instances. In this work, we theoretically show the benefit of our proposed method through a new understanding of the contrastive loss of CLIP, which we call symmetric InfoNCE. We clarify that the optimal similarity that minimizes symmetric InfoNCE is the pointwise mutual information, and show an upper bound of excess risk on downstream classification tasks of representations that achieve the optimal similarity. In addition, we show that our proposed similarity based on weighted point sets consistently achieves the optimal similarity. To verify the effectiveness of our proposed method, we demonstrate pretraining of text-image representation models and classification tasks on common benchmarks.
title Weighted Point Set Embedding for Multimodal Contrastive Learning Toward Optimal Similarity Metric
topic Machine Learning
url https://arxiv.org/abs/2404.19228