概念-SAE
概述
稀疏自编码器(Sparse Autoencoder, SAE)是机械解释性(Mechanistic Interpretability)领域的核心工具,用于从神经网络的激活中提取可解释的特征。SAE 的核心理念是:虽然单个神经元通常是多义的(polysemantic,即一个神经元响应多个不相关的概念),但通过对激活施加稀疏约束,可以学习到一组单义(monosemantic)的特征表示,每个特征对应一个可解释的概念。
核心概念
多义性(Polysemanticity)vs 单义性(Monosemanticity)
在 LLM 的隐藏层中,单个神经元往往同时响应多个不同的输入模式,这一现象称为多义性。例如,一个神经元可能同时被”猫”和”汽车”激活,使得直接解释神经元的行为变得困难。
SAE 的目标是解耦多义性:将模型的激活 映射到一个高维稀疏特征空间 (其中 ),使得每个特征维度对应一个单一的可解释概念。
SAE 数学原理
SAE 包含一个编码器(Encoder)和解码器(Decoder):
- 编码器:
- 解码器:
训练目标(L1 稀疏约束):
其中 控制稀疏程度, 是 L1 正则化项,鼓励大多数特征维度为零。
关键概念对比
| 概念 | 定义 | 提出时间 | 代表工作 |
|---|---|---|---|
| Polysemanticity | 单个神经元响应多个不相关概念 | 2021 | Anthropic: “Toy Models of Superposition” |
| Monosemanticity | 每个特征维度对应单一可解释概念 | 2023 | Anthropic: “Towards Monosemanticity” |
| SAE | 稀疏自编码器,从激活中提取可解释特征 | 2023 | Anthropic, DeepMind (GDM) |
| Transcoder | 将 SAE 应用于 MLP 层的替代方案,直接建模输入-输出变换 | 2024 | DeepMind: “Transcoders” |
| Activation Steering | 通过操控 SAE 特征或隐藏向量来控制模型行为 | 2024 | Anthropic, various |
| Control Vectors | 从对比数据中提取的方向向量,用于引导模型行为 | 2024 | Representation Engineering (RepE) |
Golden Gate Claude:SAE 的标志性案例
2024年5月,Anthropic 发布了标志性的可解释性实验:Golden Gate Claude。
实验过程:
- 在 Claude 的中间层训练 SAE,发现一个特征专门响应”金门大桥”(Golden Gate Bridge)相关的概念。
- 通过 Activation Steering,将该特征的激活值放大到正常水平的数倍。
- 结果:Claude 开始在所有对话中”认为自己是金门大桥”,无论用户问什么,它都会把话题引向金门大桥。
这个实验首次直观地展示了 SAE 提取的特征是因果可操控的(causally manipulable),而非仅仅是相关性的。
Activation Steering 与控制向量
Activation Steering(激活操控)
Activation Steering 是一种通过修改模型内部激活来控制模型行为的技术。其核心流程:
- 训练 SAE:在目标层训练 SAE,提取可解释的特征。
- 识别特征:找到与目标行为(如诚实、拒绝有害请求、特定风格)相关的特征。
- 操控激活:在推理时,对目标特征的激活值进行加法操控(additive steering): 其中 是操控向量(steering vector), 是操控强度。
Control Vectors(控制向量)
Control Vectors 是 Representation Engineering(RepE)的变体,通过对比两组数据来计算控制方向:
其中 是正例数据集(如诚实的回答), 是负例数据集(如不诚实的回答), 是模型在特定层的隐藏表示。
| 技术 | 方法 | 粒度 | 优势 | 劣势 |
|---|---|---|---|---|
| SAE 特征操控 | 操控 SAE 学习的单个特征 | 细粒度(单特征) | 高精度、可解释性强 | 需要训练 SAE(成本高) |
| Control Vectors | 对比数据集差值的均值 | 中粒度(方向) | 无需训练、快速部署 | 可能混合多个特征 |
| Prompt Engineering | 修改输入文本 | 粗粒度(输入) | 最简单、无需修改模型 | 效果不稳定、易越狱 |
| RLHF/DPO | 强化学习或偏好优化 | 全局(所有权重) | 系统化、持久 | 训练成本极高 |
2024-2025 可解释性革命
2024-2025 年被称为机械解释性的”革命期”,主要里程碑:
- 2024.05:Anthropic 发布 Golden Gate Claude,首次展示 SAE 特征操控的惊人效果。
- 2024.06:Anthropic 发布 Scaling Monosemanticity 论文,在 Claude 3 Sonnet 上训练了数千万个 SAE 特征。
- 2024.07:DeepMind 发布 Gemma Scope,为 Gemma 2 系列模型提供了开源 SAE 套件。
- 2024.10:DeepMind 发布 Transcoders,提出 SAE 的替代方案,直接建模 MLP 子层的变换。
- 2025.01:SAE Bench 发布,标准化 SAE 评估基准。
- 2025.06:OpenAI 发布 GPT-5.5 的可解释性报告,揭示 CoT 链中的内部推理特征。
- 2026:Activation Steering 进入生产环境,用于 AI 安全对齐和内容审核。
应用场景
- AI 安全:检测和抑制模型的有害行为(欺骗、谄媚、权力寻求)
- 内容审核:通过操控特征来抑制不当内容生成
- 模型调试:定位导致特定错误的特征
- 可控生成:调整模型的写作风格、语气、诚实度
- 知识编辑:通过操控特定特征来修改模型的知识
相关笔记
- 第6章 · 神经网络基础:神经网络基础概念,包括激活函数、反向传播
- 第10章 · LLM 原理与架构演进:LLM 内部架构,隐藏层与注意力机制
- AI 安全基础:AI 安全基本概念,对齐(Alignment)问题
- 概念-DeepSeek:DeepSeek 的可解释性相关工作
- 推理模型 (Reasoning Models):推理模型的可解释性挑战
来源
- Anthropic: “Towards Monosemanticity: Decomposing Language Models With Dictionary Learning” (2023): https://transformer-circuits.pub/2023/monosemantic-features
- Anthropic: “Scaling Monosemanticity” (2024): https://transformer-circuits.pub/2024/scaling-monosemanticity
- Golden Gate Claude 博客: https://www.anthropic.com/news/golden-gate-claude
- DeepMind: “Gemma Scope” (2024): https://deepmind.google/discover/blog/gemma-scope
- Representation Engineering (RepE): https://arxiv.org/abs/2310.01405
- DeepMind Transcoders: https://arxiv.org/abs/2406.11944
- SAE Bench: https://arxiv.org/abs/2501.05266
- Neel Nanda 可解释性教程: https://neelnanda.io/mechanistic-interpretability