PreprintCited by 50Topic ModelingNatural Language Processing TechniquesAdversarial Robustness in Machine Learning

Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameters

Open access
Snell, Charlie, Lee, Jaehoon, Xu, Kelvin, Kumar, Aviral·2024·arXiv

Abstract

Enabling LLMs to improve their outputs by using more test-time computation is a critical step towards building generally self-improving agents that can operate on open-ended natural language. In this paper, we study the scaling of inference-time computation in LLMs, with a focus on answering the question: if an LLM is allowed to use a fixed but non-trivial amount of inference-time compute, how much can it improve its performance on a challenging prompt? Answering this question has implications not only on the achievable performance of LLMs, but also on the future of LLM pretraining and how one should tradeoff inference-time and pre-training compute. Despite its importance, little research attempted to understand the scaling behaviors of various test-time inference methods. Moreover, current work largely provides negative results for a number of these strategies. In this work, we analyze two primary mechanisms to scale test-time computation: (1) searching against dense, process-based verifier reward models; and (2) updating the model's distribution over a response adaptively, given the prompt at test time. We find that in both cases, the effectiveness of different approaches to scaling test-time compute critically varies depending on the difficulty of the prompt. This observation motivates applying a "compute-optimal" scaling strategy, which acts to most effectively allocate test-time compute adaptively per prompt. Using this compute-optimal strategy, we can improve the efficiency of test-time compute scaling by more than 4x compared to a best-of-N baseline. Additionally, in a FLOPs-matched evaluation, we find that on problems where a smaller base model attains somewhat non-trivial success rates, test-time compute can be used to outperform a 14x larger model.

The claims and statistics of this paper haven’t been checked yet.

Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.

A few free checks a day · papers already verified open instantly · results are public

Integrity signals

Registry

DataCite · 10.48550/arxiv.2408.03314

Cites

Statistics

Claims the paper makes

The claims and statistics of this paper haven’t been checked yet.

Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.

A few free checks a day · papers already verified open instantly · results are public

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

  • Test-time scaling is widely adopted in STEM and reasoning benchmarks, where allocating more computational effort to inference has been shown to yield higher performance.

    “This approach is widely adopted in STEM and reasoning benchmarks [70], where allocating more computational effort to inference has been shown to yield higher performance.”

    From When AI Co-Scientists Fail: SPOT-a Benchmark for Automated Verification of Scientific Research (2025) · cites this paper as [70]

    The Datasets section (page not supplied) says the study focuses on the MATH benchmark, and the Discussion and Future Work section reports performance gains from allocating test-time compute adaptively. The paper does not establish that this approach is widely adopted across STEM and reasoning benchmarks; it also says gains depend on problem difficulty and can be small on the hardest questions. (LLM, conf=0.85)

    Experimental — accuracy not yet measured
  • In other domains such as math or common-sense reasoning, parallel sampling with self-consistency has been shown to be highly competitive.

    “This contrasts with findings in other domains, such as math or common-sense reasoning, where parallel sampling with self-consistency has been shown to be highly competitive (Snell et al., 2024).”

    From The Illusion of Diminishing Returns: Measuring Long Horizon Execution in LLMs (2025) · cites this paper as Snell et al., 2024

    In “Using revisions at inference-time” (page unavailable), the paper compares parallel sampling with majority voting on math problems and says sequential revisions narrowly outperform it. That supports parallel sampling as a competitive approach in the studied setting, but our copy does not establish the claimed result for common-sense reasoning. (LLM, conf=0.80)

    Experimental — accuracy not yet measured
  • Wu et al. (2024) and Snell et al. (2024) demonstrated that small models with scaling strategies such as majority voting, best-of-n, and weighted voting can outperform larger models when using the same amount of FLOPs.

    “Wu et al. (2024) and Snell et al. (2024) demonstrated that small models with scaling strategies such as majority voting, best-of-n, and weighted voting can outperform larger models when using the same amount of FLOPs.”

    From Multi-Agent Reasoning Improves Compute Efficiency: Pareto-Optimal Test-Time Scaling (2026) · cites this paper as Snell et al., 2024

    The Abstract and “Test-time Search Against a PRM Verifier” report that a smaller model using additional test-time compute can outperform a 14× larger model in a FLOPs-matched comparison, chiefly on easier and intermediate problems; the most challenging problems showed little benefit. The paper discusses majority voting and best-of-N, but its “Best-of-N weighted” method selects an answer using a verifier, not weighted voting. It does not establish that each strategy listed in the claim outperforms a larger model. (LLM, conf=0.86)

    Experimental — accuracy not yet measured
  • Test-time scaling consistently improves performance on challenging math and code problems that require reasoning by increasing computational resources allocated during inference.

    “This approach consistently improves performance on challenging math and code problems that require reasoning by increasing the computational resources allocated during inference [8,158].”

    From Stop Overthinking: A Survey on Efficient Reasoning for Large Language Models (2025) · cites this paper as [158]

    The Datasets section says the experiments focus on the MATH benchmark, and the Test-time Search Against a PRM Verifier section reports that additional test-time compute can improve outputs under some conditions. But that section reports very little benefit on the most challenging questions, while Further improving test-time compute scaling describes small gains on hard problems. The paper does not establish the claim for code problems or support 'consistently' across challenging problems (pages unavailable in our copy). (LLM, conf=0.90)

    Experimental — accuracy not yet measured
  • The agentic loop acts as a powerful test-time scaling mechanism.

    “This agentic loop acts as a powerful test-time scaling mechanism [25, 17, 32], utilizing the reasoning capabilities [31] of LLMs at each turn while grounding them through environment responses.”

    From The Complexity Trap: Simple Observation Masking Is as Efficient as LLM Summarization for Agent Context Management (2025) · cites this paper as [25]

    The ‘Test-time Search Against a PRM Verifier’ section reports that iteratively refining an answer can be an effective use of test-time compute, particularly for easier problems. It does not establish that an agentic loop grounded in environment responses is a powerful scaling mechanism; the paper studies answer revision and verifier-guided search instead. No page number is available in our copy. (LLM, conf=0.82)

    Experimental — accuracy not yet measured

Self-check

Percentages, sums and reported differences re-computed from the paper’s own numbers.

The self-check needs the paper’s full text.

Cites

No references could be resolved yet.

Research works printed but not linked: 18

  1. C. Blakeney, M. Paul, B. W. Larsen, S. Owen, and J. Frankle. Does your data spark joy? performance gains from domain upsampling at the end of training, 2024. URL https://arxiv.org/abs/ 2406.03476.
  2. J. S. B. T. Evans. Heuristic and analytic processes in reasoning. British Journal of Psychology, 75(4): 451-468, 1984.
  3. L. Gao, A. Madaan, S. Zhou, U. Alon, P. Liu, Y. Yang, J. Callan, and G. Neubig. Pal: Program-aided language models, 2023. URL https://arxiv.org/abs/2211.10435. · arXiv
  4. S. Goyal, Z. Ji, A. S. Rawat, A. K. Menon, S. Kumar, and V. Nagarajan. Think before you speak: Train- ing language models with pause tokens, 2024. URL https://arxiv.org/abs/2310.02226. · arXiv
  5. A. L. Jones. Scaling scaling laws with board games, 2021. URL https://arxiv.org/abs/2104. 03113.
  6. D. Kahneman. Maps of bounded rationality: Psychology for behavioral economics. The American Economic Review, 93(5):1449-1475, 2003.
  7. D. Kahneman. Thinking, fast and slow. Farrar, Straus and Giroux, New York, first paperback edition edition, 2013.
  8. L. Kocsis and C. Szepesv'ari. Bandit based monte-carlo planning. In European conference on machine learning, pages 282-293. Springer, 2006.
  9. Y. Qin, S. Liang, Y. Ye, K. Zhu, L. Yan, Y. Lu, Y. Lin, X. Cong, X. Tang, B. Qian, S. Zhao, L. Hong, R. Tian, R. Xie, J. Zhou, M. Gerstein, D. Li, Z. Liu, and M. Sun. Toolllm: Facilitating large language models to master 16000+ real-world apis, 2023. URL https://arxiv.org/abs/2307.16789. · arXiv
  10. C. Qu, S. Dai, X. Wei, H. Cai, S. Wang, D. Yin, J. Xu, and J.-R. Wen. Tool learning with large language models: A survey, 2024. URL https://arxiv.org/abs/2405.17935. · arXiv
  11. W. Saunders, C. Yeh, J. Wu, S. Bills, L. Ouyang, J. Ward, and J. Leike. Self-critiquing models for assisting human evaluators, 2022.
  12. A. Setlur, S. Garg, X. Geng, N. Garg, V. Smith, and A. Kumar. Rl on incorrect synthetic data scales the efficiency of llm math reasoning by eight-fold. arXiv preprint arXiv:2406.14532, 2024. · arXiv
  13. A. Sharma, S. Keh, E. Mitchell, C. Finn, K. Arora, and T. Kollar. A critical evaluation of ai feedback for aligning large language models, 2024. URL https://arxiv.org/abs/2402.12366. · arXiv
  14. C. Snell, E. Wallace, D. Klein, and S. Levine. Predicting emergent capabilities by finetuning. Conference on Language Modeling 2024, 2024.
  15. R. S. Sutton and A. G. Barto. Reinforcement learning: An introduction. Second edition, 2018.
  16. H. Touvron, L. Martin, K. Stone, P. Albert, A. Almahairi, Y. Babaei, N. Bashlykov, S. Batra, P. Bhargava, S. Bhosale, D. Bikel, L. Blecher, C. C. Ferrer, M. Chen, G. Cucurull, D. Esiobu, J. Fernandes, J. Fu, W. Fu, B. Fuller, C. Gao, V. Goswami, N. Goyal, A. Hartshorn, S. Hosseini, R. Hou, H. Inan, M. Kardas, V. Kerkez, M. Khabsa, I. Kloumann, A. Korenev, P. S. Koura, M.-A. Lachaux, T. Lavril, J. Lee, D. Liskovich, Y. Lu, Y. Mao, X. Martinet, T. Mihaylov, P. Mishra, I. Molybog, Y. Nie, A. Poulton, J. Reizenstein, R. Rungta, K. Saladi, A. Schelten, R. Silva, E. M. Smith, R. Subramanian, X. E. Tan, B. Tang, R. Taylor, A. Williams, J. X. Kuan, P. Xu, Z. Yan, I. Zarov, Y. Zhang, A. Fan, M. Kambadur, S. Narang, A. Rodriguez, R. Stojnic, S. Edunov, and T. Scialom. Llama 2: Open foundation and fine-tuned chat models, 2023. URL https://arxiv.org/abs/2307.09288. · arXiv
  17. R. Wang, E. Zelikman, G. Poesia, Y. Pu, N. Haber, and N. D. Goodman. Hypothesis search: Inductive reasoning with language models, 2024. URL https://arxiv.org/abs/2309.05660. · arXiv
  18. E. Zelikman, G. Harik, Y. Shao, V. Jayasiri, N. Haber, and N. D. Goodman. Quiet-star: Language models can teach themselves to think before speaking, 2024. URL https://arxiv.org/abs/ 2403.09629.

Cited by

Sorted by citations

Authors
Snell, C., Lee, J., Xu, K., & Kumar, A.
Affiliations
Google (United States) (US)
Venue
arXiv
Publisher
arXiv
Published
2024
ISSN
2331-8422 (ISSN-L)
OpenAlex
W4406492615
Cited by
27
References
0
Citations by year
2026: 5 · 2025: 20 · 2024: 2
Type
posted content
Language
en
Open access
Locations
  • arXiv (Cornell University) (submittedVersion, cc-by)
  • arXiv (Cornell University) (cc-by)
Topics
  • Physical Sciences › Computer Science › Artificial Intelligence › Topic Modeling
  • Physical Sciences › Computer Science › Artificial Intelligence › Natural Language Processing Techniques
  • Physical Sciences › Computer Science › Artificial Intelligence › Adversarial Robustness in Machine Learning
Keywords
large language models, test-time compute scaling, self-improving agents

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review