通过稀疏权重分解,无需训练即可从预训练模型中提取可解释的电路单元。
Sparse Weight Decomposition for Efficient Circuit Extraction

- 将权重矩阵分解为两个稀疏因子,共享中间坐标作为可单独定位的电路单元。
- 仅用不到1%的数据即可达到与主流方法相当的模型保真度,且激活边和单元更少。
- 支持零数据版本,适合每步分析等精细化可解释性研究,适用多模型架构。
预训练的稠密Transformer模型本身不自然地暴露可解释的单元用于电路提取。现有方法需学习辅助稀疏表示或训练稀疏模型,带来大量额外计算并可能引入表示与原模型之间的保真度差距。我们提出稀疏权重分解(SWD),通过将预训练线性投影重新参数化,将每个权重矩阵分解为两个稀疏因子,其共享的中间坐标作为可独立定位的电路单元。无需训练替代网络,该参数化形式支持与学习稀疏特征的方法相同的评分、选择和消融电路提取流程。在单矩阵替换中,SWD在仅使用不到1%的数据下,达到了与Transcoder等强基线相当的保留保真度。在匹配替换保真度的前提下,SWD在GPT-2、Qwen2.5和Qwen3.5-27B上以更少的活跃读写边和选中的单元达到相同的电路充分性和必要性目标。进一步证明,微调非零因子值后,SWD仍能有效实现全部注意力与MLP权重矩阵的全模型替换。此外,SWD还具备零数据变体,支持更广泛的机械可解释性分析(如每步分析)。
原文摘要 · Abstract (English)
Dense pretrained transformers do not naturally expose interpretable units for circuit extraction. Existing approaches obtain such units by learning auxiliary sparse representations or training sparse models, incurring substantial additional computation while potentially introducing a fidelity gap between the representation being analyzed and the original pretrained model. We propose Sparse Weight Decomposition (SWD), which reparameterizes pretrained linear projections by factorizing each weight matrix into two sparse factors whose shared intermediate coordinates serve as individually addressable circuit units. Without training a separate replacement network, this parametric representation supports the same scoring, selection, and ablation circuit extraction workflow used for methods that learn sparse features. Across single-matrix replacements, SWD matches the held-out fidelity achieved by Transcoder and other strong baselines while using less than 1% of the data that those baselines use to train their replacements. For matched replacement fidelity, SWD reaches the same circuit sufficiency and necessity targets with fewer active read/write edges and selected units across tasks on GPT-2, Qwen2.5, and Qwen3.5-27B. We further show that SWD remains effective for full-model replacement of all attention and MLP weight matrices after fine-tuning the nonzero factor values. Finally, SWD also features a zero-data variant, allowing broader use of mechanistic interpretability analysis (e.g., per-step analysis).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。