Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Training Verifiers to Solve Math Word Problems

About

State-of-the-art language models can match human performance on many tasks, but they still struggle to robustly perform multi-step mathematical reasoning. To diagnose the failures of current models and support research, we introduce GSM8K, a dataset of 8.5K high quality linguistically diverse grade school math word problems. We find that even the largest transformer models fail to achieve high test performance, despite the conceptual simplicity of this problem distribution. To increase performance, we propose training verifiers to judge the correctness of model completions. At test time, we generate many candidate solutions and select the one ranked highest by the verifier. We demonstrate that verification significantly improves performance on GSM8K, and we provide strong empirical evidence that verification scales more effectively with increased data than a finetuning baseline.

Karl Cobbe, Vineet Kosaraju, Mohammad Bavarian, Mark Chen, Heewoo Jun, Lukasz Kaiser, Matthias Plappert, Jerry Tworek, Jacob Hilton, Reiichiro Nakano, Christopher Hesse, John Schulman• 2021

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningGSM8K
Accuracy60
983
Code GenerationHumanEval--
850
Mathematical ReasoningGSM8K (test)
Accuracy91.2
751
Mathematical ReasoningMATH500 (test)
Accuracy71.6
381
Mathematical Problem SolvingMATH
Accuracy51.2
166
Math ReasoningGSM8K (test)
Accuracy57
155
Arithmetic ReasoningGSM8K
Accuracy55
155
Arithmetic ReasoningGSM8K (test)
Accuracy55
129
Mathematical ReasoningGK 2023
Accuracy46.8
52
Mathematical ReasoningGSM8K original (test)--
44
Showing 10 of 25 rows

Other info

Follow for update