atlas

Research map / Large language models / Reasoning and chain of thought

GLoRe: When, Where, and How to Improve LLM Reasoning via Global and Local Refinements

ICML 2024 · Large language models › Reasoning and chain of thought · approach: Efficient and adaptive thinking · 3 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 GLoRe, a pipeline that uses an ORM to decide when to refine, a synthetic-data-trained stepwise ORM (SORM) to localize where, and global and local refinement models to decide how, improving Llama-2 13B on GSM8K.

Problem

LLMs struggle to identify when and where to refine their reasoning without external feedback; ORMs under-estimate intermediate-step correctness and PRMs require costly human step labels.

Approach

Train stepwise ORMs (SORMs) only on synthetic data by rejection-sampling the student from each prefix to approximate the optimal value function V*, then use the SORM to localize the first error for local refinement, while global refinement starts from scratch; an ORM reranks draft, global and local refinements.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
train a SORM instead of an ORM for step verificationLlama-2 reward models · GSM8K test set labelsstep-level accuracy: 0.73 -> 0.81Helped
train a SORM instead of an ORM for step verificationLlama-2 reward models · SVAMP test set labelsstep-level accuracy: 0.85 -> 0.87No clear effect
use the SORM rather than ORM to predict final-answer correctnessLlama-2 13B reward models · GSM8K test set labelsfinal answer accuracy: 0.85 -> 0.8Hurt
balance positive and negative ORM labels per question (ablation)Balanced ORM · GSM8K test set labelsstep-level accuracy: 0.74 -> 0.73Hurt
use SORM instead of ORM to locate the first error for local refinementLlama-2 13B local refinement · GSM8K incorrect draftsrefinement accuracy: 0.262 -> 0.283Helped
combine global refinement with SORM-local refinementLlama-2 13B refinement pipeline · GSM8K incorrect draftsrefinement accuracy: 0.283 -> 0.412Helped
use global and local refinements together instead of global refinement aloneLlama-2 13B refinement pipeline · GSM8K incorrect draftsrefinement accuracy: 0.281 -> 0.412Helped
train global refinement with Expert Iteration rather than PPO (ablation)Llama-2 7B global refiner · GSM8K testaccuracy: 0.36 -> 0.39Helped
train global refinement with PPO rather than SFT (ablation)Llama-2 7B global refiner · GSM8K testaccuracy: 0.36 -> 0.36No clear effect
use a contrastive loss instead of classifier training for the ORM (ablation)Llama-2 7B ORM · GSM8K testrerank accuracy: 0.56 -> 0.47Hurt
filter SORM training data using SORM self-supervision (ablation)SORM · SORM test setaccuracy: 75% -> 79%Helped
filter SORM training data using the 2-epoch SORM checkpoint (ablation)SORM · SORM test setaccuracy: 78% -> 79%Helped

Surprising findings

Limitations

Experimental setup

Datasets: GSM8K SVAMP MATH PRM data

Benchmarks: GSM8K SVAMP

Metrics: step-level accuracy final answer accuracy refinement accuracy maj@1

Baselines: ORM Balanced ORM Best-of-3 (Bo3) PPO global refinement Prompted Llama-2 7B Value-based refinement

Builds on: Expert Iteration Outcome-based reward models (Cobbe et al.) Process-based reward models Chain-of-thought

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