arXiv · 2304.09707
Disentangling Neuron Representations with Concept Vectors
Abstract
Mechanistic interpretability aims to understand how models store representations by breaking down neural networks into interpretable units. However, the occurrence of polysemantic neurons, or neurons that respond to multiple unrelated features, makes interpreting individual neurons challenging. This has led to the search for meaningful vectors, known as concept vectors, in activation space instead of individual neurons. The main contribution of this paper is a method to disentangle polysemantic neurons into concept vectors encapsulating distinct features. Our method can search for fine-grained concepts according to the user's desired level of concept separation. The analysis shows that polysemantic neurons can be disentangled into directions consisting of linear combinations of neurons. Our evaluations show that the concept vectors found encode coherent, human-understandable features.
Explore related subjects
Keep this discovery
Laura O'Mahony, Vincent Andrearczyk, Henning Muller, Mara Graziani. 2023-04-19. Disentangling Neuron Representations with Concept Vectors. https://arxiv.org/abs/2304.09707
Cite the original work for its findings. Save a collection to share your selection of sources.