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

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Jain, Kushal, Miller, Moritz, Tandon, Niket, Shridhar, Kumar
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913411086942208
author Jain, Kushal
Miller, Moritz
Tandon, Niket
Shridhar, Kumar
author_facet Jain, Kushal
Miller, Moritz
Tandon, Niket
Shridhar, Kumar
contents 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). To assist smaller models in initiating the starting step, 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 over 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).
format Preprint
id arxiv_https___arxiv_org_abs_2311_07945
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle First-Step Advantage: Importance of Starting Right in Multi-Step Math Reasoning
Jain, Kushal
Miller, Moritz
Tandon, Niket
Shridhar, Kumar
Computation and Language
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). To assist smaller models in initiating the starting step, 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 over 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).
title First-Step Advantage: Importance of Starting Right in Multi-Step Math Reasoning
topic Computation and Language
url https://arxiv.org/abs/2311.07945