arXiv:2510.25791cs.LGcs.AI2025-10

揭示思维链如何影响Transformer学习速度与推理机制

The Kinetics of Reasoning: How Chain-of-Thought Shapes Learning in Transformers?

  • 通过可控复杂度符号任务研究思维链学习动态
  • 思维链加速泛化但无法突破高复杂度任务瓶颈
  • 发现推理过程存在早期不一致的动态演变现象

思维链(CoT)监督可显著提升Transformer性能,但其学习机制仍不明确。我们通过预训练Transformer于可调算法复杂度的符号推理任务,研究其泛化能力。模型在两种设置下训练:(i) 仅输出最终答案;(ii) 在回答前生成显式思维链。结果表明,尽管思维链通常提升性能,其收益依赖任务复杂度。我们用三参数逻辑曲线拟合准确率随对数训练步数的变化,揭示学习速度与形状随任务复杂度、数据分布和是否启用CoT而变化。还发现一个瞬态的推理不忠实阶段:训练初期模型常给出正确答案却跳过或矛盾地生成思维链步骤,后期才使推理与答案对齐。实证发现:(1) 思维链加速泛化但无法克服更高算法复杂度的任务(如列表交集);(2) 提出用于理解Transformer学习的动力学建模框架;(3) 揭示推理忠实性是随训练演化的动态属性;(4) 证明思维链改变了Transformer内部计算机制。

原文摘要 · Abstract (English)

Chain-of-thought (CoT) supervision can substantially improve transformer performance, yet the mechanisms by which models learn to follow and benefit from CoT remain poorly understood. We investigate these learning dynamics through the lens of grokking by pretraining transformers on symbolic reasoning tasks with tunable algorithmic complexity and controllable data composition to study their generalization. Models were trained under two settings: (i) producing only final answers, and (ii) emitting explicit CoT traces before answering. Our results show that while CoT generally improves task performance, its benefits depend on task complexity. To quantify these effects, we model the accuracy of the logarithmic training steps with a three-parameter logistic curve, revealing how the learning speed and shape vary with task complexity, data distribution, and the presence of CoT supervision. We also uncover a transient trace unfaithfulness phase: early in training, models often produce correct answers while skipping or contradicting CoT steps, before later aligning their reasoning traces with answers. Empirically, we (1) demonstrate that CoT accelerates generalization but does not overcome tasks with higher algorithmic complexity, such as finding list intersections; (2) introduce a kinetic modeling framework for understanding transformer learning; (3) characterize trace faithfulness as a dynamic property that emerges over training; and (4) show CoT alters internal transformer computation mechanistically.

思维链Transformer学习动力学推理

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