atlas

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

AlphaMath Almost Zero: Process Supervision without Process

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

Proposes AlphaMath, an iterative MCTS-based self-training framework that uses only question-answer pairs (no process annotations) to train an LLM policy with an appended value head, plus a step-level beam search for inference.

Problem

Improving mathematical reasoning of LLMs without costly human- or GPT-4-annotated process-supervised solutions.

Approach

Integrate a value head into the LLM policy and use MCTS over the LLM to generate correct and incorrect step-level solutions from only final answers; the value head is trained on MCTS-derived state values, and a step-level beam search uses the value model to select steps instead of relying on prior probabilities.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply AlphaMath iterative training instead of using the 2-shot prompt aloneDeepSeekMath-Base-7B · GSM8Kaccuracy: 59.7 -> 73.5Helped
Apply AlphaMath iterative training instead of using the 2-shot prompt aloneDeepSeekMath-Base-7B · MATHaccuracy: 33.2 -> 53.6Helped
Apply AlphaMath iterative training instead of using the 2-shot prompt aloneDeepSeekMath-Base-7B · GaoKao2023accuracy: 21.9 -> 40.5Helped
Apply AlphaMath iterative training instead of using the 2-shot prompt aloneDeepSeekMath-Base-7B · OCWCoursesaccuracy: 9.2 -> 26.1Helped
Add step-level beam search with B1=1 (ablation)AlphaMath, DeepSeekMath-Base-7B · GSM8Kaccuracy: 73.5 -> 81.1Helped
Increase SBS beam size from B1=1 to B1=3 (ablation)AlphaMath, DeepSeekMath-Base-7B · MATHaccuracy: 62.8 -> 66.3Helped
Increase SBS beam size from B1=1 to B1=3 (ablation)AlphaMath, DeepSeekMath-Base-7B · GaoKao2023accuracy: 46.2 -> 51.4Helped
Increase SBS beam size from B1=1 to B1=3 (ablation)AlphaMath, DeepSeekMath-Base-7B · OCWCoursesaccuracy: 30.5 -> 33.1Helped
Use MCTS inference instead of SBS with B1=1 (ablation)AlphaMath, DeepSeekMath-Base-7B · MATHaccuracy: 62.8 -> 64.0Helped
Use MCTS inference instead of SBS with B1=1 (ablation)AlphaMath, DeepSeekMath-Base-7B · OCWCoursesaccuracy: 30.5 -> 33.8Helped
Increase SBS beam size from B1=3 to B1=5 (ablation)AlphaMath, DeepSeekMath-Base-7B · MATHaccuracy: 66.30 -> 65.98Hurt
Apply AlphaMath iterative trainingLlama3-base · GSM8Kaccuracy: 40.7 -> 59.4Helped

Surprising findings

Limitations

Experimental setup

Datasets: GSM8K MATH GaoKao2023 OCWCourses MARIO 27k data

Benchmarks: GSM8K MATH GaoKao2023 OCWCourses

Metrics: accuracy

Baselines: GPT-4 GPT-4 (PAL) ChatGPT Gemini-1.5 Pro Claude-3.5-Sonnet MAmmoTH-Coder MathCoder ToRA MARIO MathGenie DeepSeekMath-Instruct

Builds on: AlphaGo Zero PUCT ReAct Program-aided language models DeepSeekMath

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