RLHF-Reward-Modeling 学习资料汇总 - 训练RLHF奖励模型的开源工具包

Ray

RLHF-Reward-Modeling: 训练RLHF奖励模型的开源工具包

RLHF-Reward-Modeling是一个专门用于训练强化学习中人类反馈(RLHF)奖励模型的开源项目。该项目由RLHFlow团队开发维护,提供了多种先进的奖励模型实现和训练方法,旨在帮助研究人员和开发者更好地捕捉人类偏好,从而改进RLHF的效果。

主要特点

  • 提供多种奖励模型实现,包括ArmoRM、配对偏好模型和Bradley-Terry模型等
  • 支持大规模语言模型训练,如Gemma-7B等
  • 在RewardBench基准测试中取得了SOTA性能
  • 提供详细的安装说明和数据集准备指南
  • 包含多个评估脚本和实用工具

核心模型

  1. ArmoRM (Absolute-Rating Multi-Objective Reward Model)

    • 在RewardBench上排名第一的8B模型
    • 利用多目标奖励建模和专家混合实现可解释性偏好
  2. 配对偏好模型

    • 基于SliC-HF方法实现
    • 在RewardBench上表现优异
  3. Bradley-Terry奖励模型

    • 经典的比较建模方法
    • 在某些任务上表现出色,如聊天任务

RLHF奖励模型架构

使用指南

  1. 安装 推荐为Bradley-Terry奖励模型和配对偏好模型创建单独的环境。详细安装说明请参考相应文件夹。

  2. 数据集准备 数据集需要预处理成标准格式,每个样本包含"chosen"和"rejected"两个对话,共享相同的prompt。

  3. 模型训练 提供了多种训练脚本,支持不同的模型架构和训练策略。

  4. 评估 可以使用RewardBench提供的数据集评估训练好的奖励模型。

相关资源

RLHF-Reward-Modeling为RLHF研究和应用提供了强大的工具支持。无论是想深入研究RLHF奖励建模,还是将其应用到实际项目中,这个开源项目都是一个很好的起点。欢迎探索使用,为RLHF的发展贡献力量!

avatar
0
0
0
相关项目
Project Cover

chatbot

该项目提供一个可用自定义语料训练的中文聊天机器人,并即将推出GPT版本,新增MindSpore支持。项目包含Seq2Seq和GPT两大分支,计划于2024年内实现类似mini-GPT4的图文多模态对话功能,并增强分布式集群训练能力。用户可下载小黄鸡语料进行实践,项目支持TensorFlow和PyTorch。

Project Cover

RLHF-Reward-Modeling

该项目专注于通过顺序拒绝采样微调和迭代DPO方法进行奖励和偏好模型训练,提供包括ArmoRM、Pair Preference Model和Bradley-Terry Reward Model在内的多种开源模型,并在RewardBench排行榜中表现显著。项目内容涵盖奖励建模、模型架构、数据集准备和评估结果,适用于基于DRL的RLHF及多项学术研究。

Project Cover

pykoi-rlhf-finetuned-transformers

pykoi是一个开源的Python库,利用RLHF优化大型语言模型(LLM)。它提供统一界面,包含RLHF/RLAIF数据和反馈收集、强化学习微调及模型比较等功能,支持用户存储聊天记录并进行性能对比。此外,pykoi还支持快速实现上下文感知对话生成,并确保数据隐私和安全,适用于CPU和GPU等多种计算资源。

Project Cover

LLM-RLHF-Tuning

本项目实现了RLHF的三阶段训练,包括指令微调、奖励模型训练和PPO算法训练。支持LLaMA和LLaMA2模型,并提供多种分布式加速训练方法。项目附有详细的实现文档,并对比了其他开源框架的功能,是RLHF训练的宝贵资源。

Project Cover

Vicuna-LoRA-RLHF-PyTorch

本文详述如何在普通硬件上优化Vicuna语言模型,包括环境配置、监督微调、合并PEFT适配器、训练奖励模型和PPO微调等步骤,适合中文技术人员使用。

Project Cover

awesome-llm-human-preference-datasets

本页面汇总了多个人类偏好开源数据集,适用于LLM的指令调整、强化学习和评估。涵盖的数据集包括OpenAI WebGPT Comparisons、OpenAI Summarization和Anthropic HH-RLHF等。数据集中包含的问题、模型回答和人类评价数据,旨在为语言模型的开发和改进提供资源。覆盖文本摘要、对话、人类偏好比较等领域,为LLM研究和应用提供全面支持。

Project Cover

open-chatgpt

Open-ChatGPT是一个开源库,能够使用个人数据和最少的计算资源训练个性化的ChatGPT模型。该库提供端到端训练框架,支持分布式训练和卸载,适用于使用DeepSpeed和RLHF技术训练的模型。项目还包括Stanford Alpaca Lora的最新实现,并提供丰富的公开指令调整和RLHF数据集,便于开发者和研究人员使用。

Project Cover

awesome-RLHF

此页面汇集了关于强化学习与人类反馈(RLHF)的最新研究论文,涵盖从基础概念到高级技术,包括大语言模型、视频游戏、机器人和个性化推荐系统等应用。页面将持续更新,跟踪此领域的前沿进展。

Project Cover

PaLM-rlhf-pytorch

本项目实现了基于PaLM架构的强化学习与人类反馈(RLHF),适用于开放环境下的ChatGPT复现。结合了LoRA微调和Flash Attention技术,提供详细的安装和使用指南。加入社区,探索最新的PPO和RL技术进展。

最新项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号