arXiv:2511.01190cs.LGstat.ML2025-11NeurIPS

揭示思维链对大模型推理能力的局限性,发现其本质是记忆能力问题。

Analyzing the Power of Chain of Thought through Memorization Capabilities

  • 将推理视为记忆任务,分析带与不带思维链的Transformer的记忆能力边界。
  • 证明无论是否使用思维链,记忆一个含N个样本的数据集均需约Θ̄(N)参数。
  • 发现某些无限推理任务无法被任何版本的Transformer记忆,说明思维链并非万能。

已有研究表明,思维链(CoT)可提升大语言模型在特定数学推理任务中的表现。然而,其能力边界尚未完全明确。本文指出,变压器模型的推理本质上是针对推理数据集的记忆问题。因此,评估思维链在所有推理任务上的有效性,等价于分析其记忆能力。我们首次完整描述了固定精度的变压器模型在有无思维链情况下的记忆能力,并给出上述问题的否定回答:存在某些推理任务,思维链无法提升变压器的推理能力。具体而言,我们给出了有无思维链的模型记忆有限推理数据集的充要条件,二者互不蕴含;并推导出记忆包含$N$个元素的数据集所需的参数下界与上界均为$ar{Θ}(N)$。此外,我们首次研究了思维链模型对无限推理数据集的记忆能力,表明部分简单无限数据集无法被任何有或无思维链的变压器模型记忆。

原文摘要 · Abstract (English)

It has been shown that the chain of thought (CoT) can enhance the power of large language models (LLMs) to solve certain mathematical reasoning problems. However, the capacity of CoT is still not fully explored. As an important instance, the following basic question has not yet been answered: Does CoT expand the capability of transformers across all reasoning tasks? We demonstrate that reasoning with transformers is essentially a memorization problem for reasoning datasets. Thus, examining the power of CoT across all reasoning tasks amounts to analyzing the memorization capabilities of CoT transformers. In this paper, we give a complete description of the memorization capabilities of fixed-precision transformers with or without CoT and give a negative answer to the above-mentioned question. Precisely, we first give necessary and sufficient conditions for fixed-precision transformers with and without CoT to memorize a finite reasoning dataset and show that these two conditions do not imply each other. Then, we give lower and upper bounds for the number of parameters needed for transformers with or without CoT to memorize a finite reasoning dataset with $N$ elements, which are $\overlineΘ(N)$ in all cases. This implies that there exist reasoning tasks for which CoT does not enhance the reasoning power of transformers, leading to a negative answer to the above-mentioned question. Finally, we give the first results on memorizing infinite reasoning datasets by CoT transformers and show that some simple infinite datasets cannot be memorized by transformers with or without CoT.

思维链推理能力记忆机制模型容量

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