Autoregressive Learning in Joint KL: Sharp Oracle Bounds and Lower Bounds
Not on the map. Why?
Abstract (arXiv)
We study the fundamental and timely problem of learning long sequences in autoregressive modeling and next-token prediction under model misspecification, measured by the joint Kullback--Leibler (KL) divergence. Our goal is to characterize how the sequence horizon (H) affects both approximation and estimation errors in this joint-distribution, sequence-level regime. By establishing matching upper and lower bounds, we provide, to our knowledge, the first complete characterization of long-horizon error behavior under the natural joint KL objective, with improved rates and optimality justification relative to existing work. On the approximation side, we show that joint KL admits a horizon-free approximation factor, in sharp contrast to Hellinger-based analyses that exhibit an (Omega(H)) dependence for computationally efficient methods; this isolates the choice of divergence as the source of approximation amplification. On the estimation side, we prove a fundamental information-theoretic lower bound of order (Omega(H)) that holds for both decomposable policy classes and fully shared policies, matching the (widetilde O(H)) upper bounds achieved by computationally efficient algorithms. Our analysis clarifies the landscape of recent autoregressive learning results by aligning the log-loss training objective, the sequence-level evaluation metric, and the approximation metric colorblackthrough a sharp joint-KL oracle theory. We further show that these joint-KL guarantees imply policy learning regret bounds at rates matching prior imitation learning literature.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 9 of its 21 references that are in the map. First posted on arXiv 12 May 2026.
Builds on
9 of 21 references.
References · 9
- 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 938
- 2021The Statistical Complexity of Interactive Decision Makingcited by 218
- 2023The Learnability of In-Context Learningcited by 127
- 2023What and How does In-Context Learning Learn? Bayesian Model Averaging, Parameterization, and Generalizationcited by 118
- 2023A Latent Space Theory for Emergent Abilities in Large Language Modelscited by 43
- 2024Online Estimation via Offline Estimation: An Information-Theoretic Frameworkcited by 13
- 2025Computational-Statistical Tradeoffs at the Next-Token Prediction Barrier: Autoregressive and Imitation Learning under Misspecificationcited by 13
- 2024Explicitly Encoding Structural Symmetry is Key to Length Generalization in Arithmetic Taskscited by 11
- 2025Research Program: Theory of Learning in Dynamical Systemscited by 5
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026Stochastic Autoregressive Learningcited by 1
- 2026Stochastic Autoregressive Learning7 Aug 2026
Similar, not cited
Close in wording, no citation either way
- 2026Fast Rates for Offline Q-Learning with KL Regularization
- 2025Autoregressive Language Models are Secretly Energy-Based Models: Insights into the Lookahead Capabilities of Next-Token Prediction
- 2024Theoretical guarantees on the best-of-n alignment policy
- 2025Achieving Logarithmic Regret in KL-Regularized Zero-Sum Markov Games
- 2025No-Regret Thompson Sampling for Finite-Horizon Markov Decision Processes with Gaussian Processes
- 2024Information Theoretic Guarantees For Policy Alignment In Large Language Models
- 2025Best-of-N through the Smoothing Lens: KL Divergence and Regret Analysis
- 2026Divergence Timing and Cumulative Disagreement under KV-Cache Eviction