不用训练字典,用ICA直接发现语言模型中的可解释方向。
ICA Lens: Interpreting Language Models Without Training Another Dictionary

- 用优化的FastICA算法提取语言模型激活中非高斯方向,无需额外训练。
- 在GPT-2、Gemma和Qwen上成功恢复出人类可读的语义方向。
- 比传统方法更快更稳定,适合快速探索模型内部机制。
理解与控制大语言模型行为的关键在于发现其表示空间中的可解释方向。稀疏自编码器(SAEs)是当前主流工具,但通常需训练、存储和评估大型过完备字典,带来效率瓶颈。我们提出:许多可解释方向在词元上具有选择性,其分布应显著偏离随机方向的高斯特性。因此,我们重新审视独立成分分析(ICA)——一种经典非高斯方向检测方法——作为紧凑的可解释性探针。此前对ICA的应用因依赖脆弱的现成实现且缺乏系统评估工具而被低估。为此,我们提出ICALens,首个面向大模型的稳定、高效、可审计的ICA分析工作流。它结合了GPU并行化的FastICA管道、针对大模型激活的稳定性策略及更优的拟合诊断,支持层级化高效分析。在GPT-2 Small、Gemma 2 2B和Qwen 3.5 2B Base上,ICALens无需基于梯度的字典训练即可恢复紧凑、可解释的方向。在SAEBench评测中,其稀疏探针性能与公开SAE相当,在小至中等预算下的目标探针扰动任务中表现更优。结果表明,ICA不应被视为弱基线,而应作为探索大模型表示的高效互补首选工具。
原文摘要 · Abstract (English)
Finding interpretable directions in language-model representations is critical for understanding and controlling model behavior. Sparse autoencoders (SAEs) have become the standard tool for this purpose, but using them as the default first lens often requires training, storing, and evaluating large overcomplete dictionaries. This bottleneck limits rapid exploration and raises a fundamental question: how much interpretable structure is already visible from activation geometry before training another neural dictionary? Our intuition is simple: many interpretable directions are selective on tokens, and these directions should look less Gaussian than random directions. We therefore revisit independent component analysis (ICA), a classical method for finding non-Gaussian directions, as a compact lens for language-model interpretability. We find that ICA has been underestimated for LLM interpretability, because prior uses often relied on off-the-shelf ICA implementations that are brittle on LLM activations and lacked systematic tools for inspecting and evaluating the recovered directions. To bridge these gaps, we introduce ICALens, the first practical workflow for stable, efficient, and auditable ICA analysis of LLM representations. It combines an optimized GPU-parallel FastICA pipeline with LLM-specific stability recipes and better fitting diagnostics, enabling efficient and reliable layer-wise analysis. Across GPT-2 Small, Gemma 2 2B, and Qwen 3.5 2B Base, ICALens efficiently recovers compact, human-interpretable directions without per-layer gradient-based dictionary training. On SAEBench, ICA is competitive with public SAEs in sparse probing and outperforms them in targeted probe perturbation under small-to-medium budgets. These results suggest that ICA should not be viewed as a weak baseline, but as an efficient and complementary first lens for exploring language-model representations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。