Beyond Human Data: Scaling Self-Training for Problem-Solving with Language Models

AI-generated keywords: ReST$^{EM}$ Language Models Fine-tuning Math Reasoning APPS Coding

AI-generated Key Points

  • Self-training method called ReST$^{EM}$ proposed for fine-tuning language models (LMs) on tasks with scalar feedback
  • Method involves generating samples, filtering with binary feedback, and multiple iterations of fine-tuning
  • Tested on advanced math reasoning and APPS coding benchmarks using PaLM-2 models
  • Outperforms fine-tuning only on human data, scales well with model size
  • Performance evaluated on Big-Bench Hard suite of tasks, no major degradation compared to base model
  • Problem-solving capabilities assessed on Hungarian high school finals exam in mathematics, performs well compared to existing models except GPT-4
  • Ability to generate multiple correct solutions for each problem is a key strength, provides additional training data compared to human-generated data which typically offers only a single solution per problem
  • Reduces dependence on human generated data and improves reasoning capabilities in language models
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Authors: Avi Singh, John D. Co-Reyes, Rishabh Agarwal, Ankesh Anand, Piyush Patil, Peter J. Liu, James Harrison, Jaehoon Lee, Kelvin Xu, Aaron Parisi, Abhishek Kumar, Alex Alemi, Alex Rizkowsky, Azade Nova, Ben Adlam, Bernd Bohnet, Hanie Sedghi, Igor Mordatch, Isabelle Simpson, Izzeddin Gur, Jasper Snoek, Jeffrey Pennington, Jiri Hron, Kathleen Kenealy, Kevin Swersky, Kshiteej Mahajan, Laura Culp, Lechao Xiao, Maxwell L. Bileschi, Noah Constant, Roman Novak, Rosanne Liu, Tris Warkentin, Yundi Qian, Ethan Dyer, Behnam Neyshabur, Jascha Sohl-Dickstein, Noah Fiedel

First three authors contributed equally
License: CC BY 4.0

Abstract: Fine-tuning language models~(LMs) on human-generated data remains a prevalent practice. However, the performance of such models is often limited by the quantity and diversity of high-quality human data. In this paper, we explore whether we can go beyond human data on tasks where we have access to scalar feedback, for example, on math problems where one can verify correctness. To do so, we investigate a simple self-training method based on expectation-maximization, which we call ReST$^{EM}$, where we (1) generate samples from the model and filter them using binary feedback, (2) fine-tune the model on these samples, and (3) repeat this process a few times. Testing on advanced MATH reasoning and APPS coding benchmarks using PaLM-2 models, we find that ReST$^{EM}$ scales favorably with model size and significantly surpasses fine-tuning only on human data. Overall, our findings suggest self-training with feedback can substantially reduce dependence on human-generated data.

Submitted to arXiv on 11 Dec. 2023

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2312.06585v1

In this paper, the authors propose a self-training method called ReST$^{EM}$ to fine-tune language models (LMs) on tasks where scalar feedback is available, such as math problems. The method involves generating samples from the model, filtering them using binary feedback, and fine-tuning the model on these samples through multiple iterations. The authors test ReST$^{EM}$ on advanced math reasoning and APPS coding benchmarks using PaLM-2 models and compare its performance with fine-tuning only on human data. They find that ReST$^{EM}$ scales well with model size and significantly outperforms fine-tuning only on human data. Additionally, they evaluate the performance of ReST$^{EM}$-fine-tuned models on the Big-Bench Hard suite of tasks and find no major degradation compared to the base model. Furthermore, they assess the problem-solving capabilities of their model by evaluating it on a Hungarian high school finals exam in mathematics and find that it performs well, surpassing all existing models except GPT-4. A key strength of ReST$^{EM}$ is its ability to generate multiple correct solutions for each problem, providing valuable additional training data compared to human-generated data which typically offers only a single solution per problem. Through their study, the authors demonstrate that ReST$^{EM}$ can substantially reduce dependence on human generated data and improve reasoning capabilities in language models.
Created on 13 Dec. 2023

Assess the quality of the AI-generated content by voting

Score: 0

Why do we need votes?

Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.

Similar papers summarized with our AI tools

Navigate through even more similar papers through a

tree representation

Look for similar papers (in beta version)

By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while our database grows.

Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.