土法炼钢兴趣小组的算法知识备份

【Transformer 与注意力机制】52|可解释性入门:注意力权重真的是“解释”吗

文章导航

分类入口
transformer
标签入口
#transformer#interpretability#attention#probing#explainability

目录

Transformer 有一个看似天然适合解释的部件:attention。每个 token 对其他 token 有权重,把它画成热力图,看起来就像模型在告诉我们“我关注了哪里”。问题是,关注哪里不等于为什么这样回答。

可解释性要区分几个层次。给用户看的解释可能是自然语言理由;行为解释描述输入变化如何影响输出;机制解释则试图找出模型内部哪些组件以什么方式导致了结果。attention weight 只是内部信号之一,不能自动承担所有解释责任。

本篇能让你学会三件事:

  1. 为什么 attention weight 不是完整解释;
  2. 梯度、遮挡、探针和因果干预分别能说明什么;
  3. 为什么大语言模型的解释尤其困难。

一、什么叫解释

“解释模型”可以指很多事。产品用户想知道为什么贷款申请被拒;工程师想知道为什么模型在某类样本上失败;研究者想知道某个 attention head 是否实现了某种算法。

这些问题需要不同证据。用户解释需要可理解和合规;调试解释需要定位错误来源;机制解释需要因果证据。把它们混在一起,会导致“看起来合理”的解释被误当成真实机制。

模型自己生成的解释也要小心。语言模型很会写理由,但它写出的理由未必是产生答案的真实内部原因。


二、attention weight 的诱惑

attention weight 确实提供了有用线索。某个 head 强烈关注前文中的实体、括号或分隔符,可能说明它在处理某种结构。可视化 attention 能帮助发现模型模式。

但 attention weight 只是计算路径的一部分。Transformer 还有 MLP、残差流、LayerNorm、多个 head 和多层组合。一个 token 的最终表示不是由单个 attention 图决定。

更重要的是,权重高不等于因果贡献大。某个位置被关注,不代表它决定了输出;某个位置权重低,也可能通过其他层或 MLP 间接影响结果。


三、attention is not explanation

“Attention is not Explanation” 这类工作指出,attention 权重和模型预测之间的关系不总是可靠解释。有时可以改变 attention 分布而输出不变,有时重要输入不一定对应最高 attention。

这并不意味着 attention 可视化毫无价值。更准确的结论是:attention 可以作为诊断线索,但不能单独作为因果解释。要证明某个组件重要,需要干预:修改它、遮掉它、替换它,观察输出如何变化。

解释的核心不是“看起来像”,而是“如果改变这里,结果会不会变”。


四、梯度、遮挡和探针

梯度方法看输出对输入或激活的敏感性。梯度大,说明小变化可能影响输出。但梯度容易受饱和、尺度和局部线性近似影响。

遮挡实验更直接:把某个 token、区域或特征遮住,看输出变化。它接近因果干预,但遮挡本身可能制造分布外输入。

探针(probing)训练一个小模型,从内部表示预测某种属性。探针能说明表示中包含信息,但不能说明主模型实际使用了这些信息。线性探针能读出语法特征,不代表模型决策依赖它。

这些方法各有价值,也各有边界。


五、行为解释与机制解释

行为解释关注输入输出关系。比如改变 prompt 中某句话,模型回答如何变化;加入检索证据,幻觉是否减少。这对产品调试很有用。

机制解释关注内部电路。比如某些 attention head 是否负责复制模式,某些 MLP feature 是否表示具体概念,残差流中信息如何移动。这更接近“模型内部如何计算”。

两者都重要。只有行为解释,容易停留在黑箱测试;只有机制解释,又可能脱离真实用户场景。


六、为什么 LLM 更难解释

大语言模型规模巨大,层数多、head 多、MLP 维度高,行为还经过指令微调和 RLHF 改造。它们不仅预测文本,还会遵循角色、使用工具、拒绝请求、维持对话状态。

同一个输出可能由预训练知识、上下文示例、系统指令、解码策略和安全策略共同决定。把原因归结到某个 attention head,通常过度简化。

所以 LLM 可解释性需要多层证据:行为评测、数据分析、激活干预、特征分解和安全红队。


七、关键概念回顾


八、常见误解

8.1 “attention 图就是解释”

attention 图是线索,不是完整因果解释。

8.2 “模型自己说的理由就是真实理由”

不一定。生成的理由可能是事后合理化。

8.3 “探针读出信息说明模型用了它”

探针只能说明信息存在于表示中,不保证主模型决策依赖它。


九、下一步

入门可解释性之后,下一篇进入机制可解释性:电路、特征、归因和 activation patching 如何更接近模型内部计算。


十、参考文献

  1. Jain, S. and Wallace, B. “Attention is not Explanation.” NAACL 2019.
  2. Wiegreffe, S. and Pinter, Y. “Attention is not not Explanation.” EMNLP 2019.
  3. Belinkov, Y. “Probing Classifiers: Promises, Shortcomings, and Advances.” Computational Linguistics 2022.
  4. Ribeiro, M. T. et al. “Why Should I Trust You?: Explaining the Predictions of Any Classifier.” KDD 2016.

← 上一篇:51|量化、蒸馏、剪枝 | 下一篇:53|机制可解释性

同主题继续阅读

把当前热点继续串成多页阅读,而不是停在单篇消费。

2026-06-09 · transformer

【Transformer 与注意力机制】59|推理退化:为什么大模型会输出乱码、死循环和无意义文本

大模型推理时偶尔会突然陷入死循环、输出乱码或连续无意义数字,这不是随机 bug,而是注意力机制、Causal Mask、解码策略和数值精度在自回归生成中共同作用的结果。本文从 QKV 计算坍塌出发,解释 Attention Sink、Softmax 马太效应、Causal Mask 的退路切断、FP16 溢出路径和 KV Cache 污染,并给出从架构到运行时的多层防线。

2026-04-15 · transformer

【Transformer 与注意力机制】系列总览

从《Attention Is All You Need》出发把 Transformer 注意力机制、Q/K/V、多头注意力、位置编码、Causal Mask、Softmax、FFN、训练范式、模型变体、推理工程、可解释性、未来架构以及推理退化防御串成 59 篇深度博客。

2026-04-15 · transformer

【Transformer 与注意力机制】01|为什么要从这里开始

这是【Transformer 与注意力机制】系列的第一篇,承担两件事:一是把这套五十多篇文章为谁写、解决什么问题、彼此之间是什么关系交代清楚;二是为完全没基础的读者画出一条从向量、点积、矩阵乘法走到自注意力、再走到大语言模型的爬升路径,让你在投入时间之前先知道终点在哪、路上要经过哪些坎、读完之后你会、还不会做什么事。


By .