arXiv:2511.07572cs.LG2025-11

提出新基准与结构,让小模型特征连接更稀疏清晰

SCALAR: Benchmarking SAE Interaction Sparsity in Toy LLMs

  • 设计新基准SCALAR,量化不同特征间连接稀疏性
  • 阶梯式SAE使连接稀疏度提升超60%,且适配全网络
  • 适合关注神经网络可解释性的研究者与工程师

机制可解释性旨在将神经网络分解为可理解的特征并映射其连接路径。标准方法在每层激活上独立训练稀疏自编码器(SAEs),但孤立训练无法促进跨层连接稀疏,导致上游特征无谓影响多个下游特征,夸大提取出的电路。现有评估聚焦单个SAE性能,忽视交互稀疏性。我们提出SCALAR(潜在激活关系稀疏连通性评估)基准,用于衡量SAE特征间的交互稀疏性。同时提出“阶梯式SAE”,通过权重共享限制上游特征在下游的重复使用。利用SCALAR,对比TopK SAE、雅可比SAE(JSAE)和阶梯式SAE。阶梯式SAE在前馈层和变换器块中分别相较TopK提升59.67%±1.83%和63.15%±1.35%的相对稀疏度;JSAE在前馈层仅提升8.54%±0.38%,且无法在变换器块有效训练。我们在216K参数小模型和GPT-2 Small(124M)上验证,阶梯式SAE在保持特征可解释性的前提下持续提升交互稀疏性。本工作通过基准测试凸显了交互稀疏性的重要性,并比较了多种有前景的架构。

原文摘要 · Abstract (English)

Mechanistic interpretability aims to decompose neural networks into interpretable features and map their connecting circuits. The standard approach trains sparse autoencoders (SAEs) on each layer's activations. However, SAEs trained in isolation don't encourage sparse cross-layer connections, inflating extracted circuits where upstream features needlessly affect multiple downstream features. Current evaluations focus on individual SAE performance, leaving interaction sparsity unexamined. We introduce SCALAR (Sparse Connectivity Assessment of Latent Activation Relationships), a benchmark measuring interaction sparsity between SAE features. We also propose "Staircase SAEs", using weight-sharing to limit upstream feature duplication across downstream features. Using SCALAR, we compare TopK SAEs, Jacobian SAEs (JSAEs), and Staircase SAEs. Staircase SAEs improve relative sparsity over TopK SAEs by $59.67\% \pm 1.83\%$ (feedforward) and $63.15\% \pm 1.35\%$ (transformer blocks). JSAEs provide $8.54\% \pm 0.38\%$ improvement over TopK for feedforward layers but cannot train effectively across transformer blocks, unlike Staircase and TopK SAEs which work anywhere in the residual stream. We validate on a $216$K-parameter toy model and GPT-$2$ Small ($124$M), where Staircase SAEs maintain interaction sparsity improvements while preserving feature interpretability. Our work highlights the importance of interaction sparsity in SAEs through benchmarking and comparing promising architectures.

可解释性稀疏编码模型压缩神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。