什么是 Transformer 的注意力机制(Attention)?
现代 ChatGPT、Claude 与所有大模型的灵魂中枢!看懂每个词如何通过 Q、K、V 互相“看上一眼”、为什么它能打破长文本遗忘魔咒,以及怎样把一句话里的隐形关联算得一清二楚!
旧的循环神经网络(RNN)就像排成一长队的传话游戏:必须前一个词算完才能算下一个词,无法在 GPU 上并行狂飙;更要命的是,句子只要长过几十个词,开头的信息被一层层压缩稀释,后面的人完全忘了开头说了啥(梯度消失与长期遗忘)。
2017 年《Attention Is All You Need》论文掀起的 AI 革命! 句子里的每个词同时站在大厅中央,彼此通过 Query(查询)、Key(键值标签)、Value(内容) 进行点积打分。无论相隔 1 个字还是 10 万个字,只需要 1 次矩阵相乘就能瞬间建立注意力连接,支持全并行计算!
揭秘核心三剑客:Q、K、V 的精密分工
从一维词向量通过可学习矩阵投射出的三重视角
1. Query (Q · 寻人启事)
代表当前 Token 主动发出的“需求向量”。比如动词想要找它的主语是谁,疑问词想要找它修饰的对象在哪里。
2. Key (K · 身份门牌)
代表每一个 Token 被动暴露给外界的“特征门牌号”。用来和别处的 Q 做匹配,二者越相关,内积打分就越高。
3. Value (V · 核心内容)
代表 Token 实际承载的语义信息。最终的输出,就是根据打分权重对全句所有 Token 的 V 进行加权求和打包带走。
📐 改变人类历史的数学公式:缩放点积注意力 (Scaled Dot-Product Attention)
仅用这一行优雅的线性代数公式,统治了万亿大模型时代:
🎮 交互模拟器:推演注意力机制在不同经典语境中的视线焦点
点击查看不同句子中,特定词汇把注意力百分比分给了谁: