First-Step Advantage: Importance of Starting Right in Multi-Step Math Reasoning

Language models can solve complex reasoning tasks better by learning to generate rationales for their predictions.Often these models know how to solve a task, but their auto-regressive decoding nature leads to incorrect results if they start incorrectly.We observe that smaller models, in particular, when corrected, can solve a task that they would have otherwise struggled with.We demonstrate this phenomenon by using a larger model to guide smaller models, which leads to significantly improved performance (up to +24 points on the GSM8K dataset by 7B models).Furthermore, to assist smaller models in initiating the starting step correctly, we propose QuestCoT, where a smaller model first asks itself how to start, before proceeding with a chain of reasoning.On various multistep mathematical reasoning datasets for multiple smaller models, we show that getting the right start can lead to significant performance gains across all models (gains of up to +6 points on GSM8K, +9 on SVAMP, +5 on ASDiv, and +7 on MultiArith). LMNatalia sold 48+72=120 clips in total.Natalia sold clips to 48 of her friends in April....

Authors

Publication Details

Published
2025-01-01
DOI
https://doi.org/10.18653/v1/2025.findings-acl.42
Primary Topic
Innovative Teaching and Learning Methods
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

First-Step Advantage: Importance of Starting Right in Multi-Step Math Reasoning

Kumar Shridhar, Niket Tandon, Kushal Jain, Moritz Miller
Innovative Teaching and Learning Methods
article

First-Step Advantage: Importance of Starting Right in Multi-Step Math Reasoning

Kumar Shridhar, Niket Tandon, Kushal Jain, Moritz Miller
article en

Abstract

Language models can solve complex reasoning tasks better by learning to generate rationales for their predictions.Often these models know how to solve a task, but their auto-regressive decoding nature leads to incorrect results if they start incorrectly.We observe that smaller models, in particular, when corrected, can solve a task that they would have otherwise struggled with.We demonstrate this phenomenon by using a larger model to guide smaller models, which leads to significantly improved performance (up to +24 points on the GSM8K dataset by 7B models).Furthermore, to assist smaller models in initiating the starting step correctly, we propose QuestCoT, where a smaller model first asks itself how to start, before proceeding with a chain of reasoning.On various multistep mathematical reasoning datasets for multiple smaller models, we show that getting the right start can lead to significant performance gains across all models (gains of up to +6 points on GSM8K, +9 on SVAMP, +5 on ASDiv, and +7 on MultiArith). LMNatalia sold 48+72=120 clips in total.Natalia sold clips to 48 of her friends in April....

Quality Education
Openalex Percentile: Top 79%
Innovative Teaching and Learning Methods
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.