最近,DeepSeek在国际上很火,一方面其效果比肩非开源模型ChatGPT,另一方面其运行成本很低。可以说,DeepSeek是目前为止最先进的大模型之一。
今天这篇文章,就带大家梳理一下从2024年至今DeepSeek发布的V1版本到V3版本的核心技术点,看看DeepSeek采用了哪些技术构建了最先进的大模型,每一次的升级基本都伴随着训练数据的优化、模型结构的升级、优化方式的更新。
为了方便大家阅读,本文会用比较简洁的语言进行主要技术的介绍,一些其他相关的技术细节深度解析,后续也会陆续进行更新。
1.DeepSeek V1
DeepSeek V1是2024年1月份发布的第一版DeepSeek模型,包含DeepSeek的核心构建方式,核心技术点分为数据端、模型端、优化端、对齐4个部分,前面3个部分处于模型的预训练阶段,对齐阶段使用SFT进行人类风格对齐。
数据端:在数据的处理上,包括去重、过滤、混合3个步骤,目的是构建一个多样性强、纯净的高质量预训练数据。在去重阶段,对于Common Crawl数据集进行全局的去重过滤,可以提升去重比例。在过滤阶段,构建了一套详细的包括文法语法分析在内的评估流程,去除低质量数据。在混合阶段,对不同domain的数据进行采样,平衡不同domain数据量,让数据量较少的domain也能有足够的样本占比,提升数据集多样性和全面性。
此外,在数据处理方面,使用Byte-level Byte-Pair Encoding (BBPE)作为tokenizer,相比BPE是在字符粒度进行字符串分割,BBPE在字节粒度进行处理,整体算法逻辑和BPE类似。
整体参与预训练的token数量为2 trillion。在V2和V3中,训练的token数量不断上升,V2为8 trillion,V3为14 trillion。
模型端:模型的主体结构基本沿用LLaMA。LLaMA主体就是Transformer结构,主要差异包括RMSNorm的Pre-normalization(每层Transformer输入使用RMSNorm进行归一化)、激活函数采用SwiGLU、位置编码采用Rotary Embeddings。模型包括7B和67B两种尺寸,67B尺寸的Transformer中的attention采用了Grouped Query Attention代替最普通的self-attention降低inference开销。Grouped Query Attention每组query共用同一组key和value。
优化端:使用multi-step learning rate代替LLaMA中的cosine learning rate schedule,主要原因是实验发现两者虽然最终收敛到的loss差不多,但是前者在连续学习上loss能够保证一致性,连续学习更加方便。先用2000个step的warmup将学习率提升到最大值,然后在训练完80%的训练数据后将学习率降低到31.6%,在训练完90%的训练数据后进一步降低到10%。
对齐:使用Supervised Fine-Tuning、DPO两种方式进行预训练模型的finetune,进行风格对齐。Supervised Fine-Tuning使用120w搜集到的SFT数据(一些根据指令给出答案的文本,由人类标注的高质量数据,帮助预训练模型迁移人类风格)进行finetune。DPO是针对之前ChatGPT中基于强化学习的RHLF风格迁移的一种升级,不用强化学习,只使用一个指定对应的两个答案之前的相对偏好关系作为损失函数加入到模型中。
2.DeepSeek V2
DeepSeek V2最核心的2点改动都在模型结构上,一个是提出了一种Multi-head Latent Attention提升了inference效率;另一个是构建了基于DeepSeekMoE的语言模型结构。
Multi-head Latent Attention:MLA的主要目的是减少KV缓存占用的空间。KV缓存是大模型都会使用的技术,在inference阶段,每一个token的输出都要和历史所有token计算attention,每次新增token都有很多重复计算,因此可以将前面token计算出的key和value缓存起来。但是直接缓存key和value占用较大的空间,因此MLA对KV进行了一个低维映射,只存储这个低维的向量,节省了缓存存储空间。
DeepSeekMoE:MoE是目前大模型在探索应用的一项技术,基础的MoE将原来的每个token的单个FFN层变成多个并行的FFN层(对应多个expert),并根据输入生成一个路由到各个FFN的打分,选择topN个专家,实现在单token运算量没有显著提升的前提下,扩大模型的参数空间的目的。如下图a中,即是一个激活2个专家的MoE。
而DeepSeekMoE相比MoE有2个核心优化。一个是把Expert变多了(文中称为Fine-Grained Expert),其实就是把原来每个Expert的FFN维度调小,增加Expert数量,并且最终激活的Expert数量也变多。另一个就是增加了几个所有token都走的公用Expert,通过让所有token都走这些Expert,让这些Expert提取通用信息,其他Expert就能更专注于提取差异化的信息。
3.DeepSeek V3
DeepSeekV3在模型结构上的核心优化,一方面是对DeepSeekMoE中的多专家负载均衡问题,提出了一种不需要辅助loss就能实现的解决方案,相比使用辅助loss提升了模型性能;另一方面是引入了Multi-Token Prediction技术,相比原来每次只能预测一个token,显著提升了infer的速度。
Auxiliary-Loss-Free Load Balancing:MoE的一个核心问题是有可能会出现坍缩问题,即训练到最后,激活的总是那么几个Expert,没有实现各个Expert的均衡,从而失去了多专家的意义。一般的解决方法会显示引入一个负载均衡loss(DeepSeekV2,以及一些其他MoE的做法),但是显示引入一个和目标不相关的loss会影响训练效果。因此,DeepSeek V3采用了无需loss的负载均衡方法,在每个Expert打分增加一个这个相应的bias项,bias项只影响路由不影响后续的Expert加权求和计算,每个step都会监控各个Expert的负载均衡情况,对于过载的Expert降低bias项减少其相应的激活数量,对于比较稀疏的Expert增大其bias项提升激活其的样本比例。
Multi-Token Prediction:语言模型都是逐个预测的,每次将当前预测结果作为最新的一个输入,再次预测下一个。改成多token预测,一方面可以显著提升infer的速度,另一方面也可以让模型在生成后续token的时候有一个全局性,提升生成效果,对训练数据利用的也更加充分,加速收敛。
具体做法为,在训练阶段,除了原来的主模型外,还会有几个并行的MTP模块,这些MTP模块的Embedding层和Output Head和主模型共享,内部有一个Transformer层。在主模型预测了next token后,将这个预测token的表征和之前token的Embedding拼接到一起,生成一个新的输入(超出长度的更久远的token被才减掉)。这个拼接好的Embedding输入到第一个MTP中预测next next token。以此类推,将MTP Module1的当前预测token表征和历史token拼接到一起,作为MTP Module2的输入,再预测next next next token。
文中引入Multi-Token Prediction主要为了提升训练效果,inference阶段直接去掉这些MTP模块,也可以引入这些MTP模块提升inference效率。
如何学习大模型 AI ?
由于新岗位的生产效率,要优于被取代岗位的生产效率,所以实际上整个社会的生产效率是提升的。
但是具体到个人,只能说是:
“最先掌握AI的人,将会比较晚掌握AI的人有竞争优势”。
这句话,放在计算机、互联网、移动互联网的开局时期,都是一样的道理。
我在一线互联网企业工作十余年里,指导过不少同行后辈。帮助很多人得到了学习和成长。
我意识到有很多经验和知识值得分享给大家,也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑,所以在工作繁忙的情况下还是坚持各种整理和分享。但苦于知识传播途径有限,很多互联网行业朋友无法获得正确的资料得到学习提升,故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。
第一阶段(10天):初阶应用
该阶段让大家对大模型 AI有一个最前沿的认识,对大模型 AI 的理解超过 95% 的人,可以在相关讨论时发表高级、不跟风、又接地气的见解,别人只会和 AI 聊天,而你能调教 AI,并能用代码将大模型和业务衔接。
- 大模型 AI 能干什么?
- 大模型是怎样获得「智能」的?
- 用好 AI 的核心心法
- 大模型应用业务架构
- 大模型应用技术架构
- 代码示例:向 GPT-3.5 灌入新知识
- 提示工程的意义和核心思想
- Prompt 典型构成
- 指令调优方法论
- 思维链和思维树
- Prompt 攻击和防范
- …
第二阶段(30天):高阶应用
该阶段我们正式进入大模型 AI 进阶实战学习,学会构造私有知识库,扩展 AI 的能力。快速开发一个完整的基于 agent 对话机器人。掌握功能最强的大模型开发框架,抓住最新的技术进展,适合 Python 和 JavaScript 程序员。
- 为什么要做 RAG
- 搭建一个简单的 ChatPDF
- 检索的基础概念
- 什么是向量表示(Embeddings)
- 向量数据库与向量检索
- 基于向量检索的 RAG
- 搭建 RAG 系统的扩展知识
- 混合检索与 RAG-Fusion 简介
- 向量模型本地部署
- …
第三阶段(30天):模型训练
恭喜你,如果学到这里,你基本可以找到一份大模型 AI相关的工作,自己也能训练 GPT 了!通过微调,训练自己的垂直大模型,能独立训练开源多模态大模型,掌握更多技术方案。
到此为止,大概2个月的时间。你已经成为了一名“AI小子”。那么你还想往下探索吗?
- 为什么要做 RAG
- 什么是模型
- 什么是模型训练
- 求解器 & 损失函数简介
- 小实验2:手写一个简单的神经网络并训练它
- 什么是训练/预训练/微调/轻量化微调
- Transformer结构简介
- 轻量化微调
- 实验数据集的构建
- …
第四阶段(20天):商业闭环
对全球大模型从性能、吞吐量、成本等方面有一定的认知,可以在云端和本地等多种环境下部署大模型,找到适合自己的项目/创业方向,做一名被 AI 武装的产品经理。
- 硬件选型
- 带你了解全球大模型
- 使用国产大模型服务
- 搭建 OpenAI 代理
- 热身:基于阿里云 PAI 部署 Stable Diffusion
- 在本地计算机运行大模型
- 大模型的私有化部署
- 基于 vLLM 部署大模型
- 案例:如何优雅地在阿里云私有部署开源大模型
- 部署一套开源 LLM 项目
- 内容安全
- 互联网信息服务算法备案
- …
学习是一个过程,只要学习就会有挑战。天道酬勤,你越努力,就会成为越优秀的自己。
如果你能在15天内完成所有的任务,那你堪称天才。然而,如果你能完成 60-70% 的内容,你就已经开始具备成为一名大模型 AI 的正确特征了。