GPT助手的训练流程四个主要阶段( GPT Assistant training pipeline )

GPT助手的训练流程四个主要阶段( GPT Assistant training pipeline )

flyfish

四个阶段

预训练(pre-training)

监督微调(supervised fine tuning, SFT)

奖励建模(reward modeling)

强化学习(reinforcement learning)

每个阶段又分为三个部分(从上到下):数据集、算法和输出的模型。

翻译图,最后附上原图

GPT助手(GPT Assistant)的训练流程,分为四个主要阶段:预训练(Pretraining)、有监督微调(Supervised Finetuning)、奖励建模(Reward Modeling)和强化学习(Reinforcement Learning)。每个阶段的详细内容如下:

一. 预训练阶段(Pretraining Stage)

  1. 数据集(Dataset):
    原始互联网数据(Raw Internet Data) :使用来自互联网的原始文本数据,数据量达到万亿级别的单词(tokens)。这些数据量巨大但质量较低,涵盖了各种类型的内容。

  2. 算法(Algorithm):
    语言建模(Language Modeling) :模型通过预测文本序列中的下一个token来进行训练。这是标准的自回归语言模型训练方式。

  3. 模型(Model):
    基础模型(Base Model) :经过预训练的基础语言模型,例如GPT、LLaMA、PaLM等模型,经过数月使用成千上万块GPU训练而成。这些模型可以独立部署并用于各种任务。

  4. 备注(Notes):
    使用了数千块GPU,耗费数月时间训练 。这些基础模型是诸如GPT、LLaMA、PaLM等。

二. 有监督微调阶段(Supervised Finetuning Stage)

  1. 数据集(Dataset):
    展示数据(Demonstrations): 由人工编写的理想助手响应数据集,包含约10到10万个问题及其响应。这些数据量相对较少,但质量非常高。

  2. 算法(Algorithm):
    语言建模(Language Modeling) :继续通过语言建模的方式,对模型进行微调,专注于学习这些高质量的问答对。

  3. 模型(Model):
    微调模型(SFT Model): 在基础模型的基础上,通过有监督微调训练得到的模型,例如Vicuna-13B。这些模型可以用于更具体的任务并部署。

  4. 备注(Notes):
    使用了1到100块GPU,耗时数天进行训练 。例如Vicuna-13B模型,经过微调后可以部署。

三. 奖励建模阶段(Reward Modeling Stage)

  1. 数据集(Dataset):
    比较数据(Comparisons): 使用由人工编写的10万到100万个回答的好坏对比数据。这些数据量虽然较少,但质量非常高,用于训练模型区分高质量和低质量的回答。

  2. 算法(Algorithm):
    二值分类(Binary Classification): 训练奖励模型,预测根据用户偏好进行奖励的结果。该模型用来评估和指导模型生成更好的回答。

  3. 模型(Model):
    奖励模型(RM Model): 通过二值分类训练得到的奖励模型,它不能独立部署,主要用于强化学习阶段的模型优化。

  4. 备注(Notes):
    使用了1到100块GPU,耗时数天进行训练 。此模型用于指导强化学习阶段的训练。

四. 强化学习阶段(Reinforcement Learning Stage)

  1. 数据集(Dataset):
    提示数据(Prompts): 使用由人工编写的约1到10万个提示语句进行训练。数据量较少但质量很高,帮助模型在特定上下文中生成更优质的响应。

  2. 算法(Algorithm):
    强化学习(Reinforcement Learning): 使用奖励模型指导生成token,通过生成使奖励最大化的token序列来优化模型的表现。

  3. 模型(Model):
    强化学习模型(RL Model): 经过强化学习训练的最终模型,初始化自微调模型,并利用奖励模型进行优化。这些模型可以在实际应用中部署,如ChatGPT、Claude等。

  4. 备注(Notes):
    使用了1到100块GPU,耗时数天进行训练 。最终模型例如ChatGPT、Claude,可以进行部署和实际应用。

原图

相关推荐
算法channel15 小时前
自己电脑搭建个人知识库,一般电脑也能玩(支持通义千问、GPT等)。
gpt
DongHappyyy15 小时前
gpt优化事件处理速度
python·gpt·numpy
曼城周杰伦2 天前
自然语言处理:第八十三章 Prompt格式到底多重要?
人工智能·gpt·自然语言处理·langchain·nlp·prompt·easyui
bylander2 天前
【AI学习】2024年末一些AI总结的摘录
人工智能·gpt·学习
量子位2 天前
潞晨尤洋:视频生成的 GPT-4 时刻,3 年后可以见证 | MEET 2025
人工智能·gpt
w_juesi2 天前
GPT分区 使用parted标准分区划分,以及相邻分区扩容
数据库·gpt
2301_764602232 天前
I.MX6ULL-GPT实现延时
gpt
绿萢萢t6212382 天前
智享AI三代直播软件,以GPT为刃,斩断传统直播束缚
人工智能·gpt·aigc
hao_wujing2 天前
什么是 GPT?Transformer 工作原理的动画展示
gpt·深度学习·transformer
电手3 天前
GPT与MBR之间的差异是什么?
gpt·mbr和gpt区别