Let's Verify Step by Step

Hunter Lightman, Vineet Kosaraju, Yura Burda, Harri Edwards, Bowen Baker +5 more
5/31/2023
cs.LGcs.AIcs.CL

No official code found. Best implementation: openai/prm800k (2,152 stars, MIT, last commit 2023-06-01).

Abstract

In recent years, large language models have greatly improved in their ability to perform complex multi-step reasoning. However, even state-of-the-art models still regularly produce logical mistakes. To train more reliable models, we can turn either to outcome supervision, which provides feedback for a final result, or process supervision, which provides feedback for each intermediate reasoning step. Given the importance of training reliable models, and given the high cost of human feedback, it is important to carefully compare the both methods. Recent work has already begun this comparison, but many questions still remain. We conduct our own investigation, finding that process supervision significantly outperforms outcome supervision for training models to solve problems from the challenging MATH dataset. Our process-supervised model solves 78% of problems from a representative subset of the MATH test set. Additionally, we show that active learning significantly improves the efficacy of process supervision. To support related research, we also release PRM800K, the complete dataset of 800,000 step-level human feedback labels used to train our best reward model.

View on arXivView PDF

Code Implementations(1)

openai/prm800kCommunity100%
MIT

800,000 step-level correctness labels on LLM solutions to MATH problems

2,152128PythonApr 13, 20233 years agoMIT

Similar Papers

Cite this paper

@article{lightman2023lets,
  title  = {Let's Verify Step by Step},
  author = {Hunter Lightman and Vineet Kosaraju and Yura Burda and Harri Edwards and Bowen Baker and Teddy Lee and Jan Leike and John Schulman and Ilya Sutskever and Karl Cobbe},
  year   = {2023},
  eprint = {2305.20050},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2305.20050v1},
  journal = {ICLR 2023 2023}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2305.20050.svg)](https://codeofpaper.com/paper/2305.20050)

Discussion