自注意力机制含义

标题:【Transformer 核心】自注意力机制(Self-Attention):原理、计算与作用

摘要:

自注意力是 Transformer 的灵魂:让序列中每个元素与所有元素"互相关注",直接建模全局依赖,解决 RNN 长依赖弱、无法并行的问题。

一、核心思想

对输入序列 XXX,每个词都生成:

  • Q(Query,查询):我要找什么
  • K(Key,键):我有什么
  • V(Value,值):我携带什么信息

通过 Q 与所有 K 计算相似度,加权求和 V,得到融合全局信息的新表征

二、计算公式

Attention(Q,K,V)=softmax(QKTdk)V \text{Attention}(Q,K,V) = \text{softmax}\left(\frac{QK^T}{\sqrt{d_k}}\right)V Attention(Q,K,V)=softmax(dk QKT)V

  • QKTQK^TQKT:计算每对元素相似度
  • dk\sqrt{d_k}dk :缩放,防止点积过大导致 softmax 梯度消失
  • softmax:归一化权重
  • 乘 V:加权求和得到输出

三、作用

  • 全局建模:直接捕捉长距离依赖(如句子首尾关系)
  • 并行计算:不像 RNN 必须逐时间步计算
  • 动态权重:根据输入内容自动决定关注谁

小结:

自注意力的本质是序列全局加权融合,用相似度动态聚合所有位置信息,是 Transformer 取代 RNN 的关键。

相关推荐
杨杨杨大侠23 分钟前
从零训练一个 321 万参数的小模型:用 MLX 看懂 Transformer 的训练过程
aigc·openai·ai编程
RainyJiang2 小时前
AI时代下,Android的边界正在消失
android·openai·ai编程
桦说编程2 小时前
【AtomicAgent系列1】变异测试——过去做不起,现在 agent 做得起
后端·ai编程·vibecoding
超级架构师4 小时前
先在“可能世界”中测试自治系统:PEIRAVELA 的实验控制平面
人工智能·架构·ai编程
码农胖大海4 小时前
我的第一个产品,只有一段提示词
前端·ai编程·产品
数字补给站4 小时前
Windows 10 + Hyper-V + Terraform + cloud-init:批量产出 3 台 Ubuntu 的实践笔记
运维·ai编程
学者猫头鹰5 小时前
Spring AI 教程(下篇)
ai编程
Flynt5 小时前
我给 Claude Code 装了 Ponytail,代码量直接砍了一半
agent·ai编程·claude
ServBay6 小时前
谷歌发布 Gemini 3.8 Flash,官方跑分很厉害,但又被嘲了?
aigc·ai编程·gemini