Interpreting CLIP's Image Representation via Text-Based Decomposition

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Gandelsman, Yossi, Efros, Alexei A., Steinhardt, Jacob
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910389489369088
author Gandelsman, Yossi
Efros, Alexei A.
Steinhardt, Jacob
author_facet Gandelsman, Yossi
Efros, Alexei A.
Steinhardt, Jacob
contents We investigate the CLIP image encoder by analyzing how individual model components affect the final representation. We decompose the image representation as a sum across individual image patches, model layers, and attention heads, and use CLIP's text representation to interpret the summands. Interpreting the attention heads, we characterize each head's role by automatically finding text representations that span its output space, which reveals property-specific roles for many heads (e.g. location or shape). Next, interpreting the image patches, we uncover an emergent spatial localization within CLIP. Finally, we use this understanding to remove spurious features from CLIP and to create a strong zero-shot image segmenter. Our results indicate that a scalable understanding of transformer models is attainable and can be used to repair and improve models.
format Preprint
id arxiv_https___arxiv_org_abs_2310_05916
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Interpreting CLIP's Image Representation via Text-Based Decomposition
Gandelsman, Yossi
Efros, Alexei A.
Steinhardt, Jacob
Computer Vision and Pattern Recognition
Artificial Intelligence
We investigate the CLIP image encoder by analyzing how individual model components affect the final representation. We decompose the image representation as a sum across individual image patches, model layers, and attention heads, and use CLIP's text representation to interpret the summands. Interpreting the attention heads, we characterize each head's role by automatically finding text representations that span its output space, which reveals property-specific roles for many heads (e.g. location or shape). Next, interpreting the image patches, we uncover an emergent spatial localization within CLIP. Finally, we use this understanding to remove spurious features from CLIP and to create a strong zero-shot image segmenter. Our results indicate that a scalable understanding of transformer models is attainable and can be used to repair and improve models.
title Interpreting CLIP's Image Representation via Text-Based Decomposition
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2310.05916