This is the bandit reward model introduced in the preprint Segmenting Text and Learning Their Rewards for Improved RLHF in Language Models (https://arxiv.org/abs/2501.02790). For more details, please visit our repository at https://github.com/yinyueqin/DenseRewardRLHF-PPO.

Downloads last month: 7

Safetensors

Model size

7.5B params

Tensor type

BF16

Inference Providers NEW

Text Classification

This model is not currently available via any of the supported third-party Inference Providers, and the model is not deployed on the HF Inference API.

Dataset used to train yyqoni/rlhflow-llama-3-sft-8b-v2-bandit-rm-700k

Collection including yyqoni/rlhflow-llama-3-sft-8b-v2-bandit-rm-700k

DenseRewardRLHF-PPO

Collection

This repository contains the released models for our paper Segmenting Text and Learning Their Rewards for Improved RLHF in Language Model. • 18 items • Updated 21 days ago • 1