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)
-
数据集(Dataset):
原始互联网数据(Raw Internet Data) :使用来自互联网的原始文本数据,数据量达到万亿级别的单词(tokens)。这些数据量巨大但质量较低,涵盖了各种类型的内容。 -
算法(Algorithm):
语言建模(Language Modeling) :模型通过预测文本序列中的下一个token来进行训练。这是标准的自回归语言模型训练方式。 -
模型(Model):
基础模型(Base Model) :经过预训练的基础语言模型,例如GPT、LLaMA、PaLM等模型,经过数月使用成千上万块GPU训练而成。这些模型可以独立部署并用于各种任务。 -
备注(Notes):
使用了数千块GPU,耗费数月时间训练 。这些基础模型是诸如GPT、LLaMA、PaLM等。
二. 有监督微调阶段(Supervised Finetuning Stage)
-
数据集(Dataset):
展示数据(Demonstrations): 由人工编写的理想助手响应数据集,包含约10到10万个问题及其响应。这些数据量相对较少,但质量非常高。 -
算法(Algorithm):
语言建模(Language Modeling) :继续通过语言建模的方式,对模型进行微调,专注于学习这些高质量的问答对。 -
模型(Model):
微调模型(SFT Model): 在基础模型的基础上,通过有监督微调训练得到的模型,例如Vicuna-13B。这些模型可以用于更具体的任务并部署。 -
备注(Notes):
使用了1到100块GPU,耗时数天进行训练 。例如Vicuna-13B模型,经过微调后可以部署。
三. 奖励建模阶段(Reward Modeling Stage)
-
数据集(Dataset):
比较数据(Comparisons): 使用由人工编写的10万到100万个回答的好坏对比数据。这些数据量虽然较少,但质量非常高,用于训练模型区分高质量和低质量的回答。 -
算法(Algorithm):
二值分类(Binary Classification): 训练奖励模型,预测根据用户偏好进行奖励的结果。该模型用来评估和指导模型生成更好的回答。 -
模型(Model):
奖励模型(RM Model): 通过二值分类训练得到的奖励模型,它不能独立部署,主要用于强化学习阶段的模型优化。 -
备注(Notes):
使用了1到100块GPU,耗时数天进行训练 。此模型用于指导强化学习阶段的训练。
四. 强化学习阶段(Reinforcement Learning Stage)
-
数据集(Dataset):
提示数据(Prompts): 使用由人工编写的约1到10万个提示语句进行训练。数据量较少但质量很高,帮助模型在特定上下文中生成更优质的响应。 -
算法(Algorithm):
强化学习(Reinforcement Learning): 使用奖励模型指导生成token,通过生成使奖励最大化的token序列来优化模型的表现。 -
模型(Model):
强化学习模型(RL Model): 经过强化学习训练的最终模型,初始化自微调模型,并利用奖励模型进行优化。这些模型可以在实际应用中部署,如ChatGPT、Claude等。 -
备注(Notes):
使用了1到100块GPU,耗时数天进行训练 。最终模型例如ChatGPT、Claude,可以进行部署和实际应用。