---
title: PAC Learning for Next-Token Generators
url: https://www.emergentmind.com/topics/pac-learning-framework-for-next-token-generators
type: topic
---

# PAC Learning for Next-Token Generators

Searching arXiv for the specified paper and closely related work on PAC-style/autoregressive next-token learning.
to=arxiv_search.query  北京赛车开奖ված 下载彩神争霸
{"query":"all:\"Sample Complexity of Autoregressive Reasoning: Chain-of-Thought vs. End-to-End\" OR id:2604.12013","max_results":5,"sort_by":"relevance"}
The PAC-learning framework for next-token generators studies autoregressive text generation as a statistical learning problem in which an unknown next-token rule is iteratively applied for \(T\) steps, and the learning target is the input-output map induced by that rollout rather than merely the one-step predictor. In the formulation introduced by Joshi et al. and sharply extended in "Sample Complexity of Autoregressive Reasoning: Chain-of-Thought vs. End-to-End," a next-token generator maps a sequence to its next token, rollout length \(T\) becomes an explicit complexity parameter, and the supervision model—observing only the final token or the entire intermediate trajectory—determines whether sample complexity can scale linearly in \(T\), logarithmically in \(T\), or not at all [2604.12013].

## 1. Formal autoregressive PAC model

The basic object is a next-token generator
\[
f:\Sigma^\star \to \Sigma,
\]
where \(\Sigma\) is a finite alphabet and \(\Sigma^\star\) is the set of finite strings over \(\Sigma\). One autoregressive step is represented by the apply-and-append map
\[
\bar f:\Sigma^\star\to\Sigma^\star,\qquad \bar f(x)=x\circ f(x),
\]
and iterating this map for \(T\) steps produces a generated chain whose length-\(T\) suffix is denoted \(f^{\mathsf{CoT},T}(x)\in\Sigma^T\), while its last token is denoted \(f^{\mathsf{e2e},T}(x)\in\Sigma\) [2604.12013].

This formulation makes the PAC target explicitly rollout-based. For a base class \(\mathcal F\subseteq \Sigma^{\Sigma^\star}\), the induced classes are
\[
\mathcal F^{\mathsf{CoT},T} = \{f^{\mathsf{CoT},T}:f\in\mathcal F\}, \qquad \mathcal F^{\mathsf{e2e},T} = \{f^{\mathsf{e2e},T}:f\in\mathcal F\}.
\]
Accordingly, the learning task is not necessarily to identify the one-step mechanism \(f\), but to learn the map from prompt \(x\) to the final token obtained after \(T\) rounds of autoregressive generation.

The paper studies primarily the binary case \(\Sigma=\{0,1\}\), while stating that the ideas extend to any finite alphabet. The learning setup is realizable: an unknown distribution \(D\) over \(\Sigma^\star\) supplies i.i.d. prompts \(x\sim D\), and labels are generated by some unknown \(f_\star\in F\). Test-time error is always the end-to-end error
\[
\Pr_{x\sim D}\bigl[h(x)\neq f_\star^{\mathsf{e2e},T}(x)\bigr].
\]
For finite-VC classes, the usual PAC sample complexity notation \(m_{\mathcal H}(\epsilon,\delta)\) is used, with
\[
m_{\mathcal H}(\epsilon,\delta) = \Theta\!\left(\frac{d+\log(1/\delta)}{\epsilon}\right), \qquad d=VC(\mathcal H).
\]
Thus, in the end-to-end regime, the problem is exactly PAC learning of \(F^{\mathsf{e2e},T}\), governed by \(VC(F^{\mathsf{e2e},T})\).

A common misconception is that next-token PAC theory concerns only the one-step predictor. In this framework, the relevant hypothesis class is the class induced by \(T\) rounds of rollout, so the learnability question is intrinsically sequential rather than pointwise.

## 2. Supervision models: end-to-end and Chain-of-Thought

The framework formalizes two supervision models. Under **End-to-End supervision**, a training example reveals only the final token,
\[
\bigl(x, f_\star^{\mathsf{e2e},T}(x)\bigr).
\]
Under **Chain-of-Thought supervision**, a training example reveals the full generated trajectory,
\[
\bigl(x, f_\star^{\mathsf{CoT},T}(x)\bigr),
\]
where \(f_\star^{\mathsf{CoT},T}(x)\in\Sigma^T\) [2604.12013].

The distinction is asymmetric. Test-time evaluation remains end-to-end even in the CoT regime: the learner is still judged only on the final token. The paper is explicit that CoT supervision is a training-time side-information model, not an ordinary symmetric PAC classification task. This point matters because observing the entire intermediate chain supplies \(T\) local constraints on the latent one-step generator, whereas ordinary end-to-end supervision exposes only the terminal outcome.

The central positive theorem for CoT supervision states that for every class \(F\) of binary next-token generators with \(VC(F)<\infty\), there exists a constant \(c=c(F)\) such that for every \(T\ge 1\),
\[
m_{F}^{\mathsf{CoT},T}(\epsilon,\delta) \le \frac{c}{\epsilon} \left( \log\frac{1}{\epsilon} + \log\frac{1}{\delta} \right).
\]
More precisely, the proof yields \(c=\tilde O(dd^\star)\), where \(d=VC(F)\) and \(d^\star=VC^\star(F)\) is the dual VC dimension. The stronger multiclass statement is that if \(VC(F)<\infty\), then \(F^{\mathsf{CoT},T}\) is learnable for all \(T\) with
\[
m(\epsilon,\delta)=\tilde O\!\left(\frac{VC(F)\,VC^\star(F)+\log(1/\delta)}{\epsilon}\right),
\]
while if \(VC(F)=\infty\), then \(F^{\mathsf{CoT},T}\) is not PAC learnable for any \(T\) [2604.12013].

The lower bound is already present at \(T=1\):
\[
\Omega\!\left(\frac{d+\log(1/\delta)}{\epsilon}\right).
\]
For classes with stable compression, the paper gives the sharper bound
\[
m_{F}^{\mathsf{CoT},T}(\epsilon,\delta) = O\!\left(\frac{k+\log(1/\delta)}{\epsilon}\right)
\]
whenever \(F\) admits a stable sample compression scheme of size \(k\). This applies in particular to the linear autoregressor class
\[
f_{w,\theta}(x) = \mathbf 1\!\left[\langle w,\mathrm{tail}_d(x)\rangle \ge \theta\right],
\]
yielding
\[
m = O\!\left(\frac{d+\log(1/\delta)}{\epsilon}\right).
\]

The paper’s conceptual interpretation is direct: CoT supervision turns one difficult \(T\)-step autoregressive problem into \(T\) ordinary one-step supervision constraints on the base class. This suggests that observing intermediate computation can remove the statistical burden created by hidden latent rollouts.

## 3. Sample-complexity taxonomy as a function of rollout length

The most distinctive result of the framework is a taxonomy of how sample complexity scales with generation length \(T\). In the CoT regime, the dependence on \(T\) disappears entirely. In the end-to-end regime, by contrast, the space of possibilities is “remarkably rich”: subject to mild conditions, essentially any growth rate between constant and linear can occur [2604.12013].

Joshi et al. had established the general upper bound
\[
VC(F^{e2e,T}) = O(VC(F)\cdot T),
\]
and examples with linear growth, while also showing that finite Littlestone dimension implies logarithmic growth. The later theory proves that intermediate behaviors are not artifacts but genuinely realizable. A function \(r:\mathbb N_+\to\mathbb N_+\) is called a monotone-subadditive rate if it is monotone non-decreasing and satisfies
\[
r(T_1+T_2)\le r(T_1)+r(T_2).
\]
Then for every such rate \(r\), there exists a class \(F\) of binary next-token generators such that
\[
r(T) \le VC(F^{e2e,T}) \le r(T)+r(1)\le 2r(T)
\qquad\text{for all }T\ge 1.
\]
In the normalized case \(r(1)=1\), the exact identity
\[
VC(F^{e2e,T})=r(T)
\qquad\text{for all }T
\]
is obtained.

The realizable rates include constant growth, \(\log T\), polylogarithmic growth, power laws \(T^c\) for \(0<c<1\) after integer rounding, and rates arbitrarily close to linear. Since
\[
m_{F^{e2e,T}}(\epsilon,\delta) =
\Theta\!\left( \frac{VC(F^{e2e,T})+\log(1/\delta)}{\epsilon} \right),
\]
the same taxonomy transfers immediately to end-to-end sample complexity.

The constructive part of the theory begins with the class
\[
f_a(x):= \begin{cases} a_{|x|+1} & \text{if } x \text{ is a prefix of } a,\\
0 & \text{otherwise,} \end{cases}
\]
for infinite binary sequences \(a\in\{0,1\}^{\mathbb N}\), and
\[
F_{\mathrm{full}}:=\{f_a:a\in\{0,1\}^{\mathbb N}\}.
\]
This class has
\[
VC(F_{\mathrm{full}})=1, \qquad VC(F_{\mathrm{full}}^{e2e,T})=T.
\]
Thus even a VC-dimension-1 base class can exhibit maximal linear growth in \(T\). To realize arbitrary sublinear rates, the theory restricts the informative bit positions through subsets \(N\subseteq\mathbb N\), defining
\[
r_N(T):= \max_{u\in\mathbb N}\,|N\cap[u+1,u+T]|.
\]
Then there exists a class \(F(N)\) such that
\[
VC(F(N)^{e2e,T})=r_N(T),
\]
and moreover
\[
r=r_N \iff r\text{ is monotone non-decreasing, subadditive, and }r(1)=1.
\]

A plausible implication is that end-to-end autoregressive learnability has no small list of canonical regimes. The framework instead identifies a broad continuum of possible statistical behaviors.

## 4. Combinatorial dimensions, impossibility results, and proof mechanisms

The same theory shows that no single combinatorial dimension can characterize all sublinear end-to-end rates. It defines a putative characterization as a map \(Dim(F)\in \mathbb N\cup\{\infty\}\) together with an upper bound \(M(d,T)\) such that \(Dim(F)<\infty\) iff \(VC(F^{e2e,T})=o(T)\), and such that for each fixed \(d\), \(M(d,T)=o(T)\). The resulting theorem is negative: there is no dimension that characterizes sublinear rates in this sense [2604.12013].

Although no full dimension theory exists for all sublinear behavior, the paper introduces a new sufficient parameter, the **autoregressive tree dimension** \(ATdim(F)\). For a prompt \(x\) and rollout depth \(T\), the set of realized trajectories \(\{f^{\mathsf{CoT},T}(x):f\in F\}\subseteq\{0,1\}^T\) induces a prefix tree \(G_{F,x,T}\). The quantity \(ATdim(F)\) is the largest depth of a perfect leveled binary subtree realized somewhere in these generation trees. The main quantitative upper bound is
\[
VC(F^{e2e,T}) \le 20\,ATdim(F)\,VC(F)\,\log T
\]
for sufficiently large \(T\), and in asymptotic form
\[
VC(F^{e2e,T}) = O\!\bigl(ATdim(F)\cdot VC(F)\cdot \log T\bigr).
\]

This strictly strengthens the earlier sufficient condition based on finite Littlestone dimension because
\[
\max\{VC(F),ATdim(F)\}\le L(F),
\]
yet the paper also constructs classes with
\[
VC(F)=ATdim(F)=1 \qquad\text{and}\qquad L(F)=\infty.
\]
So \(ATdim\) can be finite even when Littlestone dimension is infinite.

The proof mechanisms are equally characteristic. For CoT upper bounds, the central idea is to **inflate** each chain-of-thought sample into \(T\) binary examples for the base class. If the sample contains \((x_i,y_i)\) with \(y_i=(y_{i,1},\dots,y_{i,T})\), then
\[
u_{i,t}:=\bigl(x_i\circ y_{i,\le t-1},\, y_{i,t}\bigr), \qquad t\in[T],
\]
yields an ordinary labeled sample for \(F\). The technical difficulty is then to compress the inflated sample without paying a factor of \(T\). The paper adapts the majority-vote compression scheme of Moran and Yehudayoff to obtain compression size \(\tilde O(VC(F)VC^\star(F))\), independent of \(T\).

For logarithmic end-to-end upper bounds under finite \(ATdim\), the analysis relies on a new Sauer-type lemma for leveled binary trees: if a depth-\(T\) tree contains no perfect leveled binary subtree of depth \(d\), then the number of leaves is at most
\[
\binom{T}{\le d}.
\]
Applied to realized generation trees, this gives
\[
|B_F(T_T(x))| \le T^{2\,ATdim(F)},
\]
which replaces the trivial \(2^T\) branch count.

The framework also emphasizes that structural assumptions are essential. Outside the finite-VC setting, pathological behavior can occur: there exists a base class \(F\) such that for every even \(T\),
\[
VC(F^{e2e,T})=0,
\]
while for every odd \(T\), the problem is not learnable even under CoT supervision. This is a direct warning against extrapolating the positive theory beyond its stated hypotheses.

## 5. Related research directions around next-token PAC theory

The broader literature develops several adjacent, but non-identical, learning-theoretic frameworks for autoregressive next-token prediction.

"Auto-Regressive Next-Token Predictors are Universal Learners" defines **AR Learnable** as a PAC analogue for sequence prediction under realizability, proves a transfer theorem stating that if \(\mathcal H_1,\dots,\mathcal H_T\) are PAC learnable with sample complexity \(m(\epsilon,\delta)\), then the product class is AR learnable with sample complexity \(m(\epsilon/T,\delta/T)\), and introduces **length complexity** as the number of intermediate CoT tokens needed to compute or approximate a target function [2309.06979]. That theory is explicit that it is not a full PAC theory in the classical sense, but it gives a formal bridge between per-step PAC learnability and autoregressive sequence learning.

"Towards Auto-Regressive Next-Token Prediction: In-Context Learning Emerges from Generalization" develops a **PAC-Bayesian** framework for pre-trained next-token generators under a hierarchical topic-conditioned sequence model. Its population loss is expressed as an expected KL divergence between the true and model next-token conditionals, and the main result is a two-level decomposition of population loss into empirical loss, sequence-generalization gaps, and topic-generalization gaps, together with data-dependent, topic-dependent, and optimization-dependent PAC-Bayesian bounds [2502.17024].

"Hardness of Learning Regular Languages in the Next Symbol Prediction Setting" formalizes a PAC-style model for learning the truncated support behavior of a next-token generator from positive examples annotated with per-prefix continuation sets and acceptance bits. Its main negative result is that, despite this richer supervision, efficient PAC learning of polynomial-size acyclic DFAs in the NSP model would imply efficient PAC learning in the conventional classification model; under Kearns–Valiant cryptographic assumptions, weak learning remains hard [2510.18634]. This corrects the possible misconception that richer token-level supervision necessarily removes computational hardness.

"Provable Long-Range Benefits of Next-Token Prediction" studies a different guarantee: minimizing next-token log loss over an appropriate class of RNN language models yields generators that are \(\varepsilon\)-indistinguishable from the training distribution for all bounded-size next-\(k\)-token RNN distinguishers, with model-size bounds polynomial in \(k\) and independent of document length \(n\) [2512.07818]. The guarantee is complexity-theoretic rather than PAC in the classical sample-complexity sense.

"The Role of Generator Access in Autoregressive Post-Training" shifts attention from hypothesis complexity to oracle structure. In the no-reset or root-start regime, output sampling, generated-token log probabilities, top-\(k\) reports, and full next-token distributions along sampled trajectories collapse to one canonical experiment, and transcript distinguishability is bounded by \(q\,Reach_M(x,U)\), where \(Reach_M(x,U)\) is the on-policy probability of reaching informative prefixes. Weak prefix control breaks this barrier, and chosen-prefix sampling or logits can outperform top-1 access; for KL-regularized outcome-reward post-training, the paper proves an exponential gap between no-reset access and chosen-prefix access [2604.04855].

"Markov Chain Estimation with In-Context Learning" contributes an experimental benchmark rather than a PAC theorem, but it is directly relevant to the framework’s intuitions. Transformers trained only with next-token cross-entropy on sequences from families of Markov chains display a threshold phenomenon in model size and number of training chains: below the threshold they underfit or memorize, and above it they generalize to unseen transition matrices by estimating transition probabilities from context [2508.03934].

## 6. Assumptions, scope, and enduring significance

The PAC-learning framework for next-token generators is deliberately narrow in several respects. The main positive structural results assume a finite alphabet, realizability, deterministic next-token generators, finite \(VC(F)\), bounded rollout length \(T\) as an explicit parameter, and an information-theoretic notion of learnability rather than a general computational-efficiency guarantee [2604.12013]. The theory does not provide agnostic or noisy-label guarantees, does not analyze probabilistic next-token generators \(p_\theta(z_t\mid x,z_{<t})\) directly, and does not supply a general optimization theory for practical language-model training.

Within that scope, however, it provides a sharp answer to two foundational questions. First, end-to-end autoregressive learning does not have a single canonical dependence on reasoning length: essentially every monotone-subadditive rate between constant and linear can occur. Second, CoT supervision can eliminate dependence on reasoning length altogether. This suggests a precise statistical interpretation of reasoning traces: they expose latent intermediate states of autoregressive computation and thereby convert hidden multi-step inference into directly supervised local transitions.

The framework also reframes several debates. It shows that finite Littlestone dimension is not the right universal boundary for sublinear end-to-end growth, that no single dimension can characterize all sublinear regimes, and that richer supervision or richer generator outputs do not automatically remove computational or statistical barriers. In particular, the NSP hardness results and the generator-access separations show that one must distinguish between observing more labels, reaching the right prefixes, and solving the induced inference problem.

In that sense, the PAC-learning framework for next-token generators is best understood as a theory of **rollout-induced hypothesis classes**. Its core insight is that the learnability of autoregressive systems depends jointly on three axes: the base class \(F\), the rollout length \(T\), and the supervision or access model under which intermediate computation is hidden, revealed, or directly queryable.

Source: https://www.emergentmind.com/topics/pac-learning-framework-for-next-token-generators