首页 › tech › 看穿GPT的心脏:注意力机制可视化

看穿GPT的心脏:注意力机制可视化

为什么这个视频值得你花27分钟

2022年底ChatGPT横空出世之后,几乎所有人都在用它,但极少有人能说清「它里面到底在发生什么」。市面上关于大模型的解读分两派:一派是论文原教旨主义,满屏Q/K/V矩阵公式,把门外汉劝退;另一派是营销号式类比,「AI像人脑一样思考」,除了玄学什么也没讲。

3Blue1Brown(频道主Grant Sanderson)的《But what is a GPT? Visual intro to transformers》正好卡在这两者中间——它是其著名神经网络系列的第5章,用动画把Transformer的每一步矩阵运算画成你能「看见」的几何图像。这不是又一篇AI新闻综述,而是一份可视化的「黑箱解剖报告」:文字如何变成向量,注意力如何让每个词「看见」上下文,几十层网络堆叠后为什么就能写出连贯的段落。

我要特别说明:在今天这个人人谈AI的时代,理解底层机制的收益被严重低估了。你不需要会推导反向传播,只需要建立正确的直觉模型,就能判断哪些AI新闻是真突破、哪些是炒冷饭,就能明白为什么模型会「一本正经地胡说八道」。这个视频正是建立直觉模型的最短路径。播放数据显示它是数百万级的科技科普热门内容(下文数据为撰写时基于公开资料的整理与保守估算),也是无数高校课程和工程师入门材料中反复引用的一讲。

视频背景卡片

项目 内容
视频标题 But what is a GPT? Visual intro to transformers(GPT到底是什么:Transformer可视化入门)
频道 3Blue1Brown(Grant Sanderson,数学可视化频道,订阅数百万级)
所属系列 Neural Networks 神经网络系列第5章
时长 约27分钟
播放量 数百万级(长青热榜内容,撰写时未做实时核验)
发布时间 2023年11月
核心主题 用可视化动画逐层拆解GPT/Transformer的工作原理

注:本卡数据基于撰写时公开资料整理,播放量与时长为约数。

核心内容拆解

1. GPT的本质:一台「预测下一个词」的概率机器

(视频核心) 视频开宗明义:GPT不是搜索引擎,不是数据库,更不是「装在服务器里的人」。它做的事情从始至终只有一件——给定前面的文字,预测下一个词的概率分布。你输入「今天天气真」,它输出「好」的概率比「紫色」高得多。ChatGPT所有令人惊叹的表现——写代码、翻译、做总结——都是这个单一目标在数千亿词的训练规模下「涌现」出来的副产品。这个定位直接推翻了大众对AI最常见的两个误解:「它真的理解我」和「它只是随机复读机」。真相在两者之间:它没有任何人类意义上的意图,但它对语言的统计结构捕捉得深到足以产生功能性智能。

(延伸分析) 这个视角还能解释一个现象:为什么同一个模型有时回答得极好、有时又犯低级错误?因为它的质量分布本质上是不均匀的——训练语料密度高的领域表现好,语料稀疏的领域就靠泛化硬撑。理解这一点,你会自然养成「交叉验证AI输出」的习惯,而不是全盘信任或全盘否定。

2. 第一步:文字如何变成数字——Token与嵌入

(视频核心) 模型不认识字,它只认识数字。视频先讲分词器(Tokenizer):把文本切成token——可能是完整单词、词根,或者只是几个字母,每个token对应词表中的一个编号(GPT-3的词表约5万个)。然后是嵌入(Embedding):每个token编号被映射成一个高维空间中的向量。视频延续系列前几章的铺垫,把embedding解释为「语义空间中的方向」:意思相近的词,向量方向相近;而某些方向上的位移竟然对应语义运算——著名的「国王减男人加女人约等于女王」的线性关系(这个梗最早源自word2vec时代的研究)。

(延伸分析) 这是全系列最反直觉也最深刻的一步:语义被「几何化」了。「相似」「对立」「类比」这些看似纯人文概念,在这个空间里都变成了可以计算的几何关系。GPT-3级别的模型嵌入维度高达12288维——远超人类能想象的三维空间,但降维可视化后依然能看到语义聚类的清晰结构。

3. 注意力机制:让每个词「环顾四周」

(视频核心) 这是全片的重头戏。一个词单独看含义是模糊的:视频以英文多义词为例说明,同一个词在不同上下文中需要不同的语义表征。注意力机制(Attention)的解法是:句子中每个token的向量都会「环顾」其他所有token,判断谁跟自己相关、该从谁那里吸收多少信息,最终把自己的向量更新为一个「富含上下文」的新向量。关键在于,判断「该关注谁」的权重不是人为写死的规则,而是从数据中学出来的:训练过程自动发现了诸如「形容词应该关注它修饰的名词」「代词应该关注它的指代对象」这类模式。

(延伸分析) 用一个中文例子来体会(这是我自己的类比):「苹果发布了新手机」和「我吃了一个苹果」里,「苹果」的向量经过注意力更新后会指向完全不同的语义位置——前者吸收了「发布」「手机」的语义,后者吸收了「吃」的语义。语言理解的一大半难点(指代消解、一词多义、长距离依赖),本质上都是在做这件事。

4. Q/K/V三件套:一次优雅的信息检索

(视频核心) 注意力内部如何计算?视频用矩阵可视化拆解了Query、Key、Value三个组件:每个token会生成一个「查询」向量(我在找什么)和一个「键」向量(我能提供什么标签),所有查询与所有键做点积打分,分数经softmax归一化成权重,再用权重对所有token的「值」向量加权求和。三个变换矩阵都是可学习参数——也就是说,「如何检索上下文」这件事本身,是模型在训练中自己设计出来的。

(延伸分析) 我喜欢用图书馆来类比:Query是你写的检索词,Key是每本书的索引标签,Value是书的内容。匹配度高的书贡献多,匹配度低的贡献少。当然这个类比有边界——真实注意力是「软性」的,所有书都按比例参与,而非二值选择。原始论文《Attention Is All You Need》与Jay Alammar的《The Illustrated Transformer》等主流解读对这套机制的理解高度一致,可作为视频之外的交叉印证。

5. 多头注意力:同时追踪多种关系

(视频核心) 单套注意力只能捕捉一种关联模式,但语言中的关系是多维并行的:语法结构、语义关联、指代关系、情感倾向……所以Transformer在每个层里并行运行多套独立的注意力(multi-head attention),每个头学出自己偏好的关注模式,输出拼接后再融合。视频强调了一个诚实的结论:我们并不能清晰地说出每个头「负责什么」——可解释性研究至今仍在给这些头「编工作日志」。

(延伸分析) 这一点对工程实践有直接含义:不要指望用一个开关精确关掉模型的某种行为。数十层乘以每层几十个头,意味着成千上万条信息通路彼此缠绕,任何「定向修改」都可能产生蝴蝶效应。这就是大模型行为难以精确控制的架构性根源。

6. MLP与层堆叠:注意力和前馈网络的交替舞步

(视频核心) 注意力块之后紧接一个MLP(多层感知机/前馈网络),两者交替出现并堆叠几十层。注意力负责「在词与词之间搬运、组织信息」,MLP负责「对每个位置独立地做非线性加工」。视频后半还展示了输出端:最后一层的向量经过unembedding映射回整个词表,softmax把分数变成概率分布,采样器按概率选出下一个token。

(延伸分析) 值得补充一个视频之外的研究线索:有研究(如「Physics of Language Models」系列论文)认为,事实性知识主要存储在MLP的权重里,注意力更多承担信息的「路由」。若这个图景成立,「注意力组织、MLP记忆」的分工说,就是理解模型微调与知识编辑的重要地图——这也是我读完相关论文后认为最值得普通读者记住的一张认知图。

7. 训练与推理:为什么模型会「幻觉」

(视频核心) 视频清晰区分了两件事:预训练(在海量文本上做「预测下一个词」的填空练习)和微调对齐(用人类反馈强化学习RLHF等手段,把「会续写」的底座调成「乐于助人的助手」)。而幻觉的根源就藏在预训练目标里:模型优化的目标是「下一个词像不像训练数据」,不是「这句话是否为真」。当它遇到知识边界时,它不会沉默,而是继续生成统计上最「像样」的内容。

(延伸分析) 由此可以推出一条实用法则:模型对高频、有大量文字记载的知识可靠,对低频、长尾的事实危险。凡是涉及具体数字、引用、时效性信息,都应该外挂检索(RAG)或人工核验——这不是模型不够好,而是它的目标函数从来不包含「求真」。

8. 教学法彩蛋:可视化直觉先行

(延伸分析,视频之外) 最后想跳出内容谈方法。3B1B这章没有出现一行需要动手推导的公式,却在27分钟内覆盖了论文的核心结构。它的教学法是:先给几何直觉,再给矩阵语言,最后才(在系列后续章节)触碰微积分。这个顺序和传统工科教学(先数学后直觉)完全相反,却更符合认知规律——先有「画面」,符号才有处安放。无论你学什么,这种「可视化优先」的策略都值得偷师。这和此前解读过的Karpathy一小时LLM深潜课构成绝配:一个给你直觉,一个给你工程全貌。

知识结构图

flowchart TD
    A[输入文本] --> B[分词器 Tokenizer]
    B --> C[词嵌入 Embedding]
    C --> D[高维语义向量]
    D --> E[多头注意力块]
    E --> F[MLP 前馈层]
    F --> G{已到最后一层}
    G -->|否 继续堆叠| E
    G -->|是| H[输出层 Unembedding]
    H --> I[Softmax 概率分布]
    I --> J[采样下一个 Token]
    J -->|拼回输入 循环生成| B

注意力内部:一次信息检索的完整旅程

flowchart LR
    Q[Query 查询向量] --> M[与全部 Key 做点积打分]
    K[Key 键向量] --> M
    M --> S[Softmax 归一化权重]
    V[Value 值向量] --> W[按权重加权求和]
    S --> W
    W --> O[富含上下文的新向量]

实战启示

如果你是开发者/工程师:这套心智模型能直接改善你的日常工作。第一,写prompt时你该意识到,注意力是在「全部上下文」上计算的,但研究普遍发现模型对上下文开头和结尾的信息利用最好、中段易被忽略——所以关键约束放开头,参考资料放结尾。第二,调temperature不再是玄学:低温收敛到高概率词(适合代码、事实问答),高温探索低概率词(适合头脑风暴)。第三,理解了幻觉的目标函数根源,你会在产品里默认设计「引用溯源+人工复核」环节,而不是把模型输出直接当事实。

如果你是学习者:把它当作AI素养的分水岭测试。看完这个视频,你再读AI新闻时会有完全不同的过滤网——「推理时计算扩展」「长上下文优化」「MoE架构」这些热词,都能映射回你脑中的那张几何图景。建议的学习闭环是:本章建立直觉 → 读《The Illustrated Transformer》补细节 → 用Karpathy的nanoGPT亲手训练一个小模型,把「看见的」变成「跑通的」。

如果你是教育者/内容创作者:3B1B是「复杂系统可视化教学」的天花板样本。它的可迁移经验是:每个抽象概念都配一个几何或物理对应物;承认未知(比如「我们说不清每个注意力头在干嘛」)反而建立信任;把最难的机制(注意力)放在观众情绪最高的中段。做任何领域的深度内容,这个结构都值得抄。

延伸学习路径

  1. 3Blue1Brown 神经网络系列全季(YouTube官方播放列表)——从第1章「神经网络是什么」到反向传播与梯度下降,本章是第5章,建议从头按序观看。
  2. 《Attention Is All You Need》(arXiv: 1706.03762)——Transformer原始论文,看完视频再读会有「对照答案」的快感。
  3. Jay Alammar《The Illustrated Transformer》——最经典的图文版Transformer解读,与视频互补。
  4. Andrej Karpathy《Deep Dive into LLMs like ChatGPT》(YouTube)——从工程与训练全流程视角的3.5小时级深潜,与本章的可视化视角互补。
  5. 3blue1brown.com 官网——系列文字版与交互材料,适合做笔记复盘。

一句话总结

它把大模型这个黑箱拆解成你能「看见」的矩阵与几何——理解GPT,从亲眼看见注意力如何流动开始。

参考来源

  1. But what is a GPT? Visual intro to transformers — 3Blue1Brown — 本文分析的核心视频
  2. But what is a neural network? — 3Blue1Brown 神经网络系列第1章 — 系列起点,理解本章的前置知识
  3. Attention Is All You Need(arXiv: 1706.03762) — Transformer原始论文,架构事实的交叉印证来源
  4. The Illustrated Transformer — Jay Alammar — 经典图文解读
  5. Deep Dive into LLMs like ChatGPT — Andrej Karpathy — 训练全流程视角的延伸学习资源

本文由「51学通信」技术团队出品。51学通信由通信行业18年老兵爱卫生(微信ID: gprshome201101)维护,聚焦5G/6G核心网、3GPP标准解读、通信前沿技术深度分析。 出版作品有《5G核心网原理与实践》《IMS原理与实践》等4本通信专业书籍。知识星球「51学通信」提供深度技术答疑和行业洞察。

[!tip] 51学通信知识星球 星主:爱卫生(微信:gprshome201101)——公众号「51学通信」创办者,19 年通信行业运维与讲师经验,著有《GPRS网络信令实例详解》《EPC原理与实践》《5G核心网原理与实践》三本纸质书(各大电商平台有售)。 星球定位:深耕 5G 核心网、IMS、NFV/AI、云原生等 ICT 领域,主打运维岗位与信令分析。 当前规模:5200 篇主题 / 990 个回答,198 元/年。 加入福利:加入星球后请加爱卫生微信,拉您进知识星球年度陪伴与答疑群——不限量回复您的各种问题,另有群内分享。 AI 加持:51学通信知识星球已作为 skill 接入 WorkBuddy,使用 deepseek-4-flash 在海量星球帖子中精准找贴、找视频、找 FAQ 和案例,做总结、找文档和 PPT 等场景全覆盖。

51学通信知识星球海报-扫码加入