用神经元组合分解法,让大模型的隐藏特征更可解释。
Constructing Interpretable Features from Compositional Neuron Groups
- 通过半非负矩阵分解直接分析MLP激活值,提取稀疏线性组合的神经元组。
- 在Llama 3.1等模型上,因果操控效果优于SAE和监督基线方法。
- 发现语义相关特征共享特定神经元组合,揭示激活空间的层级结构。
大规模语言模型(LLM)的机制可解释性目标之一是找到能因果解释输出的合适分析单元。早期研究聚焦单个神经元,但神经元常编码多重概念,促使研究转向激活空间方向的分析。关键问题是如何无监督地发现具有可解释性的方向。现有方法依赖稀疏自编码器(SAEs)进行字典学习,通常在残差流激活上训练,但其在因果评估中表现不佳,且缺乏内在可解释性,因学习过程未与模型计算显式关联。本文提出直接对MLP激活值使用半非负矩阵分解(SNMF),使学习到的特征满足:(a) 是共激活神经元的稀疏线性组合;(b) 可映射至触发它们的输入,从而实现直接可解释。在Llama 3.1、Gemma 2和GPT-2上的实验表明,基于SNMF的特征在因果操控任务中优于SAEs和强监督基线(差异均值法),同时与人类可理解的概念一致。进一步分析显示,特定神经元组合在语义相关特征间重复使用,暴露了MLP激活空间中的层次结构。这些结果表明,SNMF是一种简单有效的工具,可用于识别可解释特征并解析大模型中的概念表示。
原文摘要 · Abstract (English)
A central goal for mechanistic interpretability has been to identify the right units of analysis in large language models (LLMs) that causally explain their outputs. While early work focused on individual neurons, evidence that neurons often encode multiple concepts has motivated a shift toward analyzing directions in activation space. A key question is how to find directions that capture interpretable features in an unsupervised manner. Current methods rely on dictionary learning with sparse autoencoders (SAEs), commonly trained over residual stream activations to learn directions from scratch. However, SAEs often struggle in causal evaluations and lack intrinsic interpretability, as their learning is not explicitly tied to the computations of the model. Here, we tackle these limitations by directly decomposing MLP activations with semi-nonnegative matrix factorization (SNMF), such that the learned features are (a) sparse linear combinations of co-activated neurons, and (b) mapped to their activating inputs, making them directly interpretable. Experiments on Llama 3.1, Gemma 2 and GPT-2 show that SNMF derived features outperform SAEs and a strong supervised baseline (difference-in-means) on causal steering, while aligning with human-interpretable concepts. Further analysis reveals that specific neuron combinations are reused across semantically-related features, exposing a hierarchical structure in the MLP's activation space. Together, these results position SNMF as a simple and effective tool for identifying interpretable features and dissecting concept representations in LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。