V-STaR: Training verifiers for self-taught reasoners

Arian Hosseini*, Xingdi Yuan, Nikolay Malkin, Aaron Courville, Alessandro Sordoni, Rishabh Agarwal*

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract / Description of output

Common self-improvement approaches for large language models (LLMs), such as STaR, iteratively fine-tune LLMs on self-generated solutions to improve their problem-solving ability. However, these approaches discard the large amounts of incorrect solutions generated during this process, potentially neglecting valuable information in such solutions. To address this shortcoming, we propose V-STaR that utilizes both the correct and incorrect solutions generated during the self-improvement process to train a verifier using DPO that judges correctness of model-generated solutions. This verifier is used at inference time to select one solution among many candidate solutions. Running V-STaR for multiple iterations results in progressively better reasoners and verifiers, delivering a 4% to 17% test accuracy improvement over existing self-improvement and verification approaches on common code generation and math reasoning benchmarks with LLaMA2 models.
Original languageEnglish
Title of host publicationProceedings of the 2024 Conference on Language Modeling
Publication statusAccepted/In press - 10 Jul 2024
EventConference on Language Modeling - University of Pennsylvania, Philadelphia, United States
Duration: 7 Oct 20249 Oct 2024
https://colmweb.org/

Conference

ConferenceConference on Language Modeling
Abbreviated titleCOLM 2024
Country/TerritoryUnited States
CityPhiladelphia
Period7/10/249/10/24
Internet address

Fingerprint

Dive into the research topics of 'V-STaR: Training verifiers for self-taught reasoners'. Together they form a unique fingerprint.

Cite this