chatGPT训练过程

news2024/10/4 1:19:12

强化学习基础

强化学习是指智能体在不确定环境中最大化其获得的奖励从而达到自主决策的目的。其执行过程为：智能体依据策略决策从而执行动作，然后感知环境获取环境的状态，进而得到奖励(以便下次再到相同状态时能采取更优的动作)，然后再继续按此流程“依据策略执行动作-感知状态–得到奖励”循环进行。

在这里插入图片描述

【强化学习是一种无监督学习】强化学习没有标签告诉它在某种情况下应该做出什么样的行为，只有一个做出一系列行为后最终反馈回来的reward，然后判断当前选择的行为是好是坏。某一状态的价值函数等于即时奖励+折扣因子×后续一系列状态的奖励。

基于值函数的方法

通过求解一个状态或者状态下某个动作的估值为手段，从而寻找最佳的价值函数，找到价值函数后，再提取最佳策略。【总结】通过做出动作后，找最佳的价值函数，提取最佳策略。

基于策略的方法

一般先进行策略评估，即对当前已经搜索到的策略函数进行估值，得到估值后，进行策略改进，不断重复这两步直至策略收敛。【总结】对策略进行估值，优化使策略估值最大化。

马尔科夫

当且仅当某时刻的状态只取决于上一时刻的状态时，一个随机过程被称为具有马尔可夫性质。具备马尔科夫性质的随机过程称为马尔科夫过程。

马尔科夫奖励

在马尔科夫过程的基础上加入奖励函数和折扣因子就变为了马尔科夫奖励过程。

【奖励函数】某个状态s的奖励是指转移到该状态s时可以获得的奖励期望。

【回报】其实某一状态获得的奖励是持久的，因为当前状态导致下一状态获得的奖励乘以折扣因子就是持久奖励。回报=当前奖励+持久奖励。

【状态价值】某一状态可以获得回报的期望。

在这里插入图片描述

马尔科夫决策过程

在马尔科夫奖励过程中加入来自外界的刺激如智能体的动作，就得到了马尔科夫决策过程。

【状态价值函数】某一状态可以获得奖励的期望乘以该状态的价值。

【动作价值函数】在某一状态下采取某一动作所获得奖励的期望。—Q函数

强化学习的分类

基于模型的强化学习

可以简单的使用动态规划求解，任务可定义为预测和控制，预测的目的是评估当前策略的好坏，即求解状态价值函数。

无模型的强化学习

基于价值的强化学习，其会学习并贪婪的选择奖励值最大的动作。

基于策略的强化学习，其对策略进行建模和优化。

重要性采样

假设有一个函数，x需要从分布p中采样数据；但是当不能从分布p中采样数据而只能从另一个分布q中采用数据时，需要做一些变换。

在这里插入图片描述

如此便可以从分布q中采样x，再进行计算。

而异策略就是基于重要性采样的原理实现的。

策略学习

【策略梯度更新的思想】参数为θ的策略πθ接受状态s，输出动作概率分布，在动作概率分布中采样动作，执行动作(形成运动轨迹)，得到奖励r，跳到下一个状态。在这样的步骤下，可以使用策略π收集一批样本，然后使用梯度下降算法学习这些样本，不过当策略π的参数更新后，这些样本不能继续被使用，还要重新使用策略π与环境互动收集数据。

TRPO：加进KL散度解决两个分布相差大的问题

在目标函数里面加入了约束。TRPO的问题在于把 KL 散度约束当作一个额外的约束，没有放在目标里面，导致TRPO很难计算

PPO(相对TRPO减少了计算量)

通过KL散度(相对熵)加入惩罚项、截断的方式使得更新的策略不与原策略相差太大。

RLHF:基于人类偏好的强化学习

1.首先，智能体的一对1-2秒的行为片段定期地回馈给人类操作员，人类基于偏好对智能体的行为作出某种偏好性的选择评判。

2.接着，人类这种基于偏好的选择评判被预测器来预测奖励函数。

3.智能体通过预测器预测出的奖励函数作出更优的行为。

ChatGPT的训练过程

第一阶段：利用人类的问答数据对GPT3微调进行有监督训练出SFT模型（作为baseline）。

1.一共进行了16个epochs的训练。

2.采用了余弦学习率衰减策略。

3.残差丢弃率。

第二阶段：通过RLHF的思路训练一个奖励模型RM。

首先使用第一阶段训练的SFT模型初始化第二阶段的RM模型。针对每个问题收集4-9个不同的回答，人工对这些回答的好坏进行标注且排序，排序的结果来训练一个RM模型，使模型从排序数据中理解人类的偏好。

不同回答两两组合，计算奖励差值。

第三阶段：通过最大化奖励函数的目标下，通过PPO算法继续微调GPT4模型。

首先使用第一阶段训练的SFT模型初始化一个PPO模型，使用不带任何人工标注的数据集训练，使用第二阶段训练的RM奖励模型去给PPO模型的预测结果进行打分和排序。之后通过奖励最大化优化PPO模型的策略参数，PPO算法限制策略更新范围。

总目标函数

在这里插入图片描述

第一部分是使奖励最大化，利用重采样的思想展开如下所示：

在这里插入图片描述

第二部分是惩罚项，目的是不让新学习到的策略函数偏离baseline策略SFT太多。

第三部分是偏置项，防止训练出的模型过于讨好人类偏好，而不根据问题回答答案。(个人理解可以是防止数据集中的噪声带来干扰)

从GPT1到GPT2

虽然GPT1的预训练加微调的范式仅需要少量的微调和些许的架构改动，但能不能有一种模型完全不需要对下游任务进行适配就可以表现优异？GPT2便是在往这个方向努力：不微调但给模型一定的参考样例以帮助模型推断如何根据任务输入生成相应的任务输出。

针对【小样本/零样本】分为三种：零样本学习(是指在没有任何样本/示例情况下，让预训练语言模型完成特定任务)、单样本学习(是指在只有一个样本/示例的情况下，预训练语言模型完成特定任务)、少样本学习(是指在只有少量样本/示例的情况下，预训练语言模型完成特定任务)。【注】零样本、单样本、少样本并没有被模型去学习和微调，模型学习了样本输入输出的分布。

prompt learning

让模型逐步学会人类的各种自然指令，而不用根据下游任务去微调模型或更改模型的参数，直接根据指令去干活，这个指令就叫做prompt。

指令微调技术(IFT)

IFT的数据通常是由人工手写指令和语言模型引导的指令实例的集合，这些指令数据由三个主要组成部分组成：指令、输入和输出，对于给定的指令，可以有多个输入和输出实例。

关于「prompt learning」最简单粗暴的理解，其实就是让模型逐步学会人类的各种自然指令或人话，而不用根据下游任务去微调模型或更改模型的参数，直接根据人类的指令直接干活，这个指令就是prompt，而设计好的prompt很关键也需要很多技巧，是一个不算特别小的工程，所以叫prompt engineering。