Supplementary material from "A Theory of Inference Compute Scaling: Reasoning through Directed Stochastic Skill Search"

Read full text
Authors: Austin R. Ellis-Mohr, Anuj Keshava Nayak, Lav R. Varshney
Cited by: 0

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 learned 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 and tree-of-thought---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 behavior elicited by reasoning despite parameter plateaus; and both best-of-N and majority voting captured within one framework. By characterizing training-inference interdependencies, our framework deepens theoretical understanding and supports principled algorithmic design and resource allocation.

Fields of Study & Subjects

Topic Modeling Natural Language Processing Techniques Advanced Neural Network Applications

Publication details

Journal / source
Figshare
Publication date
1 January 2026
Language
EN