TalentMe
How To
🗺️
AI Industry Map
NEW
Knowledge ▾
Intro & Usage
Machine Learning Repo
Data Science Repo
Resources ▾
Intro & Usage
AI Tech Vault (Tech Wiki)
Industry News
Tech Blogs
Research Papers
Open Source Projects
Tools Hub ▾
🛠️ Tools & Skills Overview
🌐 Interactive Web Tools
🗺️ AI Industry & Career Map
🧭 AI Career Transition Navigator & Roadmap
📚 AI Multi-Module Practice Hub
🎯 AI Skill Assessment
🧠 AI Skills Library
AI Skills & Prompts Overview
Local Agent Guide
Cloud Skills Templates
⚡ MCP Tools Suite
TalentMe MCP Guide
CLI Tools & Commands
Services ▾
🚀 Services & Plans Suite
🧭 1v1 Coaching & Services
💎 Plans & Pricing
💬 Contact & Consultation
Contact Us
💬 Discord
🌐
中
☀️
🔑
Login / Register
☰
技术知识库
›
复习路线图
›
大语言模型 思维导图
›
缩放点积注意力
← 返回 大语言模型 思维导图
中文
·
English
⚡ 大语言模型
ID:
attention-scaled-dot
缩放点积注意力
Scaled Dot-Product Attention
🎯
核心定义
Scaled Dot-Product Attention (缩放点积注意力) 是 Transformer 的注意力核心算子,计算
A
t
t
e
n
t
i
o
n
(
Q
,
K
,
V
)
=
s
o
f
t
m
a
x
(
Q
K
T
d
k
)
V
Attention(Q,K,V) = softmax(\frac{Q K^T}{\sqrt{d_k}}) V
A
tt
e
n
t
i
o
n
(
Q
,
K
,
V
)
=
so
f
t
ma
x
(
d
k
Q
K
T
)
V
。除以
d
k
\sqrt{d_k}
d
k
用于归一化点积方差:当 Q、K 各分量方差为 1 时,
d
k
d_k
d
k
维点积的方差与
d
k
d_k
d
k
成正比,若不缩放,softmax 输入随维度膨胀进入饱和区(如
d
k
=
128
d_k = 128
d
k
=
128
时
128
≈
11.3
\sqrt{128} \approx 11.3
128
≈
11.3
),梯度趋近 0。
💡
使用场景
每个注意力头对序列内所有 token 做软匹配,是 MHA/GQA/MLA、FlashAttention、RoPE 等一切注意力话题的基础;面试常从"为什么除以
d
k
\sqrt{d_k}
d
k
"切入,再追问复杂度、mask 与多头拆分。
⚡
解决的核心痛点
相比不缩放的点积,缩放保证任意维度下 softmax 的熵与梯度稳定,深层网络可训练;但时间与显存仍是
O
(
n
2
)
O(n^2)
O
(
n
2
)
,长序列下成为主要瓶颈,需要 FlashAttention 等 IO 优化来缓解。
🎯
5 个高频面试考点 (Exam Points)
1
为什么注意力要除以
d
k
\sqrt{d_k}
d
k
?不除以会怎样(方差分析与 softmax 饱和)?
2
缩放点积注意力的时间与空间复杂度是多少?为什么
O
(
n
2
)
O(n^2)
O
(
n
2
)
是长上下文瓶颈?
3
Q、K、V 的投影矩阵形状是什么?多头如何拆分与拼接?
4
padding mask 与 causal mask 的区别与实现方式?
5
单头 vs 多头:为什么拆成 h 个头而不是一个
d
d
d
维大头?
📖 关联深度指南:
📄 transformer-architecture →
更新于 2026-08-12
🎯
检验攻克程度:针对「缩放点积注意力」专属刷题排雷
做单选排雷题、推导选项机制,答错自动收录进专属错题本。
🚀 开始本考点专项刷题 ➔
← 上一个知识点
KV Cache 管理
下一个知识点 →
注意力变体 MHA/MQA/GQA
🔗 更多 大语言模型 知识点卡片
Agent 与工具调用
对齐税与偏好数据
评测基准 MMLU/GSM8K
分词 BPE/WordPiece