A Measure-Theoretic Analysis of Reasoning: Structural Generalization and Approximation Limits
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
While empirical scaling laws for LLM reasoning are well-documented, the theoretical mechanisms governing out-of-distribution (OOD) generalization remain elusive. We formalize reasoning via optimal transport, projecting discrete trajectories into a continuous metric space to quantify domain shifts using the Wasserstein-1 distance. Invoking Kantorovich duality, we bound OOD generalization via architectural Lipschitz continuity and functional approximation limits. This exposes two primary constraints. First, position-dependent attention (e.g., Absolute Positional Encoding) fails to preserve shift invariance, yielding an Omega(1) Lipschitz constant and expected risk, whereas shift-invariant mechanisms (e.g., Rotary Embeddings) preserve equivariance and bound the error. Second, by mapping sequential backtracking to a Dyck-k language, we establish a strict circuit depth lower bound for TC0 Transformers. Scaling physical layer depth is necessary to avert representation collapse -- a constraint that scaling representation width cannot bypass due to irreducible approximation bounds in Barron spaces. Evaluations across 54 Transformer configurations on combinatorial search corroborate these bounds, demonstrating that generalization risk degrades monotonically with the Wasserstein domain shift.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 24 of its 35 references that are in the map. First posted on arXiv 19 May 2026.
Builds on
24 of 35 references.
Closest ancestors · 2
Same direction, newest first
- 2024Stream of Search (SoS): Learning to Search in Language1 Apr
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Models17 May
Other references · 22
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
- 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,316
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,103
- 2024Tulu 3: Pushing Frontiers in Open Language Model Post-Trainingcited by 1,053
- 2015Optimal Transport for Domain Adaptationcited by 463
- 2023The Impact of Positional Encoding on Length Generalization in Transformerscited by 327
- 2023Towards Revealing the Mystery behind Chain of Thought: A Theoretical Perspectivecited by 294
- 2022The Parallelism Tradeoff: Limitations of Log-Precision Transformerscited by 193
- 2015The Power of Depth for Feedforward Neural Networkscited by 183
10 more
- 2023Randomized Positional Encodings Boost Length Generalization of Transformerscited by 117
- 2016Theoretical Analysis of Domain Adaptation with Optimal Transportcited by 105
- 2023Why think step by step? Reasoning emerges from the locality of experiencecited by 103
- 2025Is Chain-of-Thought Reasoning of LLMs a Mirage? A Data Distribution Lenscited by 43
- 2020The Depth-to-Width Interplay in Self-Attentioncited by 42
- 2023Length Extrapolation of Transformers: A Survey from the Perspective of Positional Encodingcited by 32
- 2024DAPE: Data-Adaptive Positional Encoding for Length Extrapolationcited by 31
- 2024Two Stones Hit One Bird: Bilevel Positional Encoding for Better Length Extrapolationcited by 24
- 2023Logit Margin Matters: Improving Transferable Targeted Adversarial Attack by Logit Calibrationcited by 18
- 2023Functional Interpolation for Relative Positions Improves Long Context Transformerscited by 18
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Bridging Reasoning to Learning: Unmasking Illusions using Complexity Out of Distribution Generalization
- 2022Achieving and Understanding Out-of-Distribution Generalization in Systematic Reasoning in Small-Scale Transformers
- 2025Large Language and Reasoning Models are Shallow Disjunctive Reasoners
- 2026Beyond What Seems Necessary: Hidden Gains from Scaling Training-Time Reasoning Length under Outcome Supervision
- 2026LLM Reasoning as Trajectories: Step-Specific Representation Geometry and Correctness Signals
- 2026An Asymptotic Theory of Chain-of-Thought in In-Context Learning
- 2026Boule or Baguette? A Study on Task Topology, Length Generalization, and the Benefit of Reasoning Traces
- 2026Tool Use Reduces Depth-Induced Collapse in OOD Reasoning