ComAlign: Compositional Alignment in Vision-Language Models

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Abdollah, Ali, Izadi, Amirmohammad, Saghafian, Armin, Vahidimajd, Reza, Mozafari, Mohammad, Mirzaei, Amirreza, Samiei, Mohammadmahdi, Baghshah, Mahdieh Soleymani
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914948130537472
author Abdollah, Ali
Izadi, Amirmohammad
Saghafian, Armin
Vahidimajd, Reza
Mozafari, Mohammad
Mirzaei, Amirreza
Samiei, Mohammadmahdi
Baghshah, Mahdieh Soleymani
author_facet Abdollah, Ali
Izadi, Amirmohammad
Saghafian, Armin
Vahidimajd, Reza
Mozafari, Mohammad
Mirzaei, Amirreza
Samiei, Mohammadmahdi
Baghshah, Mahdieh Soleymani
contents Vision-language models (VLMs) like CLIP have showcased a remarkable ability to extract transferable features for downstream tasks. Nonetheless, the training process of these models is usually based on a coarse-grained contrastive loss between the global embedding of images and texts which may lose the compositional structure of these modalities. Many recent studies have shown VLMs lack compositional understandings like attribute binding and identifying object relationships. Although some recent methods have tried to achieve finer-level alignments, they either are not based on extracting meaningful components of proper granularity or don't properly utilize the modalities' correspondence (especially in image-text pairs with more ingredients). Addressing these limitations, we introduce Compositional Alignment (ComAlign), a fine-grained approach to discover more exact correspondence of text and image components using only the weak supervision in the form of image-text pairs. Our methodology emphasizes that the compositional structure (including entities and relations) extracted from the text modality must also be retained in the image modality. To enforce correspondence of fine-grained concepts in image and text modalities, we train a lightweight network lying on top of existing visual and language encoders using a small dataset. The network is trained to align nodes and edges of the structure across the modalities. Experimental results on various VLMs and datasets demonstrate significant improvements in retrieval and compositional benchmarks, affirming the effectiveness of our plugin model.
format Preprint
id arxiv_https___arxiv_org_abs_2409_08206
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ComAlign: Compositional Alignment in Vision-Language Models
Abdollah, Ali
Izadi, Amirmohammad
Saghafian, Armin
Vahidimajd, Reza
Mozafari, Mohammad
Mirzaei, Amirreza
Samiei, Mohammadmahdi
Baghshah, Mahdieh Soleymani
Computer Vision and Pattern Recognition
Multimedia
Vision-language models (VLMs) like CLIP have showcased a remarkable ability to extract transferable features for downstream tasks. Nonetheless, the training process of these models is usually based on a coarse-grained contrastive loss between the global embedding of images and texts which may lose the compositional structure of these modalities. Many recent studies have shown VLMs lack compositional understandings like attribute binding and identifying object relationships. Although some recent methods have tried to achieve finer-level alignments, they either are not based on extracting meaningful components of proper granularity or don't properly utilize the modalities' correspondence (especially in image-text pairs with more ingredients). Addressing these limitations, we introduce Compositional Alignment (ComAlign), a fine-grained approach to discover more exact correspondence of text and image components using only the weak supervision in the form of image-text pairs. Our methodology emphasizes that the compositional structure (including entities and relations) extracted from the text modality must also be retained in the image modality. To enforce correspondence of fine-grained concepts in image and text modalities, we train a lightweight network lying on top of existing visual and language encoders using a small dataset. The network is trained to align nodes and edges of the structure across the modalities. Experimental results on various VLMs and datasets demonstrate significant improvements in retrieval and compositional benchmarks, affirming the effectiveness of our plugin model.
title ComAlign: Compositional Alignment in Vision-Language Models
topic Computer Vision and Pattern Recognition
Multimedia
url https://arxiv.org/abs/2409.08206