paper-with-me

홈 › Papers

Reasoning about Reasoning: BAPO Bounds on Chain-of-Thought Token Complexity in LLMs

2026-02-02 · Kiran Tomlinson, Tobias Schnabel, Adith Swaminathan, Jennifer Neville arxiv

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.

📄 PDF Abstract BibTeX arXiv:2602.02909

Code (0)

등록된 구현이 없습니다.

Similar Papers 제목 키워드 기반

Lost in Transmission: When and Why LLMs Fail to Reason Globally

2025-05-13 · Tobias Schnabel, Kiran Tomlinson, Adith Swaminathan, Jennifer Neville

Despite their many successes, transformer-based large language models (LLMs) continue to struggle with tasks that require complex reasoning over large parts of their input. We argue that these failures arise due to capac…

BAPO: Boundary-Aware Policy Optimization for Reliable Agentic Search

2026-01-16 · Shiyu Liu, Yongjing Yin, Jianhao Yan, Yunbo Tang 외 arxiv

RL-based agentic search enables LLMs to solve complex questions via dynamic planning and external search. While this approach significantly enhances accuracy with agent policies optimized via large-scale reinforcement le…

Reinforcement Learning

Buffer Matters: Unleashing the Power of Off-Policy Reinforcement Learning in Large Language Model Reasoning

2026-02-24 · Xu Wan, Yansheng Wang, Wenqi Huang, Mingyang Sun arxiv

Traditional on-policy Reinforcement Learning with Verifiable Rewards (RLVR) frameworks suffer from experience waste and reward homogeneity, which directly hinders learning efficiency on difficult samples during large lan…

Reinforcement LearningVisual Reasoning

Lower Bounds for Chain-of-Thought Reasoning in Hard-Attention Transformers

2025-02-04 · Alireza Amiri, Xinting Huang, Mark Rofin, Michael Hahn

Chain-of-thought reasoning and scratchpads have emerged as critical tools for enhancing the computational capabilities of transformers. While theoretical results show that polynomial-length scratchpads can extend transfo…

Hard Attention

BAPO: Stabilizing Off-Policy Reinforcement Learning for LLMs via Balanced Policy Optimization with Adaptive Clipping

2025-10-21 · Zhiheng Xi, Xin Guo, Yang Nan, Enyu Zhou 외 arxiv

Reinforcement learning (RL) has recently become the core paradigm for aligning and strengthening large language models (LLMs). Yet, applying RL in off-policy settings--where stale data from past policies are used for tra…

Reinforcement Learning