Math-Shepherd: Verify and Reinforce LLMs Step-by-step without Human Annotations

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wang, Peiyi, Li, Lei, Shao, Zhihong, Xu, R. X., Dai, Damai, Li, Yifei, Chen, Deli, Wu, Y., Sui, Zhifang
Format: Preprint
Publié: 2023
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913236883865600
author Wang, Peiyi
Li, Lei
Shao, Zhihong
Xu, R. X.
Dai, Damai
Li, Yifei
Chen, Deli
Wu, Y.
Sui, Zhifang
author_facet Wang, Peiyi
Li, Lei
Shao, Zhihong
Xu, R. X.
Dai, Damai
Li, Yifei
Chen, Deli
Wu, Y.
Sui, Zhifang
contents In this paper, we present an innovative process-oriented math process reward model called \textbf{Math-Shepherd}, which assigns a reward score to each step of math problem solutions. The training of Math-Shepherd is achieved using automatically constructed process-wise supervision data, breaking the bottleneck of heavy reliance on manual annotation in existing work. We explore the effectiveness of Math-Shepherd in two scenarios: 1) \textit{Verification}: Math-Shepherd is utilized for reranking multiple outputs generated by Large Language Models (LLMs); 2) \textit{Reinforcement Learning}: Math-Shepherd is employed to reinforce LLMs with step-by-step Proximal Policy Optimization (PPO). With Math-Shepherd, a series of open-source LLMs demonstrates exceptional performance. For instance, the step-by-step PPO with Math-Shepherd significantly improves the accuracy of Mistral-7B (77.9\%$\to$84.1\% on GSM8K and 28.6\%$\to$33.0\% on MATH). The accuracy can be further enhanced to 89.1\% and 43.5\% on GSM8K and MATH with the verification of Math-Shepherd, respectively. We believe that automatic process supervision holds significant potential for the future evolution of LLMs.
format Preprint
id arxiv_https___arxiv_org_abs_2312_08935
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Math-Shepherd: Verify and Reinforce LLMs Step-by-step without Human Annotations
Wang, Peiyi
Li, Lei
Shao, Zhihong
Xu, R. X.
Dai, Damai
Li, Yifei
Chen, Deli
Wu, Y.
Sui, Zhifang
Artificial Intelligence
Computation and Language
Machine Learning
In this paper, we present an innovative process-oriented math process reward model called \textbf{Math-Shepherd}, which assigns a reward score to each step of math problem solutions. The training of Math-Shepherd is achieved using automatically constructed process-wise supervision data, breaking the bottleneck of heavy reliance on manual annotation in existing work. We explore the effectiveness of Math-Shepherd in two scenarios: 1) \textit{Verification}: Math-Shepherd is utilized for reranking multiple outputs generated by Large Language Models (LLMs); 2) \textit{Reinforcement Learning}: Math-Shepherd is employed to reinforce LLMs with step-by-step Proximal Policy Optimization (PPO). With Math-Shepherd, a series of open-source LLMs demonstrates exceptional performance. For instance, the step-by-step PPO with Math-Shepherd significantly improves the accuracy of Mistral-7B (77.9\%$\to$84.1\% on GSM8K and 28.6\%$\to$33.0\% on MATH). The accuracy can be further enhanced to 89.1\% and 43.5\% on GSM8K and MATH with the verification of Math-Shepherd, respectively. We believe that automatic process supervision holds significant potential for the future evolution of LLMs.
title Math-Shepherd: Verify and Reinforce LLMs Step-by-step without Human Annotations
topic Artificial Intelligence
Computation and Language
Machine Learning
url https://arxiv.org/abs/2312.08935