atlas

Research map / Large language models / Agents, code and math

Let's Verify Step by Step

ICLR 2024 · Large language models › Agents, code and math · approach: Mathematical and structural reasoning benchmarks · 31 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.

Compares outcome- and process-supervised reward models for math reasoning on MATH, finds process supervision is more reliable, uses active learning for step-level human labels, and releases the PRM800K dataset.

Problem

Training reliable reward models for multi-step mathematical reasoning: whether step-level (process) feedback is better than final-outcome feedback, given the high cost of human labels.

Approach

Collects 800K step-level human correctness labels on GPT-4-generated MATH solutions, selected by active learning toward convincing wrong-answer solutions, and trains a process reward model that scores solutions by the product of step correctness probabilities; compares it against outcome reward models at large and small scale, using a large PRM as a synthetic labeler for ablations.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use process supervision instead of outcome supervisionReward models trained on small-scale generator samples · MATH, best-of-500accuracyHelped
Use process supervision rather than outcome supervisionLarge-scale reward model · MATH test subset, best-of-NaccuracyHelped
Use process supervision rather than outcome supervisionLarge-scale reward model · OOD STEM tests, best-of-100accuracyHelped
Select convincing samples with active learning rather than uniform data labellingSmall-scale process-supervised reward model · MATHdata efficiency: 2.6xHelped
Add RM-weighted voting to combine PRM scores and majority votingPRM and majority-vote search · MATH test subsetaccuracyNo clear effect
Iteratively retrain the selector between data-collection rounds (ablation)Small-scale active-learning reward models · MATHaccuracyNo clear effect
Use product rather than minimum to reduce step scores, with neutral treated as positive (ablation)PRM · MATH test subset, best-of-1860accuracy: 77.6% -> 78.2%Helped
Treat neutral step labels as positive rather than negative, using product reduction (ablation)PRM · MATH test subset, best-of-1860accuracy: 77.4% -> 78.2%Helped
Treat neutral step labels as positive rather than negative, using minimum reduction (ablation)PRM · MATH test subset, best-of-1860accuracy: 77.8% -> 77.6%Hurt

Surprising findings

Limitations

Experimental setup

Datasets: MATH MathMix PRM800K

Benchmarks: MATH (500-problem test subset) AP Calculus AP Chemistry AP Physics AMC10/12

Metrics: best-of-N accuracy best-of-500 accuracy accuracy on held-out STEM tests

Baselines: outcome-supervised reward model majority voting final-answer outcome supervision RM-weighted voting

Builds on: token-level verifiers (Cobbe et al. 2021) Minerva Uesato et al. 2022 process supervision

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