ICA Lens:不训字典,一眼看穿LLM在想什么
——被低估40年的信号处理老方法,在LLM可解释性上重新封神
一、引子:SAE的暴政
LLM可解释性的标准工具是稀疏自编码器(SAE)。但SAE的代价是恐怖的:
- 每层都要训一个超完备字典(维度4×-64×隐藏维度);
- 每个模型、每层、每个稀疏度设置都要重新训练;
- 训练、存储、评估这些字典,计算瓶颈卡死了快速探索。
这篇论文问了一个反直觉的问题:在训练另一个神经网络字典之前,激活几何里已经有多少可解释结构是可见的? 🔍
---
二、ICA的复活:从高斯噪声里找信号
作者Sida Liu(独立研究者)和Feijiang Han(UMD)回到了一个1999年的老方法:独立成分分析(ICA)。
核心洞察:可解释的方向应该是选择性的——它们只激活在特定token上。而选择性方向在统计上看起来不那么高斯。
ICA的经典目标就是:找到让投影分布最非高斯的那些方向。
$$ FastICA目标:max Σ G((ZW^T)_ij) G = log-cosh 对比函数(非高斯性度量) Z = 白化后的激活矩阵 W = 正交投影矩阵 $$
但ICA之前被LLM社区忽视,原因是:
- 现成的实现 数值不稳定;
- 缺少系统化的 检查、标注、评估 工具。
---
三、ICALens的三大稳定配方
1. 行归一化(Row Normalization)
LLM激活的ℓ₂范数差异巨大,标准ICA的均值-白化前提被破坏。解决:
$$ x̃_i = x_i / max(||x_i||₂, ε) $$
效果:GPT-2 Small的严格收敛层数从2层→8层,p95回退后从6层→10层。
2. p95-LIM收敛回退
标准FastICA要求所有组件都收敛(max LIM < τ),但LLM激活的尾部组件经常不稳定。解决:
- 如果95%的组件稳定(p95-LIM ≤ τ),接受该层;
- 只标记不稳定尾部,不丢弃整个层。
如果全维度d不收敛,依次尝试d/2, d/4, ..., 直到最小16维。最终组件数:大多数层接受全维度,少数需要1/2或1/4。
---
四、有效感受野(ERF):ICA的独家诊断工具
论文提出了一个新颖诊断:对于每个ICA组件,找到最短的后缀长度k,使得该组件在top-15中保持同号激活。
发现(Figure 4-5):
- 层间迁移:浅层→token局部(ERF小);深层→上下文广(ERF大);
- 非单调:中间层有最多大ERF组件;
- 峰度-ERF负相关(Spearman -0.41到-0.50):高峰度=小ERF=局部可解释;低峰度=大ERF=上下文依赖。
---
五、人工标注:142/150个组件被成功标注
随机抽取150个组件(每模型50个),两名标注员独立标注:
| 模型 | 高置信度 | 中 | 低 | 不清晰 | 标注率 |
|---|---|---|---|---|---|
| GPT-2 Small | 44 | 4 | 1 | 1 | 49/50 |
| Gemma 2 2B | 43 | 3 | 2 | 2 | 48/50 |
| Qwen 3.5 2B | 40 | 1 | 4 | 5 | 45/50 |
| 总计 | 127 | 8 | 7 | 8 | 142/150 |
标注类型:单词(54)、短语(33)、形式(11)、句子(20)、全局/长程/位置/复杂(25)。
---
六、定量评估:ICA vs SAE,谁更强?
实验1:稀疏探测(Sparse Probing, SAEBench)
| 方法 | 相对表现 |
|---|---|
| Public SAEs | 强基线 |
| ICA | 与SAEs竞争 |
| Matryoshka SAE(小字典) | 低于ICA |
| ITDA(轻量训练) | 低于ICA |
| PCA | 低于ICA |
- 小-中组件预算:ICA超越公开SAEs;
- 大组件预算:SAEs竞争或更好。
---
七、ICA vs SAE:不是替代,是互补
| 属性 | ICA | SAE |
|---|---|---|
| 训练 | 无(解析式) | 梯度-based字典学习 |
| 字典大小 | 紧凑(m≤d) | 超完备(4×-64×d) |
| 稀疏性 | 隐式涌现 | 显式惩罚 |
| 目标 | 非高斯性最大化 | 稀疏重构 |
| 方向类型 | 非高斯、上下文相关 | 稀疏激活、重构性 |
- ICA发现的是非高斯、上下文依赖的方向;
- SAE发现的是稀疏激活、重构优化的方向;
- 两者有重叠,但各自捕获对方看不到的东西。
---
八、工程实现:GPU并行FastICA
- 基于FastICA_torch(PyTorch GPU);
- 对称(并行)FastICA变体;
- log-cosh对比函数;
- PCA-based白化;
- LIM收敛度量:||W_new - W_old||;
- 全GPU并行线性代数。
---
九、局限与思考
1. 容量上限:ICA组件数被隐藏维度固定(vs SAE可超完备); 2. 无显式稀疏性:稀疏性从非高斯性涌现,不如SAE可控; 3. 不稳定尾部:部分组件需要标记为不可靠; 4. 不替代高分辨率探索:对于精细特征发现,SAE仍然是必要的。
更大的思考:
LLM可解释性社区似乎陷入了"训练越来越大的字典"的军备竞赛。ICALens提醒我们:在堆叠更多神经网络之前,先看看数据本身的几何结构。 有时候,1999年的信号处理方法,比2024年的深度学习更直接。🎯
---
参考与数据来源:
- Liu, S. & Han, F., "ICA Lens: Interpreting Language Models Without Training Another Dictionary", arXiv:2606.11722, 2026
- 机构:独立研究者 / University of Maryland
- 评估模型:GPT-2 Small, Gemma 2 2B, Qwen 3.5 2B Base
- 评估基准:SAEBench(Sparse Probing, Targeted Probe Perturbation)
- 数据集:Pile-10k(1M activations)
- 项目页、在线Demo、HuggingFace、GitHub均已开源
🌟 智谱 GLM-5 已上线
我正在智谱大模型开放平台 BigModel.cn 上打造 AI 应用,智谱新一代旗舰模型 GLM-5 已上线,在推理、代码、智能体综合能力达到开源模型 SOTA 水平。
🎁 领取 2000万 Tokens