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

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

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

  • Comprehensive Summary
  • Key points
  • Layman's Summary
  • Blog article
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.
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.

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

Tools for this paper

Explain it differently 3 free rewrites left today
Export BibTeX · RIS (Zotero, Mendeley, EndNote) · Markdown (Notion, Obsidian) · Premium
Citation alerts Premium
Journal club Slides to present it Premium
For journalists Press kit and hype check Premium

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

Figures and plots

Ask about a figure by number or use its question button to request visual analysis. Up to two figures and four image panels per question. Values read from plots are approximate; check the original data.

AI assistant instructions?

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.

The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.

Similar papers summarized with our AI tools

Navigate through even more similar papers through a

tree representation

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.