Computational-Statistical Tradeoffs at the Next-Token Prediction Barrier: Autoregressive and Imitation Learning under Misspecification
On the map: Offline RL and imitation learning from data → Language-conditioned policies and VLAs
Abstract (arXiv)
Next-token prediction with the logarithmic loss is a cornerstone of autoregressive sequence modeling, but, in practice, suffers from error amplification, where errors in the model compound and generation quality degrades as sequence length H increases. From a theoretical perspective, this phenomenon should not appear in well-specified settings, and, indeed, a growing body of empirical work hypothesizes that misspecification, where the learner is not sufficiently expressive to represent the target distribution, may be the root cause. Under misspecification -- where the goal is to learn as well as the best-in-class model up to a multiplicative approximation factor C≥ 1 -- we confirm that C indeed grows with H for next-token prediction, lending theoretical support to this empirical hypothesis. We then ask whether this mode of error amplification is avoidable algorithmically, computationally, or information-theoretically, and uncover inherent computational-statistical tradeoffs. We show: (1) Information-theoretically, one can avoid error amplification and achieve C=O(1). (2) Next-token prediction can be made robust so as to achieve C=tilde O(H), representing moderate error amplification, but this is an inherent barrier: any next-token prediction-style objective must suffer C=Omega(H). (3) For the natural testbed of autoregressive linear models, no computationally efficient algorithm can achieve sub-polynomial approximation factor C=e^(log H)1-Omega(1); however, at least for binary token spaces, one can smoothly trade compute for statistical power and improve on C=Omega(H) in sub-exponential time. Our results have consequences in the more general setting of imitation learning, where the widely-used behavior cloning algorithm generalizes next-token prediction.
Key facts. Cited by 12 papers in the map (data through 7 Oct 2026). Placed in Language-conditioned policies and VLAs, one of 13 directions in Offline RL and imitation learning from data. Builds on 19 of its 85 references that are in the map. First posted on arXiv 18 Feb 2025.
Builds on
19 of 85 references.
References · 19
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2024Gemini 1.5: Unlocking multimodal understanding across millions of tokens of contextcited by 3,175
- 2019The Curious Case of Neural Text Degenerationcited by 2,881
- 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusioncited by 2,730
- 2023Learning Fine-Grained Bimanual Manipulation with Low-Cost Hardwarecited by 1,490
- 2018ChauffeurNet: Learning to Drive by Imitating the Best and Synthesizing the Worstcited by 391
- 2015Unifying distillation and privileged informationcited by 218
- 2021The Statistical Complexity of Interactive Decision Makingcited by 208
- 2024The pitfalls of next-token predictioncited by 156
- 2014Reinforcement and Imitation Learning via Interactive No-Regret Learningcited by 116
- 2023Robot Parkour Learningcited by 115
7 more
- 2021Feedback in Imitation Learning: The Three Regimes of Covariate Shiftcited by 68
- 2024DextrAH-G: Pixels-to-Action Dexterous Arm-Hand Grasping with Geometric Fabricscited by 36
- 2023Foundations of Reinforcement Learning and Interactive Decision Makingcited by 33
- 2019Lecture Notes: Selected topics on robust statistical learning theorycited by 17
- 2021Provably Breaking the Quadratic Error Compounding Barrier in Imitation Learning, Optimallycited by 12
- 2024Online Estimation via Offline Estimation: An Information-Theoretic Frameworkcited by 11
- 2024Exploration is Harder than Prediction: Cryptographically Separating Reinforcement Learning from Supervised Learningcited by 6
Built on by
Citing papers per year: 2025: 8; 2026: 4.
12 papers.
- 2025Dream-VL & Dream-VLA: Open Vision-Language and Vision-Language-Action Models with Diffusion Language Model Backbonecited by 18
- 2025On the Limits of Test-Time Compute: Sequential Reward Filtering for Better Inferencecited by 6
- 2025Inverse Q-Learning Done Right: Offline Imitation Learning in Qπ-Realizable MDPscited by 5
- 2025SAMPO:Scale-wise Autoregression with Motion PrOmpt for generative world modelscited by 4
- 2025To Distill or Decide? Understanding the Algorithmic Trade-off in Partially Observable Reinforcement Learningcited by 3
- 2025In Good GRACEs: Principled Teacher Selection for Knowledge Distillationcited by 3
- 2025EMA Without the Lag: Bias-Corrected Iterate Averaging Schemescited by 2
- 2026When Does Online Imitation Learning Help in LLM Post-Training? The Role of (Non-)Realizability Beyond Horizoncited by 2
- 2025Alternatives To Next Token Prediction In Text Generation -- A Surveycited by 1
- 2026Your Self-Play Algorithm is Secretly an Adversarial Imitator: Understanding LLM Self-Play through the Lens of Imitation Learningcited by 1
- 2026Multi-agent imitation learning with function approximation: Linear Markov games and beyondcited by 1
- 2026Fast Rates for Inverse Reinforcement Learningcited by 1
- 2026When Does Online Imitation Learning Help in LLM Post-Training? The Role of (Non-)Realizability Beyond Horizon29 Jun 2026
- 2026Fast Rates for Inverse Reinforcement Learning14 May 2026
- 2026Multi-agent imitation learning with function approximation: Linear Markov games and beyond26 Feb 2026
- 2026Your Self-Play Algorithm is Secretly an Adversarial Imitator: Understanding LLM Self-Play through the Lens of Imitation Learning1 Feb 2026
- 2025Dream-VL & Dream-VLA: Open Vision-Language and Vision-Language-Action Models with Diffusion Language Model Backbone27 Dec 2025
- 2025On the Limits of Test-Time Compute: Sequential Reward Filtering for Better Inference4 Dec 2025
- 2025In Good GRACEs: Principled Teacher Selection for Knowledge Distillation4 Nov 2025
- 2025To Distill or Decide? Understanding the Algorithmic Trade-off in Partially Observable Reinforcement Learning3 Oct 2025
- 2025Alternatives To Next Token Prediction In Text Generation -- A Survey29 Sep 2025
- 2025SAMPO:Scale-wise Autoregression with Motion PrOmpt for generative world models19 Sep 2025
- 2025EMA Without the Lag: Bias-Corrected Iterate Averaging Schemes31 Jul 2025
- 2025Inverse Q-Learning Done Right: Offline Imitation Learning in Qπ-Realizable MDPs26 May 2025
Similar, not cited
Close in wording, no citation either way
- 2024Where is the signal in tokenization space?
- 2026Sample Complexity of Autoregressive Reasoning: Chain-of-Thought vs. End-to-End
- 2023SequenceMatch: Imitation Learning for Autoregressive Sequence Modelling with Backtracking
- 2025Unifying Autoregressive and Diffusion-Based Sequence Generation
- 2024Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion
- 2026Autoregressive Learning in Joint KL: Sharp Oracle Bounds and Lower Bounds
- 2025How Reinforcement Learning After Next-Token Prediction Facilitates Learning
- 2026ALoDLM: Adaptively Looped Diffusion Language Models