arXiv:2605.13687cs.LGcs.AI2026-05

提出可分析的分层语言模型,证明推理能极大降低上下文需求。

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

论文配图:A Hierarchical Language Model with Predictable Scaling Laws and Provable Benefits of Reasoning
图 1 · 摘自论文原文
  • 用树状广播过程生成分层语言,精确建模上下文与推理作用。
  • 上下文不足时生成序列会严重偏离真实分布,需至少线性长度上下文。
  • 引入推理机制仅需对数级记忆即可准确采样,适合研究大模型原理。

我们提出一类具有层级结构的合成语言,由树上的广播过程生成,可精确分析自回归生成中上下文长度与推理的作用。核心是用精确的 $k$-gram 假设替代变压器模型,其上下文长度为 $k$,并通过实证验证该假设的有效性。利用该假说,我们推导出训练后模型生成序列的分布统计量的渐近预测,分别在两种设定下实现:对于(软约束)伊辛广播过程,生成和的方差随上下文深度呈对数线性增长,峰度收敛于高斯分布——但任何亚线性上下文均偏离真实语言;对于(硬约束)着色广播过程在冻结区域,有限上下文的自回归生成在高概率下产生与任何有效着色不一致的序列。上述结果共同表明,忠实采样长度为 $n$ 的序列所需上下文长度至少为 $Ω(n)$。相比之下,我们证明仅需 $Θ(\log n)$ 工作内存的自回归推理模型可精确采样真实语言——实现指数级改进。我们通过在合成语言上训练的变压器模型,实证验证了下界预测与基于推理的上界,训练模型在广泛上下文尺寸下与渐近预测高度一致。

原文摘要 · Abstract (English)

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 \emph{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 \emph{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 \emph{coloring broadcast process} (a hard-constrained language) in the freezing regime, bounded-context autoregression produces sequences that, with high probability, are inconsistent with \emph{any} valid coloring of the underlying tree. Together these results imply an $Ω(n)$ lower bound on the context length required to faithfully sample length-$n$ sequences. In contrast, we prove that an autoregressive \emph{reasoning} model with only $Θ(\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.

语言建模推理机制理论分析上下文长度

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。