Towards Compositionality in Concept Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Stein, Adam, Naik, Aaditya, Wu, Yinjun, Naik, Mayur, Wong, Eric
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909231929622528
author Stein, Adam
Naik, Aaditya
Wu, Yinjun
Naik, Mayur
Wong, Eric
author_facet Stein, Adam
Naik, Aaditya
Wu, Yinjun
Naik, Mayur
Wong, Eric
contents Concept-based interpretability methods offer a lens into the internals of foundation models by decomposing their embeddings into high-level concepts. These concept representations are most useful when they are compositional, meaning that the individual concepts compose to explain the full sample. We show that existing unsupervised concept extraction methods find concepts which are not compositional. To automatically discover compositional concept representations, we identify two salient properties of such representations, and propose Compositional Concept Extraction (CCE) for finding concepts which obey these properties. We evaluate CCE on five different datasets over image and text data. Our evaluation shows that CCE finds more compositional concept representations than baselines and yields better accuracy on four downstream classification tasks. Code and data are available at https://github.com/adaminsky/compositional_concepts .
format Preprint
id arxiv_https___arxiv_org_abs_2406_18534
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Towards Compositionality in Concept Learning
Stein, Adam
Naik, Aaditya
Wu, Yinjun
Naik, Mayur
Wong, Eric
Computation and Language
Machine Learning
Concept-based interpretability methods offer a lens into the internals of foundation models by decomposing their embeddings into high-level concepts. These concept representations are most useful when they are compositional, meaning that the individual concepts compose to explain the full sample. We show that existing unsupervised concept extraction methods find concepts which are not compositional. To automatically discover compositional concept representations, we identify two salient properties of such representations, and propose Compositional Concept Extraction (CCE) for finding concepts which obey these properties. We evaluate CCE on five different datasets over image and text data. Our evaluation shows that CCE finds more compositional concept representations than baselines and yields better accuracy on four downstream classification tasks. Code and data are available at https://github.com/adaminsky/compositional_concepts .
title Towards Compositionality in Concept Learning
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2406.18534