arXiv:2609.02049cs.LGcs.CL2026-09

揭示大模型推理中连续混合状态崩溃的三大成因

The Dynamics of Continuous Mixture Collapse in Language Models

论文配图:The Dynamics of Continuous Mixture Collapse in Language Models
图 1 · 摘自论文原文
  • 用理论与实验分析混合状态在模型中的演化机制
  • 发现混合状态会因软最大读出和自回归反馈而失真或坍缩
  • 适合研究模型内部表征与推理稳定性的研究人员

大型语言模型的潜在状态推理方法用连续状态(如词嵌入加权混合)替代离散中间标记,以保留多种可能的推理方向。然而预训练语言模型往往无法保持这些混合状态。我们通过理论分析与受控实验,在多种模型上研究其原因,识别出三种独立且不同的失败来源:第一,变压器架构本身会扭曲混合几何结构,而训练过程进一步放大该效应;第二,即使模型能完美线性传输混合状态,软最大读出与自回归反馈仍构成一个动力系统,导致混合成分间微小差异被放大至一方主导,或使不同混合状态趋于不可区分;我们实证验证了这一预测:观测到的收缩与放大转变恰好位于理论阈值附近,且预训练模型的推演结果主要位于放大侧;最后,我们将结论推广至多成分混合情形,发现精确保持通常需依赖上下文的校正,其所需维度随成分数量增长。

原文摘要 · Abstract (English)

LLMs latent-state reasoning methods replace discrete intermediate tokens with continuous states, such as weighted mixtures of token embeddings, to retain multiple possible reasoning directions rather than committing to one. Yet pretrained language models often fail to preserve these mixtures. We study why through a combination of theoretical analysis and controlled empirical investigations on a variety of models. We identify three independent, distinct sources of failure. First, transformer architectures already distort mixture geometry, and training substantially amplifies this effect. Moreover, the failure can occur even if the model transports mixtures perfectly linearly: the softmax readout and autoregressive feedback form a dynamical system that either amplifies small differences until one component of the mixture dominates or contracts different mixtures until they become indistinguishable. We verify this theoretical prediction empirically: the observed transition between contraction and amplification occurs near the theoretical threshold derived by our analysis, and pretrained-model rollouts lie predominantly on the amplifying side. Finally, we generalize to mixtures of many components and show that exact preservation generally requires context-dependent correction, whose required dimensionality can grow with the number of components.

语言模型推理机制状态坍缩混合表示

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