ICLR 2024 | RLHF有了通用平台和基准,天大开源,专攻现实决策场景

RLHF 通过进修人类偏好,可以在难以手工设想赞美函数的复杂决策工作中进修到正确的赞美引导,得到了很高的关注,在不同情况中选择合适的人类反应典型和不同的进修方法至关重要。然而,当前钻研社区缺乏可以支持这一需求的标准化标注平台和统一基准,量化和比较 RLHF 的最新进展是有挑战性的。本文中,天津大学深度强化进修实验室的钻研团队推出了面向现实决策场景的 Uni-RLHF 平台,这是一个专为 RLHF 量身定制的综合系统实施方案。它旨在根据真实的人类反应提供完整的工作流程,一站式解决实际问题。论文题目:Uni-RLHF:

RLHF 通过进修人类偏好,可以在难以手工设想赞美函数的复杂决策工作中进修到正确的赞美引导,得到了很高的关注,在不同情况中选择合适的人类反应典型和不同的进修方法至关重要。

然而,当前钻研社区缺乏可以支持这一需求的标准化标注平台和统一基准,量化和比较 RLHF 的最新进展是有挑战性的。

本文中,天津大学深度强化进修实验室的钻研团队推出了面向现实决策场景的 Uni-RLHF 平台,这是一个专为 RLHF 量身定制的综合系统实施方案。它旨在根据真实的人类反应提供完整的工作流程,一站式解决实际问题。

ICLR 2024 | RLHF有了通用平台和基准,天大开源,专攻现实决策场景

论文题目:Uni-RLHF: Universal Platform and Benchmark Suite for Reinforcement Learning with Diverse Human Feedback

项目主页:https://uni-rlhf.github.io/

平台链接:https://github.com/pickxiguapi/Uni-RLHF-Platform

算法代码库:https://github.com/pickxiguapi/Clean-Offline-RLHF

论文链接:https://arxiv.org/abs/2402.02423

作者主页:http://yifu-yuan.github.io/

Uni-RLHF 包含三个部分:1)通用多反应标注平台,2)大规模众包反应数据集,3)模块化离线 RLHF 基线代码库

具体流程来看,Uni-RLHF 首先针对各种反应典型开发了用户友好的标注界面,与各种主流 RL 情况兼容。然后建立了一个系统的众包标注流水线,产生了包含 32 个工作、超过 1500 万个时间步的大规模标注数据集。最后,鉴于大规模反应数据集,实现了最先进的 RLHF 算法的基线结果和模块化组件以供其他钻研者应用。

Uni-RLHF 希望通过评价各种设想选择,深入了解它们的优势和潜在的改进领域,构建有价值的开源平台、数据集和基线,以促进鉴于真实人类反应开发更强大、更可靠的 RLHF 解决方案。目前平台、数据集和基线代码库均已开源。

多反应典型通用标注平台

Uni-RLHF 标注平台提供了众包标准标注工作流程:

接口支持多种在线情况 (Online Mode) 和离线数据集 (Offline Mode),并且可以通过简单的接口扩展方式接入定制化的情况;

查询采样器 (Query Sampler) 可决定哪些数据需要被标注,支持多种典型的采样策略;

交互式用户界面 (User Interface) 可让众包查看可用轨迹片段并提供反应响应,提供包含选择、拖动、框选和关键帧捕捉等一系列视频片段和图像标注方式;

反应翻译器 (Feedback Translator) 可将不同的反应标签转换为标准化格式。

ICLR 2024 | RLHF有了通用平台和基准,天大开源,专攻现实决策场景

                               Uni-RLHF 包括平台、数据集和离线 RLHF 基线代码库三个部分

ICLR 2024 | RLHF有了通用平台和基准,天大开源,专攻现实决策场景

                                Uni-RLHF 可以支持大量主流的强化进修情况

适用于强化进修的标准反应编码格式

为了更好地捕捉和利用来自标注者的各种不同典型的反应标签,Uni-RLHF 对一系列相关钻研从事了总结,提出一种标准化的反应编码格式和对应的训练方法。应用者可以根据工作和标注成本需求,选择不同典型的标注方法。一般来说,信息密度越高,标注成本相应也会更大,但是反应效率也会随之提升。

Uni-RLHF 支持以下五种反应典型:

比较反应 (Comparative Feedback):对两段轨迹给出相对性的二元反应比较

属性反应 (Attribute Feedback):对两段轨迹给出鉴于多属性的相对反应比较

评价反应 (Evaluative Feedback):对一段轨迹给出多个级别的评价选项

视觉反应 (Visual Feedback):对一段轨迹中的视觉重点从事选择和标记

关键帧反应 (Keypoint Feedback):对一段轨迹中的关键帧从事捕捉和标记

大规模众包标注流水线

在 RLHF 训练过程中,数据标注是一项复杂的工程问题。钻研职员围绕 Uni-RLHF 构建众包数据正文流水线,通过并行的众包数据正文和过滤,促进大规模正文数据集的创建。

为了验证 Uni-RLHF 平台各方面的易用性和对 RLHF 前沿算法性能从事验证,钻研职员应用广受认可的离线 RL 数据集实现了大规模众包标注工作,以收集反应标签。

在完成数据收集后,钻研职员从事了两轮数据过滤,以尽量减少有噪声的众包数据量,最终建立了一个系统化的众包正文流水线,形成了大规模标注数据集,包括 32 个主流工作中的 1,500 多万个时间步。

ICLR 2024 | RLHF有了通用平台和基准,天大开源,专攻现实决策场景

                                          标注流水线中每个组件的验证

为了证明数据过滤的有效性。钻研职员首先在 SMARTS 中抽取了 300 个轨迹片段从事专家正文,称为「Oracle」。接下来,钻研职员请了五位众包在三种不同的设置下分别标注 100 条轨迹。「Naive」意味着只能看到工作描述,「Example」允许查看专家提供的五个正文样本和详细分析,而「Filter」则添加了过滤器。

以上实验结果表明,每个组件都显著提高了标注的可靠性,最终实现了与专家正文 98% 的一致率。

离线 RLHF 基准实验

钻研职员利用收集到的众包反应数据集对下游决策工作从事了大量实验,以评价各种不同的设想选择及其对应的优势。

首先,Uni-RLHF 应用了三种不同的赞美模型设想结构,分别是 MLP、TFM (Transformer) 和 CNN,其中 MLP 结构便于处理向量输入,而 CNN 结构便于处理图像输入。TFM 赞美结构则可以更好地拟合 non-Markovian 赞美。同时Uni-RLHF 应用了三种广泛应用的离线强化进修算法作为底座,包括 IQL、CQL 和 TD3BC。

Oracle 代表应用手工设想的工作赞美训练的模型;CS (CrowdSource) 代表一种是通过 Uni-RLHF 系统众包获得的众包标签;而 ST (Script Teacher) 代表根据实际工作赞美生成的合成标签,可视为专家标签供比较。

ICLR 2024 | RLHF有了通用平台和基准,天大开源,专攻现实决策场景

实验结论表明:

鉴于 IQL 基线效果最稳定,并且比较 IQL-CS 可以表现出和 IQL-Oracle 相当的优异性能,而 CQL 基线偶尔会出现策略崩溃的结果;

总体看来,TFM 结构在稳定性和性能两方面均领先于 MLP 结构,尤其是在稀疏赞美设置的情况中;

和合成标签 (ST) 相比,众包标签 (CS) 在大多数情况中可以达到相当甚至超越的效果,这也证明了 Uni-RLHF 具有高质量的数据标注。

在图像输入的情况中,众包标签 (CS) 则全面领先于合成标签 (ST),钻研职员认为这种优异表现来源于人类可以更敏感的捕捉到游戏过程中的细节过程,这些细节则很难用简单的积分赞美来概括。

ICLR 2024 | RLHF有了通用平台和基准,天大开源,专攻现实决策场景

RLHF 方法是否能在真实的复杂工作上成功替代手工设想的赞美函数?钻研职员应用了 NeurIPS 2022 中 SMARTS 自动驾驶竞赛的情况,该情况提供了相对真实和多样化的自动驾驶场景,并应用成功率,速度和舒适度等多个指标评价模型的性能。其中,冠军方案针对该工作设想赞美函数会经过多次试错,并在多次训练过程中不断调整完善各项赞美时间及系数,最终形成了以下极为复杂的赞美函数构成,设想成本极高:

ICLR 2024 | RLHF有了通用平台和基准,天大开源,专攻现实决策场景

                                  自动驾驶场景赞美函数设想

而通过众包标注的简单反应标签从事赞美函数训练,Uni-RLHF 就可以达到超越专家赞美的工作成功率,并且在舒适度指标上也有所领先。

ICLR 2024 | RLHF有了通用平台和基准,天大开源,专攻现实决策场景

                                   针对 SMARTS 自动驾驶场景的多指标评测

ICLR 2024 | RLHF有了通用平台和基准,天大开源,专攻现实决策场景

                                Uni-RLHF 方法和 Oracle 赞美函数对比。(左:Oracle,右:Uni-RLHF)

Uni-RLHF 还针对其他多种典型的反应形式从事了更多验证,这里以多属性反应 (Attribute Feedback) 举一个简单的例子:用户希望训练一个 Walker 机器人,使其速度和躯干高度在运动的过程中从事自由的变化,而不是简单的最大化速度。此时简单的比较反应就很难准确的表述用户的偏好,Uni-RLHF 则提供了针对多属性反应的标注模式。在本实验中,Walker 会运行 1000 步,并每 200 步调整姿态,速度的属性值设定为 [慢,快,中,慢,快],高度的属性值设定为 [高,中,高,低,高]。从曲线和相应的视频中可以清楚地观察到经过 Uni-RLHF 标注后训练的模型可以灵活的从事姿态转换。

ICLR 2024 | RLHF有了通用平台和基准,天大开源,专攻现实决策场景

                            Walker 遵循用户偏好从事灵活姿态转换

总结和未来展望

Uni-RLHF 展示了在决策工作中鉴于 RLHF 方法取代手工设想赞美函数的重要前景,钻研职员希望通过建设平台、大规模数据集和代码库以促进更加可靠,鉴于真实人类反应标注的 RLHF 解决方案。该领域仍存在一些挑战和可能的未来方向:

评价人类的非理性和偏向性:众包提供反应标签势必会带来反应标签的噪音,即对工作认知不统一、标注错误、有偏向性等问题,如何在嘈杂的标签数据中从事进修是值得钻研的方向。

不完美赞美函数修正:反应标签的噪音和数据分布狭窄等问题会导致进修到次优的赞美函数、如何鉴于赞美塑形、先验知识等进一步的鉴于该赞美函数从事修正也是重要的钻研问题。

多反应典型的组合作用:尽管目前的钻研已经证实,应用更细粒度的反应方式会给进修效率带来巨大提升,但在同一个工作中聚合图像、评价、关键帧等各典型反应方式依然值得进一步钻研。

给TA打赏
共{{data.count}}人
人已打赏
工程

CNN、Transformer、Uniformer之外,我们终于有了更高效的视频懂得技术

2024-3-25 11:19:00

工程

两分钟1200帧的长视频生成器StreamingT2V来了,代码将开源

2024-3-28 11:45:00

0 条回复 A文章作者 M管理员
    暂无讨论,说说你的看法吧
个人中心
今日签到
搜索