Transformer 架构详解
一、Transformer 的诞生
1.1 发展时间线
| 时期 | 阶段 | 描述 |
|---|---|---|
| 2014-2016 | RNN/LSTM 时代 | 序列模型主导,但存在长序列处理困难 |
| 2017 | Transformer 诞生 | Vaswani 团队发表《Attention Is All You Need》 |
| 2018-2019 | BERT & GPT 兴起 | 基于 Transformer 的预训练模型大获成功 |
| 2020-现在 | 大语言模型时代 | GPT-3, GPT-4, ChatGPT 等改变世界 |
1.2 三大突破点
1. 注意力机制替代循环计算
- 彻底摆脱了 RNN 的串行处理限制
- 并行处理,速度提升数百倍
- 能够关联文本中任意距离的词语
- 更好地理解长文本上下文
- 为后续大模型提供了坚实基础
- 支持千亿、万亿参数模型
二、核心组件
2.1 嵌入层(Embedding)
文字变数字的魔法工厂- 词元分割:把"我爱北京"拆成["我", "爱", "北京"],类似把句子拆成乐高积木
- 向量转换:每个词元转为100-1024维的数字向量,捕捉语义信息
- 语义捕捉:相似词的向量距离近,体现语义关系
2.2 Transformer 块
信息处理的智能工厂- 注意力机制:让每个词"听"到其他所有词的信息,实现全局感知
- MLP层:对注意力输出进行非线性变换和特征提取
- 残差连接:帮助梯度传播,使深层网络训练更稳定
2.3 输出概率层
答案揭晓的"投票站"- 线性层:将向量转换为词表长度的维度(如512维 → 50000维)
- Softmax函数:将数值转为0-1的概率,所有概率和为1
- 概率选择:根据概率分布选择最可能的下一个词
三、预测下一个词的底层逻辑
3.1 预测过程详解
1. 输入文本分析
- 模型接收输入文本"今天天气",开始分析每个词之间的关系
- 将文本分割为词元,每个词转换为数字向量
- 计算"今天"与"天气"的关联度,理解上下文语义
- 注意力机制评估词语间的相关性强度
- 基于上下文分析,为可能的下一个词生成概率分布
- 使用 softmax 函数将得分转换为概率
- 选择概率最高的词作为预测结果
- 可以使用贪心搜索或采样方法选择结果
3.2 类比理解
- 古诗猜测类比:就像根据"床前明月光"预测下句是"疑是地上霜"
- 规律学习:Transformer 通过学习海量文本的词语关联规律
- 概率猜词:实现更精准的"猜词"游戏
四、Transformer 的优缺点
4.1 三大优势
| 优势 | 描述 | 类比理解 |
|---|---|---|
| 并行加速 | 处理1000词的句子时速度快数百倍,充分利用GPU并行计算 | RNN像排队买票,Transformer像10个窗口同时售票 |
| 长距离记忆 | 能直接关联文本中任意距离的词语 | 处理"三年前小明在巴黎买的书"时,能直接关联"巴黎"和"书" |
| 万能适配 | 同一架构可适用于翻译、生成、问答等不同NLP任务 | 就像同一台机器既能生产汽车零件,也能生产家电零件 |
4.2 三大挑战
| 挑战 | 描述 | 影响 |
|---|---|---|
| 计算量大 | 注意力机制计算复杂度为O(n²),序列越长计算量越大 | 处理1000词需计算100万次关联,限制了处理超长文本的能力 |
| 数据饥渴 | 训练基础模型需数亿单词语料(相当于10万本小说) | 小数据集上表现可能不如传统方法 |
| 位置感知缺失 | 原生不理解词语位置顺序,需额外的位置编码 | 处理位置敏感任务时需要特别设计 |
五、从 Transformer 到大语言模型
5.1 演进时间线
| 年份 | 模型 | 参数量 | 能力 |
|---|---|---|---|
| 2017 | Original Transformer | 65M | 机器翻译、基础文本生成 |
| 2018 | BERT | 340M | 文本理解、情感分析、问答系统 |
| 2019 | GPT-2 | 1.5B | 流畅文本生成、零样本学习 |
| 2020 | GPT-3 | 175B | 代码生成、创意写作、复杂推理 |
| 2023 | GPT-4 | 1.8T | 多模态理解、专业级问答、复杂任务 |
5.2 三大改进方向
1. 规模升级
- 从基础 Transformer 的 1 亿参数,扩展到 GPT-4 的 1.8 万亿参数
- 相当于从"小型图书馆"升级为"国家图书馆"
- 自监督学习:遮住句子中的词让模型猜测
- 人类反馈微调(RLHF):让模型更懂人话
- 指令微调:提升任务执行能力
- GPT-4 能解释图片中的梗
- 支持文本、图像、音频输入
- 跨模态推理和生成能力
来源:Easy AI 教程系列 #EasyAI #AI教学 #教程 #Transformer