A framework for analyzing concept representations in neural models
This work provides a systematic framework for evaluating concept representations in neural models, benefiting researchers studying interpretability and fairness.
The paper introduces a unified framework for analyzing concept representations in neural models along two axes: containment and disentanglement. Experiments on text and speech models show that concept subspaces are not uniquely determined, and that the choice of estimator impacts these properties, with LEACE performing well but struggling to generalize to unseen data.
Understanding how neural models represent human-interpretable concepts is challenging. Prior work has explored linear concept subspaces from diverse perspectives, such as probing and concept erasure. We introduce a unified framework to study these subspaces along two axes: \textit{containment}, which tests if a concept is fully represented in a subspace but not outside it, and \textit{disentanglement}, which tests for isolation from other concepts. In experiments on both text and speech models, we first highlight that concept subspaces may not be uniquely determined, and discuss the implications for concept subspace analysis. Then, we compare properties of concept subspaces estimated using five estimators, proposed in different communities. We find that (1) the choice of estimator impacts the containment and disentanglement properties; (2) the state-of-the-art concept erasure method, LEACE, performs well on both testing axes, but still struggles to generalize to unseen data; and (3) in HuBERT speech representations, phone information is both contained and disentangled from speaker information, while speaker information is hard to contain in a compact subspace, despite being disentangled from phones.