mlsanityData to 14 Oct 2026

A Hierarchical Language Model with Predictable Scaling Laws and Provable Benefits of Reasoning

arXiv 2605.13687 ↗First posted 13 May 2026cs.LG cs.AI stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

We introduce a family of synthetic languages with hierarchical structure -- generated by a broadcast process on trees -- for which the role of context length and reasoning in autoregressive generation can be analyzed precisely. At the heart of our analytic approach is an exact k-gram ansatz in place of transformers with context length k, a substitution we then validate empirically. Using this ansatz we derive explicit asymptotic predictions for distributional statistics of the sequences produced by a trained model, instantiated in two settings. For the Ising broadcast process (a soft-constrained language), we prove that the variance of the generated sum scales log-linearly in the context depth and its kurtosis converges to that of a Gaussian -- both deviating from the true language for any sublinear context. For the coloring broadcast process (a hard-constrained language) in the freezing regime, bounded-context autoregression produces sequences that, with high probability, are inconsistent with any valid coloring of the underlying tree. Together these results imply an Omega(n) lower bound on the context length required to faithfully sample length-n sequences. In contrast, we prove that an autoregressive reasoning model with only Theta(log n) working memory can sample exactly from the true language -- an exponential improvement. We confirm both the lower-bound predictions and the reasoning-based upper bound empirically with transformers trained on the synthetic language; the trained models track our asymptotic predictions quantitatively across a wide range of context sizes.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 23 of its 55 references that are in the map. First posted on arXiv 13 May 2026.

Builds on

23 of 55 references.

References · 23

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Scaling Laws for Neural Language Modelscited by 6,586
  3. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
  4. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  5. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  6. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
  7. 2020Linformer: Self-Attention with Linear Complexitycited by 1,474
  8. 2025DeepSeek-V3.2: Pushing the Frontier of Open Large Language Modelscited by 815
  9. 2015Improved Semantic Representations From Tree-Structured Long Short-Term Memory Networkscited by 306
  10. 2023Effective Long-Context Scaling of Foundation Modelscited by 303
  11. 2023Towards Revealing the Mystery behind Chain of Thought: A Theoretical Perspectivecited by 294
  12. 2022The Parallelism Tradeoff: Limitations of Log-Precision Transformerscited by 193
11 more
  1. 2023How Deep Neural Networks Learn Compositional Data: The Random Hierarchy Modelcited by 71
  2. 2023Auto-Regressive Next-Token Predictors are Universal Learnerscited by 67
  3. 2023Tighter Bounds on the Expressivity of Transformer Encoderscited by 66
  4. 2024Theoretical limitations of multi-layer Transformercited by 45
  5. 2023Logical Languages Accepted by Transformer Encoders with Hard Attentioncited by 38
  6. 2016Deep Learning and Hierarchal Generative Modelscited by 25
  7. 2025A Theory of Learning with Autoregressive Chain of Thoughtcited by 23
  8. 2016Prediction with a Short Memorycited by 13
  9. 2021Reconstruction on Trees and Low-Degree Polynomialscited by 10
  10. 2024How Deep Networks Learn Sparse and Hierarchical Data: the Sparse Random Hierarchy Modelcited by 7
  11. 2026Provable Learning of Random Hierarchy Models and Hierarchical Shallow-to-Deep Chainingcited by 6

Built on by

Not cited yet.