Transformer模型自2017年由Vaswani等人提出以来,已成为自然语言处理领域的基石性架构。本文将深入剖析Transformer的每一层结构,详细讲解从输入到输出的完整计算流程,不遗漏任何关键细节。

输入表示层:词嵌入与位置编码

Transformer的输入处理包含两个关键步骤:词嵌入和位置编码。

词嵌入(Word Embedding)

词嵌入将离散的词汇符号映射为连续的向量表示。假设词汇表大小为,嵌入维度为,则嵌入矩阵为:

对于输入序列中的每个词,其嵌入向量为:

计算示例:
假设:

  • 词汇表大小
  • 嵌入维度
  • 输入序列[“the”, “cat”],对应词汇索引

随机初始化嵌入矩阵E:

E = np.random.randn(10000, 512) * 0.02  # 标准差0.02的正态分布

获取词向量:

e1 = E[5]   # shape (512,)
e2 = E[20]  # shape (512,)

位置编码(Positional Encoding)

由于Transformer不包含循环或卷积结构,需要显式注入位置信息。位置编码使用不同频率的正弦和余弦函数:

其中是位置,是维度索引。最终输入表示是词嵌入和位置编码的逐元素和:

位置编码的设计使得模型可以学习到相对位置关系,且能够处理比训练时更长的序列。

计算示例:
计算pos=0(第一个词)的位置编码:

position = 0
d_model = 512
pe = np.zeros(512)
for i in range(0, 512, 2):
denominator = np.power(10000, 2*i/d_model)
pe[i] = np.sin(position / denominator)
pe[i+1] = np.cos(position / (10000**(2*i/d_model)))

最终输入表示:

h1_0 = e1 + pe(pos=0)  # shape (512,)
h2_0 = e2 + pe(pos=1)  # shape (512,)
H_0 = np.stack([h1_0, h2_0])  # shape (2, 512)

编码器结构:多头注意力与前馈网络

Transformer编码器由N个相同层堆叠而成,每层包含两个子层:多头自注意力机制和位置全连接前馈网络。

多头自注意力机制(Multi-Head Self-Attention)

注意力基础计算

首先将输入通过线性变换得到查询(Q)、键(K)、值(V)向量:

其中参数矩阵维度为:

注意力分数计算:

计算示例:
假设:

  • d_k = d_v = 64
  • h=8个头
  • 当前输入H_0 shape (2,512)

初始化参数矩阵:

W_Q = np.random.randn(512, 64) * 0.1
W_K = np.random.randn(512, 64) * 0.1
W_V = np.random.randn(512, 64) * 0.1

计算Q,K,V:

Q = H_0 @ W_Q  # shape (2,64)
K = H_0 @ W_K  # shape (2,64)
V = H_0 @ W_V  # shape (2,64)

计算注意力分数:

attn_scores = Q @ K.T / np.sqrt(64)  # shape (2,2)
attn_weights = softmax(attn_scores)  # 按行softmax
output = attn_weights @ V  # shape (2,64)

多头注意力实现

多头注意力并行执行次不同的注意力计算:

其中参数矩阵:

通常设置,保持计算量不变。

计算示例:
假设8个头的输出:

head1 = np.random.randn(2,64)  # 模拟第一个头输出
...
head8 = np.random.randn(2,64)  # 模拟第八个头输出
multi_head = np.concatenate([head1,...,head8], axis=-1)  # shape (2,512)
W_O = np.random.randn(512,512) * 0.1
output = multi_head @ W_O  # shape (2,512)

残差连接与层归一化

每个子层后都应用残差连接和层归一化:

层归一化计算:

其中是均值和方差,是可学习参数。

计算示例:
假设:

  • 输入h shape (2,512)
  • 子层输出sublayer_out shape (2,512)

残差连接:

residual = h + sublayer_out  # shape (2,512)

层归一化计算:

mean = np.mean(residual, axis=-1, keepdims=True)  # shape (2,1)
var = np.var(residual, axis=-1, keepdims=True)   # shape (2,1)
gamma = np.ones(512)  # 初始化为1
beta = np.zeros(512)  # 初始化为0
h_out = gamma * (residual - mean) / np.sqrt(var + 1e-5) + beta

位置全连接前馈网络(Position-wise FFN)

前馈网络对每个位置独立应用相同的两层全连接:

其中参数维度:

通常,提供足够的表达能力。

计算示例:
参数设置:

d_ff = 2048
W1 = np.random.randn(512, 2048) * 0.1
b1 = np.zeros(2048)
W2 = np.random.randn(2048, 512) * 0.1
b2 = np.zeros(512)

计算过程:

x = np.random.randn(2,512)  # 输入
hidden = np.maximum(0, x @ W1 + b1)  # ReLU激活
output = hidden @ W2 + b2  # shape (2,512)

解码器结构:掩码自注意力与编码器-解码器注意力

解码器也由N个相同层堆叠,但包含三种子层:掩码自注意力、编码器-解码器注意力和前馈网络。

掩码自注意力(Masked Self-Attention)

为防止解码器看到未来信息,在自注意力计算中应用掩码:

计算时:

计算示例:
假设解码器输入3个词:

Q = np.random.randn(3,64)
K = np.random.randn(3,64)
attn_scores = Q @ K.T / np.sqrt(64)  # shape (3,3)

应用掩码:

mask = np.triu(np.ones((3,3)), k=1)  # 上三角矩阵
masked_scores = attn_scores - 1e9 * mask  # 未来位置设为-∞
attn_weights = softmax(masked_scores)  # 按行softmax

编码器-解码器注意力(Encoder-Decoder Attention)

此处的查询来自解码器前一层的输出,而键和值来自编码器的最终输出:

这使得解码器可以关注输入序列的相关部分。

输出层:线性变换与softmax

最终输出层将解码器输出转换为预测词的概率分布:

  • 线性变换:将维向量映射到词汇表大小

  • Softmax归一化:

计算示例:
假设:

  • 解码器最后输出 shape (3,512)
  • 词汇表V=10000
W_out = np.random.randn(512, 10000) * 0.1
logits = decoder_output @ W_out  # shape (3,10000)
probs = softmax(logits, axis=-1)  # 得到每个词的概率分布

关键技术与优化细节

缩放点积注意力

缩放因子防止点积过大导致softmax梯度消失:

多头注意力的优势

多头机制允许模型:

  • 在不同位置共同关注来自不同表示子空间的信息
  • 学习多种依赖关系模式(如局部/全局、语法/语义)

残差连接的作用

残差连接:

  • 缓解深层网络梯度消失问题
  • 使模型能够学习恒等映射,保留原始信息

层归一化的位置

Transformer采用后归一化(Post-LN):

  • 归一化在残差连接之后进行
  • 相比前归一化(Pre-LN)训练更稳定,但需要更仔细的参数初始化

数学视角下的完整前向传播

给定输入序列,完整计算流程:

输入嵌入:

编码器每层计算:

解码器每层计算:

输出概率:

扩展讨论:Transformer的变体与改进

高效注意力机制

原始自注意力复杂度O(n²)限制了长序列处理,改进包括:

  • 稀疏注意力(如Longformer)

  • 局部窗口注意力(如Swin Transformer)

  • 低秩近似(如Linformer)

位置表示的改进

  • 相对位置编码(如Transformer-XL)

  • 可学习的位置嵌入

  • 旋转位置编码(RoPE)

模型架构演进

  • 编码器-解码器(BERT/GPT)

  • 仅解码器(GPT系列)

  • 仅编码器(BERT)

Transformer架构通过自注意力机制实现了强大的序列建模能力,其精心设计的各层组件协同工作,克服了传统RNN/CNN的局限性。理解每一层的数学细节对于模型调优和架构创新至关重要。随着研究的深入,Transformer仍在不断演进,但其核心思想将继续影响深度学习的发展方向。

普通人如何抓住AI大模型的风口?

领取方式在文末

为什么要学习大模型?

目前AI大模型的技术岗位与能力培养随着人工智能技术的迅速发展和应用 , 大模型作为其中的重要组成部分 , 正逐渐成为推动人工智能发展的重要引擎 。大模型以其强大的数据处理和模式识别能力, 广泛应用于自然语言处理 、计算机视觉 、 智能推荐等领域 ,为各行各业带来了革命性的改变和机遇 。

目前,开源人工智能大模型已应用于医疗、政务、法律、汽车、娱乐、金融、互联网、教育、制造业、企业服务等多个场景,其中,应用于金融、企业服务、制造业和法律领域的大模型在本次调研中占比超过 30%。
在这里插入图片描述

随着AI大模型技术的迅速发展,相关岗位的需求也日益增加。大模型产业链催生了一批高薪新职业:
在这里插入图片描述

人工智能大潮已来,不加入就可能被淘汰。如果你是技术人,尤其是互联网从业者,现在就开始学习AI大模型技术,真的是给你的人生一个重要建议!

最后

只要你真心想学习AI大模型技术,这份精心整理的学习资料我愿意无偿分享给你,但是想学技术去乱搞的人别来找我!

在当前这个人工智能高速发展的时代,AI大模型正在深刻改变各行各业。我国对高水平AI人才的需求也日益增长,真正懂技术、能落地的人才依旧紧缺。我也希望通过这份资料,能够帮助更多有志于AI领域的朋友入门并深入学习。

真诚无偿分享!!!
vx扫描下方二维码即可
加上后会一个个给大家发

【附赠一节免费的直播讲座,技术大佬带你学习大模型的相关知识、学习思路、就业前景以及怎么结合当前的工作发展方向等,欢迎大家~】
在这里插入图片描述

大模型全套学习资料展示

自我们与MoPaaS魔泊云合作以来,我们不断打磨课程体系与技术内容,在细节上精益求精,同时在技术层面也新增了许多前沿且实用的内容,力求为大家带来更系统、更实战、更落地的大模型学习体验。

图片

希望这份系统、实用的大模型学习路径,能够帮助你从零入门,进阶到实战,真正掌握AI时代的核心技能!

01 教学内容

图片

  • 从零到精通完整闭环:【基础理论 →RAG开发 → Agent设计 → 模型微调与私有化部署调→热门技术】5大模块,内容比传统教材更贴近企业实战!

  • 大量真实项目案例: 带你亲自上手搞数据清洗、模型调优这些硬核操作,把课本知识变成真本事‌!

02适学人群

应届毕业生‌: 无工作经验但想要系统学习AI大模型技术,期待通过实战项目掌握核心技术。

零基础转型‌: 非技术背景但关注AI应用场景,计划通过低代码工具实现“AI+行业”跨界‌。

业务赋能突破瓶颈: 传统开发者(Java/前端等)学习Transformer架构与LangChain框架,向AI全栈工程师转型‌。

image.png

vx扫描下方二维码即可
【附赠一节免费的直播讲座,技术大佬带你学习大模型的相关知识、学习思路、就业前景以及怎么结合当前的工作发展方向等,欢迎大家~】
在这里插入图片描述

本教程比较珍贵,仅限大家自行学习,不要传播!更严禁商用!

03 入门到进阶学习路线图

大模型学习路线图,整体分为5个大的阶段:
图片

04 视频和书籍PDF合集

图片

从0到掌握主流大模型技术视频教程(涵盖模型训练、微调、RAG、LangChain、Agent开发等实战方向)

图片

新手必备的大模型学习PDF书单来了!全是硬核知识,帮你少走弯路(不吹牛,真有用)
图片

05 行业报告+白皮书合集

收集70+报告与白皮书,了解行业最新动态!
图片

06 90+份面试题/经验

AI大模型岗位面试经验总结(谁学技术不是为了赚$呢,找个好的岗位很重要)图片
在这里插入图片描述

07 deepseek部署包+技巧大全

在这里插入图片描述

由于篇幅有限

只展示部分资料

并且还在持续更新中…

真诚无偿分享!!!
vx扫描下方二维码即可
加上后会一个个给大家发

【附赠一节免费的直播讲座,技术大佬带你学习大模型的相关知识、学习思路、就业前景以及怎么结合当前的工作发展方向等,欢迎大家~】
在这里插入图片描述

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐