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

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-N 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'.

Authors

Institutions

Publication Details

Journal
Philosophical Transactions of the Royal Society A Mathematical Physical and Engineering Sciences
Published
2026-02-26
DOI
https://doi.org/10.1098/rsta.2024.0510
Citations
1
Primary Topic
Multimodal Machine Learning Applications
Type
article
Field-Weighted Citation Impact
46.84
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

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

Anuj K. Nayak, Austin Ellis‐Mohr
1 citations
Philosophical Transactions of the Royal Society A Mathematical Physical and Engineering Sciences
Multimodal Machine Learning Applications
46.84
article

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

Anuj K. Nayak, Austin Ellis‐Mohr
article en
1 citations

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-N 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'.

Philosophical Transactions of the Royal Society A Mathematical Physical and Engineering SciencesVol. 384(2315)
University of Illinois Urbana-Champaign (US), Stony Brook University (US)
Responsible consumption and production
Openalex Percentile: Top 0%
Multimodal Machine Learning Applications
46.84
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.