arXiv:2603.28744cs.LG2026-03被引 5

稀疏自编码器在组合泛化上失败,根源是词典学习不佳而非推理效率问题。

Stop Probing, Start Coding: Why Linear Probes and Sparse Autoencoders Fail at Compositional Generalisation

  • 用逐样本优化替代固定编码器,提升稀疏推断精度
  • 即使换用更优推理方式,词典方向错误仍导致性能下降
  • 词典学习质量是决定性因素,适用于需要可解释性的场景

线性表征假说认为神经网络激活以线性组合形式编码高层概念。但在超叠加(superposition)下,这种编码是从高维概念空间到低维激活空间的投影,概念空间中的线性决策边界在投影后未必保持线性。经典稀疏编码方法通过每样本迭代推断利用压缩感知保证恢复潜在因子。而稀疏自编码器(SAEs)将稀疏推断固化为固定编码器,引入系统性差距。我们发现该差距在不同训练集规模、潜在维度和稀疏度下均持续存在,导致SAE在分布外(OOD)组合变化下失效。通过受控实验分解失败原因,我们识别出词典学习才是关键瓶颈:SAE学到的词典方向严重偏离,即便将编码器替换为同一词典下的逐样本FISTA,差距依然无法弥合。一个理想基线证明,在所有测试尺度下,良好词典足以解决该问题。结果表明,SAE失败本质是词典学习问题,而非近似推断问题,提示可扩展的词典学习是超叠加下稀疏推断的核心开放挑战。

原文摘要 · Abstract (English)

The linear representation hypothesis states that neural network activations encode high-level concepts as linear mixtures. However, under superposition, this encoding is a projection from a higher-dimensional concept space into a lower-dimensional activation space, and a linear decision boundary in the concept space need not remain linear after projection. In this setting, classical sparse coding methods with per-sample iterative inference leverage compressed sensing guarantees to recover latent factors. Sparse autoencoders (SAEs), on the other hand, amortise sparse inference into a fixed encoder, introducing a systematic gap. We show this amortisation gap persists across training set sizes, latent dimensions, and sparsity levels, causing SAEs to fail under out-of-distribution (OOD) compositional shifts. Through controlled experiments that decompose the failure, we identify dictionary learning -- not the inference procedure -- as the binding constraint: SAE-learned dictionaries point in substantially wrong directions, and replacing the encoder with per-sample FISTA on the same dictionary does not close the gap. An oracle baseline proves the problem is solvable with a good dictionary at all scales tested. Our results reframe the SAE failure as a dictionary learning challenge, not an amortisation problem, and point to scalable dictionary learning as the key open problem for sparse inference under superposition.

稀疏编码词典学习组合泛化神经表征

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