RLHF(从人类反馈中进行强化学习)详解(三)

在经过了前两节的内容学习之后,我们对于RLHF(从人类反馈中进行强化学习)有了比较深入的认知,并且初步了解了RLHF中偏好数据集的引入,奖励模型的设置以及baseLLM的训练过程。在本节的学习中,我们将深入LLM的tune步骤,了解LLM的微调工作。

ML流水线

如上图所示,在我们想要批量实现机器学习的任务时,我们可以开发一条机器学习流水线,将其中的每个步骤封装成通用的类,在需要时进行调用即可。比如该图中就包括创建数据集,训练模型等步骤。该工作流图,橙色框表示工作中用到的一些组件,包含通用的代码框架,而蓝色框则表示组件生成的工件,比如生成的模型和指标可以帮助我们评估模型。

RLHF流水线

该图为RLHF工作流的基本框架,包含创建偏好数据集、训练奖励模型、创建prompt数据集以及用RL微调大模型等组件。在各组件共同配合下,能完成一个复杂的RLHF任务。

简单来说,就是创建两个数据集,训练两个模型。

  • 40
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值