arXiv:2601.00923cs.AI2026-01被引 1

揭示大模型上下文学习与模型坍缩的内在关联机制。

Context Collapse: In-Context Learning and Model Collapse

  • 用线性Transformer建模上下文学习,发现参数相变现象。
  • 超过临界上下文长度后,解出现反对称分量导致旋转偏差。
  • 提出上下文坍缩概念,解释长文本生成中的稳定性问题。

本论文研究大语言模型中的两个关键现象:上下文学习(ICL)与模型坍缩。通过在权重共享的线性Transformer上训练线性回归任务,发现最小化上下文损失会导致参数的相变:当上下文长度超过临界值时,解中产生反对称分量。我们通过将权重共享下的前向传播简化为预条件梯度下降,并分析最优预条件器,证明该预条件器包含反对称项,从而引发梯度方向旋转。对于模型坍缩,我们运用鞅理论和随机游走理论,在线性回归与高斯拟合的简化设置下,分别在替换与累积数据模式下进行分析,强化了已有结论:除非数据增长足够快或被长期保留,否则几乎必然发生坍缩。最后,我们引入‘上下文坍缩’概念——即长生成过程中上下文质量退化,尤其在链式思维推理中表现显著,这一概念将ICL动态与生成模型的长期稳定性挑战联系起来。

原文摘要 · Abstract (English)

This thesis investigates two key phenomena in large language models (LLMs): in-context learning (ICL) and model collapse. We study ICL in a linear transformer with tied weights trained on linear regression tasks, and show that minimising the in-context loss leads to a phase transition in the learned parameters. Above a critical context length, the solution develops a skew-symmetric component. We prove this by reducing the forward pass of the linear transformer under weight tying to preconditioned gradient descent, and then analysing the optimal preconditioner. This preconditioner includes a skew-symmetric component, which induces a rotation of the gradient direction. For model collapse, we use martingale and random walk theory to analyse simplified settings - linear regression and Gaussian fitting - under both replacing and cumulative data regimes. We strengthen existing results by proving almost sure convergence, showing that collapse occurs unless the data grows sufficiently fast or is retained over time. Finally, we introduce the notion of context collapse: a degradation of context during long generations, especially in chain-of-thought reasoning. This concept links the dynamics of ICL with long-term stability challenges in generative models.

上下文学习模型坍缩大模型机制

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