揭示自回归生成中误差累积的理论根源,提出计算与统计权衡的新视角。
Computational-Statistical Tradeoffs at the Next-Token Prediction Barrier: Autoregressive and Imitation Learning under Misspecification
- 在模型不充分时,预测误差随序列长度线性增长,无法避免
- 理论上可实现恒定误差放大因子,但实际算法受限于计算复杂度
- 为行为克隆等模仿学习方法提供理论支持,适合研究模型偏差的学者
以对数损失进行下一步词预测是自回归序列建模的核心,但在实践中存在误差放大现象:随着序列长度H增加,模型错误累积导致生成质量下降。理论上,在模型充分的情况下不应出现此现象;而越来越多的实证工作认为,模型表达力不足(即模型不充分)可能是根本原因。在模型不充分的设定下(目标是达到最优模型乘性近似因子C≥1),我们证实C确实随H增长,为这一假设提供了理论支持。进一步探讨该误差放大是否可通过算法、计算或信息论手段规避,发现存在内在的计算-统计权衡:(1) 信息论上可避免误差放大,实现C=O(1);(2) 下一步预测可被设计为仅承受温和的误差放大,达到C=~O(H),但这是一种固有瓶颈——任何下一步预测类目标函数都必然面临C=Ω(H);(3) 对于自回归线性模型,在二元词空间下,虽无高效算法能实现亚多项式近似因子C=e^{(log H)^{1−Ω(1)}},但可在次指数时间内通过牺牲计算资源换取统计性能提升,优于C=Ω(H)。这些结果对更广泛的模仿学习场景具有影响,其中广泛使用的行为克隆本质上是下一步预测的特例。
原文摘要 · Abstract (English)
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\geq 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=Ω(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-Ω(1)}}$; however, at least for binary token spaces, one can smoothly trade compute for statistical power and improve on $C=Ω(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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。