揭示大模型如何通过训练自然形成多推理路径并存的超叠加机制。
Emergence of Superposition: Unveiling the Training Dynamics of Chain of Continuous Thought
- 分两阶段分析训练动态:生成思想与预测答案
- 索引匹配逻辑值先升后稳,平衡探索与利用
- 为连续思维推理提供理论依据,适合研究模型机制者
已有研究表明,连续思维链(continuous CoT)通过隐式并行思考提升大语言模型的推理能力。后续工作从理论上证明,配备连续思维的两层Transformer可高效解决有向图可达性问题,通过在连续思维中维持多个推理路径的超叠加状态。然而,这种超叠加机制如何通过梯度训练自然习得仍不明确。本文针对简化版两层Transformer在有向图可达性任务上的训练动态进行理论分析,揭示了超叠加机制在两个训练阶段的涌现过程:(i) 思考生成阶段,模型自回归扩展连续思维;(ii) 预测阶段,将思维转化为最终答案。分析表明,在连续思维训练下,索引匹配逻辑值(index-matching logit)在温和假设下会先上升后趋于稳定。该稳定值有效平衡了推理过程中的探索与利用:模型利用局部问题结构识别合理搜索路径,并对多个可能路径赋予相近权重,以在不确定时进行探索,从而形成超叠加。实验追踪逻辑值增长结果进一步验证了该理论。
原文摘要 · Abstract (English)
Previous work shows that the chain of continuous thought (continuous CoT) improves the reasoning capability of large language models (LLMs) by enabling implicit parallel thinking, and a subsequent work provided theoretical insight by showing that a two-layer transformer equipped with continuous CoT can efficiently solve directed graph reachability by maintaining a superposition of multiple reasoning traces in the continuous thought. However, it remains unclear how the superposition mechanism is naturally learned from gradient-based training methods. To fill this gap, we theoretically analyze the training dynamics of a simplified two-layer transformer on the directed graph reachability problem to unveil how the superposition mechanism emerges during training in two training stages -- (i) a thought-generation stage that autoregressively expands the continuous thought, and (ii) a prediction stage that converts the thought into the final answer. Our analysis reveals that during training using continuous thought, the index-matching logit, an important quantity which reflects the strength of the model's local search ability, will first increase and then remain bounded under mild assumptions. The bounded index-matching logit effectively balances exploration and exploitation during the reasoning process: the model will exploit local problem structures to identify plausible search traces, and assign comparable weights to multiple such traces to explore when it is uncertain about which solution is correct, which results in superposition. Our experimental results tracking the growth of logits further validate our theory.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。