This is the bandit reward based ppo 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: 13

Safetensors

Model size

3.82B params

Tensor type

BF16

Inference Providers NEW

Text Generation

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.

Model tree for yyqoni/Phi-3-mini-4k-bandit-ppo-60k

Base model

microsoft/Phi-3-mini-4k-instruct

Finetuned

(206)

this model

Dataset used to train yyqoni/Phi-3-mini-4k-bandit-ppo-60k

Collection including yyqoni/Phi-3-mini-4k-bandit-ppo-60k

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 20 days ago • 1