HINT-lab/mistral-7b-ppo-c-hermes

Model

0

stars

6

commits

1

linked in READMEs

Oct 17, 2024

updated

conversational
endpoints_compatible
mistral
safetensors
text-generation
text-generation-inference
transformers
Browse cluster: LLM Fine-tuning with Reinforcement Learning

README

Model Card for Model ID

PPO-C (PPO with Calibrated Reward Calculation) is an RLHF algorithm to mitigate verbalized overconfidence in RLHF-trained Large Language Models. PPO-C adjusts standard reward model scores during PPO training. It maintains a running average of past reward scores as a dynamic threshold to classify responses, and adjusts the reward scores based on model expressed verbalized confidence. Please refer to our preprint (Taming Overconfidence in LLMs: Reward Calibration in RLHF) and repo for more details.

Model Details

Model Description

We train teknium/OpenHermes-2.5-Mistral-7B on our HINT-lab/prompt-collections-final-v0.3 with a vanilla reward model HINT-lab/mistral-7b-hermes-rm-skywork.

Model Sources [optional]

Contributors

Sean123321

3 commits

teapot123

3 commits

HINT-lab/mistral-7b-ppo-c-hermes

Model

0

stars

6

commits

1

linked in READMEs

Oct 17, 2024

updated

conversational
endpoints_compatible
mistral
safetensors
text-generation
text-generation-inference
transformers
Browse cluster: LLM Fine-tuning with Reinforcement Learning

README

Model Card for Model ID

PPO-C (PPO with Calibrated Reward Calculation) is an RLHF algorithm to mitigate verbalized overconfidence in RLHF-trained Large Language Models. PPO-C adjusts standard reward model scores during PPO training. It maintains a running average of past reward scores as a dynamic threshold to classify responses, and adjusts the reward scores based on model expressed verbalized confidence. Please refer to our preprint (Taming Overconfidence in LLMs: Reward Calibration in RLHF) and repo for more details.

Model Details

Model Description

We train teknium/OpenHermes-2.5-Mistral-7B on our HINT-lab/prompt-collections-final-v0.3 with a vanilla reward model HINT-lab/mistral-7b-hermes-rm-skywork.

Model Sources [optional]

Contributors

Sean123321

3 commits

teapot123

3 commits