arXiv · 2406.18534
Towards Compositionality in Concept Learning
Abstract
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 .
Explore related subjects
Keep this discovery
Adam Stein, Aaditya Naik, Yinjun Wu, Mayur Naik, Eric Wong. 2024-06-26. Towards Compositionality in Concept Learning. https://arxiv.org/abs/2406.18534
Cite the original work for its findings. Save a collection to share your selection of sources.