Large Language Monkeys: Scaling Inference Compute with Repeated Sampling

AI-generated keywords: Language model training Inference compute Scaling Coverage Robust verifiers

AI-generated Key Points

  • Scaling compute has enhanced language model training capabilities
  • Inference compute as a dimension for scaling by increasing generated samples
  • Coverage scales with number of samples, leading to improved performance in coding and formal proofs
  • Repeated sampling increased problem-solving success rates significantly
  • Amplifying more affordable models with multiple samples can be cost-effective and efficient
  • Log-linear relationship between coverage and number of samples identified
  • Identifying correct samples crucial for domains lacking automatic verifiers
  • Traditional methods like majority voting or reward models plateaued beyond several hundred samples
  • Cautionary tales shared regarding inconsistencies in test suites and limitations in verification tools
  • Scaling inference compute through repeated sampling can lead to significant improvements in coverage across tasks and models
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Authors: Bradley Brown, Jordan Juravsky, Ryan Ehrlich, Ronald Clark, Quoc V. Le, Christopher Ré, Azalia Mirhoseini

License: CC BY 4.0

Abstract: Scaling the amount of compute used to train language models has dramatically improved their capabilities. However, when it comes to inference, we often limit the amount of compute to only one attempt per problem. Here, we explore inference compute as another axis for scaling by increasing the number of generated samples. Across multiple tasks and models, we observe that coverage - the fraction of problems solved by any attempt - scales with the number of samples over four orders of magnitude. In domains like coding and formal proofs, where all answers can be automatically verified, these increases in coverage directly translate into improved performance. When we apply repeated sampling to SWE-bench Lite, the fraction of issues solved with DeepSeek-V2-Coder-Instruct increases from 15.9% with one sample to 56% with 250 samples, outperforming the single-attempt state-of-the-art of 43% which uses more capable frontier models. Moreover, using current API pricing, amplifying the cheaper DeepSeek model with five samples is more cost-effective and solves more issues than paying a premium for one sample from GPT-4o or Claude 3.5 Sonnet. Interestingly, the relationship between coverage and the number of samples is often log-linear and can be modelled with an exponentiated power law, suggesting the existence of inference-time scaling laws. Finally, we find that identifying correct samples out of many generations remains an important direction for future research in domains without automatic verifiers. When solving math word problems from GSM8K and MATH, coverage with Llama-3 models grows to over 95% with 10,000 samples. However, common methods to pick correct solutions from a sample collection, such as majority voting or reward models, plateau beyond several hundred samples and fail to fully scale with the sample budget.

Submitted to arXiv on 31 Jul. 2024

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: 2407.21787v1

In the realm of language model training, scaling compute has significantly enhanced the capabilities of models. However, when it comes to inference, we often limit the amount of compute to just one attempt per problem. In this study, we delve into the concept of inference compute as another dimension for scaling by increasing the number of generated samples. Across various tasks and models, it is observed that coverage - which refers to the fraction of problems solved by any attempt - scales with the number of samples over four orders of magnitude. This increase in coverage directly translates into improved performance in domains like coding and formal proofs where answers can be automatically verified. For instance, when applying repeated sampling to SWE-bench Lite, it was found that the fraction of issues solved with DeepSeek-V2-Coder-Instruct increased from 15.9% with one sample to 56% with 250 samples, surpassing the single-attempt state-of-the-art success rate of 43%. Moreover, utilizing current API pricing revealed that amplifying the more affordable DeepSeek model with five samples proved to be more cost-effective and efficient in solving problems compared to paying a premium for one sample from more advanced models like GPT-4o or Claude 3.5 Sonnet. Interestingly, a log-linear relationship between coverage and the number of samples was identified across multiple experiments, suggesting the existence of inference-time scaling laws. Furthermore, it was noted that while identifying correct samples out of numerous generations remains crucial for domains lacking automatic verifiers. from GSM8K and MATH datasets using Llama-3 models resulted in coverage exceeding 95% with 10,000 samples. However, traditional methods such as majority voting or reward models plateaued beyond several hundred samples and did not fully scale with the sample budget. Additionally, cautionary tales were shared regarding software development tasks and verification tools. were identified in SWE-bench Lite where inconsistencies in test suites led to false positives/negatives even on ground-truth solutions. Similarly, false negatives were encountered in CodeContests due to limitations in existing verification tools. Overall, this study highlights that scaling inference compute through repeated sampling can lead to significant improvements in coverage across various tasks and models. It demonstrates how amplifying weaker models with multiple samples can sometimes outperform single attempts from more powerful models. The findings underscore the importance of robust verifiers and highlight areas for further research in improving decision-making processes based on sampled outputs.
Created on 21 Aug. 2024

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.