Hyperbolic Image-Text Representations

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Desai, Karan, Nickel, Maximilian, Rajpurohit, Tanmay, Johnson, Justin, Vedantam, Ramakrishna
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929213515235328
author Desai, Karan
Nickel, Maximilian
Rajpurohit, Tanmay
Johnson, Justin
Vedantam, Ramakrishna
author_facet Desai, Karan
Nickel, Maximilian
Rajpurohit, Tanmay
Johnson, Justin
Vedantam, Ramakrishna
contents Visual and linguistic concepts naturally organize themselves in a hierarchy, where a textual concept "dog" entails all images that contain dogs. Despite being intuitive, current large-scale vision and language models such as CLIP do not explicitly capture such hierarchy. We propose MERU, a contrastive model that yields hyperbolic representations of images and text. Hyperbolic spaces have suitable geometric properties to embed tree-like data, so MERU can better capture the underlying hierarchy in image-text datasets. Our results show that MERU learns a highly interpretable and structured representation space while being competitive with CLIP's performance on standard multi-modal tasks like image classification and image-text retrieval. Our code and models are available at https://www.github.com/facebookresearch/meru
format Preprint
id arxiv_https___arxiv_org_abs_2304_09172
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Hyperbolic Image-Text Representations
Desai, Karan
Nickel, Maximilian
Rajpurohit, Tanmay
Johnson, Justin
Vedantam, Ramakrishna
Computer Vision and Pattern Recognition
Machine Learning
Visual and linguistic concepts naturally organize themselves in a hierarchy, where a textual concept "dog" entails all images that contain dogs. Despite being intuitive, current large-scale vision and language models such as CLIP do not explicitly capture such hierarchy. We propose MERU, a contrastive model that yields hyperbolic representations of images and text. Hyperbolic spaces have suitable geometric properties to embed tree-like data, so MERU can better capture the underlying hierarchy in image-text datasets. Our results show that MERU learns a highly interpretable and structured representation space while being competitive with CLIP's performance on standard multi-modal tasks like image classification and image-text retrieval. Our code and models are available at https://www.github.com/facebookresearch/meru
title Hyperbolic Image-Text Representations
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2304.09172