Reasoning about Reasoning: BAPO Bounds on Chain-of-Thought Token Complexity in LLMs
Kiran Tomlinson, Tobias Schnabel, Adith Swaminathan, Jennifer Neville
TL;DR
This work analyzes how many reasoning tokens a large language model must generate as input size grows, using the bounded attention prefix oracle (BAPO) framework to model information flow within transformers. It proves that three classic BAPO-hard tasks—Majority, Match3$_n$, and Reachability—require linear CoT length, i.e., $c(n)=\Omega(n)$, under constant-bandwidth BAPO-CoTs, and provides matching upper bounds via explicit constructions. The authors introduce a self-consistent variant, cBAPO, to avoid pathological input-doubling loopholes and show that the same lower bounds hold while enabling tighter, constructive upper bounds (e.g., Majority $O(n\log n)$, Match3$_n$ $O(n)$, Reachability $O(n^2)$). Experiments with frontier reasoning models corroborate the theory, revealing approximately linear scaling of internal reasoning tokens and showcasing the practical impact on inference-time costs, while also highlighting opportunities for efficiency gains through decomposition or new architectures. Overall, the paper offers a principled framework to reason about optimal reasoning length and its consequences for real-world scaling and compression of chain-of-thought in LLMs.
Abstract
Inference-time scaling via chain-of-thought (CoT) reasoning is a major driver of state-of-the-art LLM performance, but it comes with substantial latency and compute costs. We address a fundamental theoretical question: how many reasoning tokens are required to solve a problem as input size grows? By extending the bounded attention prefix oracle (BAPO) model--an abstraction of LLMs that quantifies the information flow required to solve a task--we prove lower bounds on the CoT tokens required for three canonical BAPO-hard tasks: binary majority, triplet matching, and graph reachability. We show that each requires $Ω(n)$ reasoning tokens when the input size is $n$. We complement these results with matching or near-matching upper bounds via explicit constructions. Finally, our experiments with frontier reasoning models show approximately linear reasoning token scaling on these tasks and failures when constrained to smaller reasoning budgets, consistent with our theoretical lower bounds. Together, our results identify fundamental bottlenecks in inference-time compute through CoT and offer a principled tool for analyzing optimal reasoning length.
