Skip to main navigation Skip to search Skip to main content

A theory of inference compute scaling: reasoning through directed stochastic skill search

  • University of Illinois at Urbana-Champaign

Research output: Contribution to journalArticlepeer-review

1 Scopus citations

Abstract

Abstract – Large language models (LLMs) require considerable computation and energy resources during training and deployment. While scaling laws for training have guided much recent progress, inference costs represent a significant and growing component of the overall resource burden, particularly for reasoning models. Existing compute-optimality characterizations that consider model size, dataset size and inference tokens in isolation or fixed combinations may overlook more efficient operating points. We introduce directed stochastic skill search (DS3), a general framework that represents inference as stochastic traversal over a learnt skill graph. From a simplified yet expressive instantiation, we derive closed-form expressions for task success and compute cost across a wide range of inference strategies—including chain-of-thought (CoT) and tree-of-thought (ToT)—enabling comparisons by task difficulty and model capability. We extend a prior graph framework of LLM training to include inference and bridge DS3 with empirical scaling laws. We theoretically recover observed patterns, including linear accuracy scaling with log-compute, variation in preferred inference strategies by task and capability, emergent behaviour elicited by reasoning despite parameter plateaus and both best-of-(Formula presented) and majority voting (MV) captured within one framework. By characterizing training-inference interdependencies, our framework deepens theoretical understanding and supports principled algorithmic design and resource allocation. This article is part of the discussion meeting issue ‘Bits, neurons and qubits for sustainable AI’.

Original languageEnglish
Article number20240510
JournalPhilosophical Transactions of the Royal Society A: Mathematical, Physical and Engineering Sciences
Volume384
Issue number2315
DOIs
StatePublished - Feb 26 2026

Keywords

  • inference scaling
  • inference-time compute
  • large language models
  • reasoning
  • scaling laws
  • test-time compute
  • train-time compute

Fingerprint

Dive into the research topics of 'A theory of inference compute scaling: reasoning through directed stochastic skill search'. Together they form a unique fingerprint.

Cite this