atlas

Research map / Large language models / Alignment and preferences

HelpSteer 2: Open-source dataset for training top-performing reward models

NeurIPS 2024 · Large language models › Alignment and preferences · approach: Reward model design and preference models · 4 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Releases HelpSteer2, a CC-BY-4.0 10k-pair multi-attribute helpfulness preference dataset, trains Llama 3 70B and Nemotron-4 340B SteerLM regression reward models (92.0% on RewardBench primary), and proposes SteerLM 2.0 for alignment.

Problem

Permissively licensed preference datasets (Open Assistant, HH-RLHF, HelpSteer) have become less effective for reward modeling, while stronger preference datasets distilled from GPT-4 carry commercial-use restrictions.

Approach

Collect a small but high-quality multi-attribute (helpfulness, correctness, coherence, complexity, verbosity) Likert-scored dataset using responses from permissively usable in-house LLMs, with more annotators per sample and sequential pairwise rating; train SteerLM-style regression reward models on it and use them to align models via Iterative DPO, PPO, and a new SteerLM 2.0 KL-based policy-approx

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Train the Llama 3 70B reward model with HelpSteer2 instead of Open AssistantLlama 3 70B reward model · RewardBench primary datasetOverall accuracy: 79.1 -> 88.8Helped
Train the Llama 3 70B reward model with HelpSteer2 instead of HH-RLHFLlama 3 70B reward model · RewardBench primary datasetOverall accuracy: 73.9 -> 88.8Helped
Scale the reward-model base from Llama 3 70B to Nemotron-4 340B, using HelpSteer2HelpSteer2 reward model · RewardBench primary datasetOverall accuracy: 88.8 -> 92.0Helped
Replace Daring Anteater with Open Assistant for SFT (ablation)Llama 3 70B SFT model · MT BenchGPT-4-Turbo score: 7.96 -> 6.75Hurt
Use the reward model after SFT on Open Assistant (ablation)Llama 3 70B · MT BenchGPT-4-Turbo score: 6.75 -> 7.44Helped
Use SteerLM 2.0 after the first iteration (ablation)Llama 3 70B SteerLM model · MT BenchGPT-4-Turbo score: 8.24 -> 8.28Helped
Apply DPO using HelpSteer2 preference pairsLlama 3 70B SFT model · TruthfulQAMC2: 0.6025 -> 0.6321Helped
Apply PPO using HelpSteer2 prompts and reward modelLlama 3 70B SFT model · TruthfulQAMC2: 0.6025 -> 0.5629Hurt
Apply PPO using HelpSteer2 prompts and reward modelLlama 3 70B SFT model · AlpacaEval 2.0 LCLength-controlled score: 32.87 (1.40) -> 33.17 (1.38)Helped
Apply SteerLM 2.0 for two iterations rather than the original SteerLMLlama 3 70B · MT BenchGPT-4-Turbo score: 8.17 -> 8.28Helped
Apply SteerLM 2.0 for two iterations rather than the original SteerLMLlama 3 70B · AlpacaEval 2.0 LCLength-controlled score: 31.10 (1.37) -> 29.93 (1.35)Hurt
Replace Daring Anteater with Open Assistant for SFT (ablation)Llama 3 70B SFT model · TruthfulQAMC2: 0.6025 -> 0.5137Hurt

Surprising findings

Limitations

Experimental setup

Datasets: HelpSteer2 Daring Anteater Open Assistant HH-RLHF ShareGPT HelpSteer RewardBench MT Bench TruthfulQA AlpacaEval 2.0 Arena Hard

Benchmarks: RewardBench MT Bench TruthfulQA MC2 AlpacaEval 2.0 LC Arena Hard

Metrics: accuracy MT Bench score TruthfulQA MC2 AlpacaEval 2.0 length-controlled win rate Arena Hard win rate mean response length inter-annotator quadratic weighted Cohen's kappa

Baselines: Llama 3 70B Instruct GPT-4-0613 ArmoRM-Llama 3 8B RLHFlow-Llama 3 8B Eurus RM Mistral 7B Starling RM Yi 34B Llama 3 70B RM trained on Open Assistant, HH-RLHF, HelpSteer

Builds on: SteerLM HelpSteer Iterative DPO RewardBench

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy