← 返回 2026-01-12

思维的分子结构:长链推理的拓扑映射 The Molecular Structure of Thought: Mapping the Topology of Long Chain-of-Thought Reasoning

Qiguang Chen, Yantao Du, Ziniu Li, Jinhao Liu, Songyao Duan, Jiarui Guo, Minghao Liu, Jiaheng Liu, Tong Yang, Ge Zhang, Libo Qin, Wanxiang Che, Wenhao Huang 📅 2026-01-09 👍 60 2026-07-13 08:35
大语言模型 强化学习 思维链 蒸馏 认知科学 长链推理

用分子结构类比Long CoT,揭示三种推理键的稳定分布决定学习成败

前置知识

Chain-of-Thought (CoT)

思维链是一种让大语言模型在回答问题前生成中间推理步骤的技术。通过显式展示推理过程,模型能够处理需要多步逻辑的问题,比如数学证明或多步计算。CoT的核心思想是将复杂问题分解为一系列小步骤,每一步都基于前一步的结论继续推理。长链推理(Long CoT)是CoT的扩展形式,推理步骤可达数十甚至上百步,包含更多的回溯、探索和反思。

本文研究的核心对象就是Long CoT的有效结构,理解传统CoT与Long CoT的区别是理解本文动机的关键

知识蒸馏 (Knowledge Distillation)

知识蒸馏是一种模型压缩技术,通过让小模型(学生)模仿大模型(教师)的输出行为来转移知识。在推理任务中,蒸馏通常指让学生模型学习教师模型生成的推理链(rationales),而不仅仅是最终答案。这包括步骤级别的模仿学习和推理行为的转移。蒸馏的效果高度依赖教师模型的质量和生成轨迹的结构特性。

本文发现只有从强推理模型蒸馏才能有效转移Long CoT结构,而从弱模型或人类标注数据蒸馏则失败,这是论文的核心发现之一

语义同分异构体 (Semantic Isomers)

借用化学中的同分异构体概念,本文定义的语义同分异构体是指解决相同任务、访问相似语义区域,但行为分布和转换模式不同的Long CoT轨迹。就像化学中分子式相同但结构不同的化合物一样,语义同分异构体的差异在于推理行为的组合方式和转换概率,而非表面的词汇选择。这种结构差异会显著影响模型的学习效果。

这是本文的核心概念创新,解释了为什么表面相似的推理链会产生截然不同的学习效果

行为转换图 (Behavior Transition Graph)

行为转换图是一种有向图表示,节点是推理行为类型(深推理、自我反思、自我探索、正常操作),边的权重表示从一种行为转换到另一种行为的概率。通过分析大量推理轨迹,可以估计出模型在推理过程中的行为转换分布矩阵P(b'|b)。这种图结构编码了推理的全局拓扑特性,而不仅仅是单个步骤的内容。

行为转换图是Mole-Syn方法的核心,用于从强模型转移推理结构到弱模型

注意力能量 (Attention Energy)

本文将Transformer的注意力权重重新参数化为能量形式。注意力权重αij遵循softmax分布,可以看作玻尔兹曼分布:αij = exp(-Eij) / Σ exp(-Eil),其中Eij = -qi·kj/√dk定义为注意力能量。较低的能量意味着较高的注意力权重,即更强的依赖关系。不同推理行为(深推理、反思、探索)表现出特征性的能量分布,深推理能量最高(最弱),反思次之,探索能量最低(最强)。

注意力能量分析为分子结构类比提供了物理化学层面的理论支撑

研究动机

大语言模型在学习长链推理(Long CoT)时面临一个关键困境:从人类标注的推理轨迹或弱指令模型的In-Context Learning演示中进行标准监督微调,无法可靠地让模型掌握Long CoT能力。具体而言,Du et al. (2025) 的研究表明人类无法通过模仿DeepSeek-R1来生成有效的Long CoT,本文的初步实验进一步发现,使用随机采样的Long CoT示例进行标准SFT或从人类/弱指令LLM进行蒸馏,模型要么在长轨迹上失去连贯性,要么无法将模式迁移到新任务。以LLaMA-3.1-8B-Instruct为例,在GSM8K上从基础的75.89%通过ICL蒸馏后性能下降,在AIME2024上从4.17%降到更低水平。相比之下,从强推理模型(如DeepSeek-R1、QwQ-32B、OpenAI-OSS-120B)蒸馏则能显著提升性能,OSS蒸馏在AIME2024上可达10.83%。这种巨大差异表明存在某种结构性因素决定了Long CoT学习的成败,而不仅仅是数据质量或数量的问题。

本文的目标是本文的核心目标是理解大语言模型如何学习和表示有效的长链推理结构。具体而言,作者希望回答三个关键问题:第一,有效Long CoT的结构特性是什么?第二,为什么只有从强推理模型蒸馏才能成功,而人类模仿和ICL蒸馏失败?第三,能否在不依赖强推理模型的情况下,从弱指令模型合成有效的Long CoT结构?通过回答这些问题,作者希望建立一个理论框架来解释Long CoT的学习机制,并提出实用的方法来降低获取Long CoT能力的成本。

与已有工作不同的是,本文的独特切入角度是将Long CoT类比为分子结构,而非传统的序列、树或图结构。先前的工作将推理建模为逻辑节点的序列或树结构,关注单个节点的质量和连接方式。但本文发现,有效Long CoT在全局视角下呈现出稳定的分子状拓扑结构,由三种'化学键'构成:深推理(类比共价键,形成主链)、自我反思(类比氢键,实现远程折叠)、自我探索(类比范德华力,桥接远距离簇)。这种分子视角的核心洞察是:Long CoT的有效性不取决于单个推理步骤的质量,而取决于这三种行为键的全局分布和相互作用。这解释了为什么表面相似但结构不同的推理链(语义同分异构体)会产生截然不同的学习效果。

核心方法

本文的方法论建立在一个核心直觉上:有效的Long CoT不是一个简单的线性推理链,而是一个具有稳定分子状拓扑结构的推理过程。就像蛋白质通过折叠形成稳定的三维结构一样,Long CoT通过三种推理行为的相互作用形成稳定的推理拓扑。作者首先通过大规模实验验证了这一假说:分析多个强推理模型(DeepSeek-R1、QwQ-32B、OpenAI-OSS-120B)在多种任务上的推理轨迹,发现它们的行为转换分布在不同模型和任务间具有高度一致性(Pearson相关系数>0.9)。基于这一发现,作者提出了Mole-Syn框架,其技术路线是:首先从强推理模型估计行为转换图(behavior transition graph),然后将这种行为结构转移到便宜的指令LLM上,通过控制轨迹合成来生成匹配目标行为分布的Long CoT数据,而不是直接复制教师输出。

本文的核心创新点是将Long CoT的学习问题从'复制推理内容'转变为'转移推理结构'。已有方法(如直接蒸馏)试图让学习模型模仿教师模型的具体推理步骤和词汇表达,但本文发现真正重要的是推理行为的全局分布模式。具体而言,三种推理行为形成特征性的'化学键':深推理(D)像共价键一样形成紧密的逻辑依赖链,自我反思(R)像氢键一样实现对先前步骤的检验和修正(81.72%的反思步骤会回连到之前的语义簇),自我探索(E)像范德华力一样在远距离概念间建立松散桥接。这种行为分布的稳定性体现在:不同模型(R1、OSS、QwQ)在不同任务上的行为转换图Pearson相关系数超过0.95。Mole-Syn正是利用这一特性,只转移行为结构而非具体内容,从而实现从指令LLM合成有效的Long CoT。

方法步骤详情

Mole-Syn方法包含三个核心步骤。第一步,行为标注与转换图估计:从20K条强推理模型(如QwQ-32B或OSS-120B)生成的Long CoT蒸馏数据中,使用Qwen2.5-32B-Instruct将每个推理步骤之间的转换标注为四种行为之一(正常操作N、深推理D、自我反思R、自我探索E),然后计算经验转换矩阵P̂(b'|b)和边缘分布π(b),形成行为转换图。第二步,结构引导的轨迹合成:以探索状态为起始,按照估计的转换概率随机游走,在每个状态下使用专门设计的提示词(包含行为定义和指令)引导指令LLM生成对应行为类型的推理步骤。例如,当采样到自我反思状态时,使用包含'请对响应进行自我反思'的提示词;采样到深推理状态时,使用'请进一步深化推理'的提示词。第三步,合成数据微调:使用合成的20K条Long CoT数据对学生模型(如Llama-3.1-8B-Instruct)进行一个epoch的监督微调,学习率2e-5,全局batch size 128,最大序列长度16K-32K tokens。

技术新颖性

Mole-Syn的技术新颖性体现在三个方面。首先,分子结构类比本身是全新的视角:与传统将CoT建模为序列、树或图不同,本文将推理行为的边缘(而非节点)作为分析单元,三种行为键的全局分布决定了推理结构的稳定性。这种从节点中心到边缘分布的范式转换,使得能够捕捉传统方法遗漏的全局拓扑特性。其次,行为分布的可转移性发现具有重要价值:不同强推理模型的行为转换图高度一致(r>0.95),表明存在一个任务无关的'最优推理拓扑',这使得结构转移成为可能。第三,Mole-Syn实现了推理结构与内容的解耦:传统蒸馏直接复制教师输出,而Mole-Syn只转移行为分布,让指令LLM在自己擅长的领域内生成符合目标结构的推理链。这种解耦不仅降低了对强推理模型的依赖,还避免了结构冲突问题——实验表明同时训练两个高度相关(r≈0.9)但结构不同的推理框架会导致'结构混乱'和性能下降。

稳定分子结构假说:Long CoT的三种化学键
Figure 1: 稳定分子结构假说:Long CoT的三种化学键
先前结构与分子结构推理的对比
Figure 2: 先前结构与分子结构推理的对比
SFT学习的特征分析
Figure 6: SFT学习的特征分析
逻辑折叠结构的验证
Figure 7: 逻辑折叠结构的验证
不同键的注意力能量水平
Figure 8: 不同键的注意力能量水平
人类和推理模型的信息流分析
Figure 10: 人类和推理模型的信息流分析
单个键在推理中的形成功能
Figure 13: 单个键在推理中的形成功能

实验结果

本文的核心实验发现可以归纳为六个方面。第一,只有从强推理模型蒸馏才能有效转移Long CoT结构:在LLaMA-3.1-8B-Base上,使用20K OSS蒸馏数据在6个基准上平均达到33.69%,而ICL蒸馏和人类标注蒸馏效果远差。第二,强推理模型的行为转换图在不同模型间高度稳定:DeepSeek-R1、OpenAI-OSS-120B、QwQ-32B的转换图Pearson相关系数>0.9(p<0.001),采样超过2000例时相关系数>0.95。第三,SFT学习的是推理结构而非表面关键词:通过稀疏自编码器分析发现,Long CoT行为集中在少数话语控制特征上;将关键词替换为同义词后,模型仍能达到相当的推理性能。第四,语义同分异构体的结构差异决定学习成败:R1和OSS蒸馏链的结构相关性可达0.95,但某些模型使用R1蒸馏后性能下降超过10%。第五,混合两个稳定的分子结构会导致结构混乱:同时训练R1和OSS蒸馏数据(r≈0.9)使自相关性不超过0.8,性能显著下降。第六,Mole-Syn合成的Long CoT结构接近直接蒸馏效果:在Llama-3.1-8B-Instruct上,QwQ-Mole-Syn在6个基准上平均32.29%,QwQ直接蒸馏为35.73%,且Mole-Syn初始化的模型在RL训练中表现更稳定。

六个基准上的蒸馏结果
Table 1: 六个基准上的蒸馏结果
Mole-Syn与直接蒸馏的性能对比
Table 2: Mole-Syn与直接蒸馏的性能对比
从Gemini和Claude的蒸馏结果
Table 3: 从Gemini和Claude的蒸馏结果
总结Long CoT轨迹的结果
Table 4: 总结Long CoT轨迹的结果
六个基准上的完整结果
Table 5: 六个基准上的完整结果
强推理LLM蒸馏与Mole-Syn的完整对比
Table 6: 强推理LLM蒸馏与Mole-Syn的完整对比
不同指令模型的Mole-Syn效果分析
Table 7: 不同指令模型的Mole-Syn效果分析
三种初始化方法的强化学习性能
Table 8: 三种初始化方法的强化学习性能
三个不同模型的行为转换图
Figure 5: 三个不同模型的行为转换图
不同ICL蒸馏设置下的性能
Figure 9: 不同ICL蒸馏设置下的性能
两个稳定分子结构之间的冲突学习
Figure 11: 两个稳定分子结构之间的冲突学习
Mole-Syn初始化模型的RL持续改进
Figure 12: Mole-Syn初始化模型的RL持续改进
ICL蒸馏和人类标注推理轨迹的完整失败结果
Figure 15: ICL蒸馏和人类标注推理轨迹的完整失败结果
不同推理键比例下的性能对比分析
Figure 16: 不同推理键比例下的性能对比分析
查看结构化数据
任务指标本文基线提升
GSM8K (小学数学) Accuracy (%) 84.31 (QwQ-Mole-Syn on Llama-8B-Instruct) 82.41 (QwQ-Distill), 75.89 (原始) +2.3% vs QwQ-Distill (Instruct+Mole-Syn), +11.6% vs 原始
MATH-500 (高中数学) Accuracy (%) 50.20 (QwQ-Mole-Syn on Llama-8B-Instruct) 60.80 (QwQ-Distill), 35.20 (原始) -17.4% vs QwQ-Distill, +42.6% vs 原始
AIME2024 (数学邀请赛) Avg@16 Accuracy (%) 5.21 (QwQ-Mole-Syn on Llama-8B-Instruct) 4.38 (QwQ-Distill), 4.17 (原始) +18.9% vs QwQ-Distill, +24.9% vs 原始
AIME2025 (数学邀请赛) Avg@16 Accuracy (%) 1.67 (QwQ-Mole-Syn on Llama-8B-Instruct) 8.33 (QwQ-Distill), 1.04 (原始) -80.0% vs QwQ-Distill, +60.6% vs 原始
AMC2023 (高中竞赛) Avg@16 Accuracy (%) 32.34 (QwQ-Mole-Syn on Llama-8B-Instruct) 32.97 (QwQ-Distill), 23.59 (原始) -1.9% vs QwQ-Distill, +37.1% vs 原始
OlympiadBench (奥赛级别) Accuracy (%) 20.00 (QwQ-Mole-Syn on Llama-8B-Instruct) 25.48 (QwQ-Distill), 12.00 (原始) -21.5% vs QwQ-Distill, +66.7% vs 原始
6个基准平均 Avg Accuracy (%) 32.29 (QwQ-Mole-Syn on Llama-8B-Instruct) 35.73 (QwQ-Distill), 25.32 (原始) -9.6% vs QwQ-Distill, +27.5% vs 原始

局限与改进

作者在论文末尾明确承认了四个主要局限性。第一,分析依赖有限的教师模型和学生骨干网络:受限于成本和规模,实验主要使用DeepSeek-R1、QwQ-32B、OpenAI-OSS-120B作为教师,Llama和Qwen系列作为学生,这可能引入特定架构或训练方法的统计偏差。第二,方法局限于离线蒸馏和监督微调:论文未探讨Mole-Syn在在线或交互式强化学习环境中的扩展性,实际应用中可能需要更多探索。第三,分子结构的可视化和量化仍不精确:虽然通过t-SNE和注意力能量分析提供了几何特征的近似可视化,但准确刻画通用Long CoT大分子结构仍是开放问题。第四,行为标注依赖自动标注器:尽管提供了初步的鲁棒性检查(macro-F1>0.85),但标注噪声或偏差不可避免地会影响估计的行为分布。此外,从我自己的分析来看,Mole-Syn在某些高难度基准(如AIME2025)上的表现显著弱于直接蒸馏,这表明合成结构在处理极端复杂的推理任务时可能丢失了一些关键的细粒度推理模式;同时,方法需要先从强模型获取蒸馏数据来估计转换图,这在某种程度上仍然依赖强推理模型的存在。

独立分析的弱点

基于独立分析,本文存在以下几个值得关注的弱点。首先,Mole-Syn在高难度任务上的性能差距明显:在AIME2025上,QwQ-Mole-Syn仅1.67%而QwQ-Distill为8.33%,下降80%,这表明合成的分子结构在需要极深推理的任务上可能不够精细,改进方向是引入更细粒度的行为分类(如将深推理进一步细分为演绎、归纳、类比等子类型)或增加结构复杂度的控制机制。其次,行为转换图的估计依赖固定的数据集:当前方法使用20K条蒸馏数据估计转换矩阵,但这可能不代表所有任务类型的最优分布,改进方向是开发自适应的转换图估计方法,根据输入问题的特征动态调整行为分布。第三,分子结构类比的理论基础仍有待加强:虽然论文证明了注意力能量的玻尔兹曼分布对应关系,但三种'化学键'的能量排序(D > R > E)缺乏独立的物理化学验证,改进方向是探索与统计力学或信息论的更深层联系。第四,Mole-Syn的合成质量高度依赖指令LLM的能力:Table 7显示Llama-3.1-8B-Instruct作为合成骨干时性能仅为13.95%(远低于Qwen-32B的35.73%),这意味着方法对基础模型的推理能力有较高要求。

未来方向

本文开启了多个有前景的研究方向。作者提出的方向包括:探索分子结构在在线强化学习环境中的应用,当前仅验证了离线SFT和初始化后RL的效果;研究更通用的Long CoT大分子结构表示,当前的可视化依赖于特定的嵌入空间和t-SNE投影;以及开发更精确的行为自动标注方法以减少标签噪声。基于本文成果可以延伸的方向包括:将分子结构分析扩展到代码生成、科学推理等非数学领域,验证行为分布的跨域稳定性;研究分子结构与模型规模、架构的关系,理解为什么更大的模型更善于形成长程推理折叠;开发基于分子结构的主动学习策略,根据当前推理轨迹的结构特征动态选择下一步行为;以及将Mole-Syn与模型压缩技术结合,探索在资源受限场景下合成最优推理结构的方法。此外,论文提到的隐私保护机制(总结压缩可破坏Long CoT结构)也值得深入研究,可能启发新的模型保护方法。

复现评估

本文的复现条件相对较好。数据方面,论文使用OpenThoughts-3公开数据集作为主要训练语料,随机采样20K-35K条高质量Long CoT查询。模型方面,所有实验模型均为公开可用:Llama-3.1-8B/70B-Instruct、Qwen-2.5-7B/32B-Instruct/Base、DeepSeek-R1-671B、QwQ-32B、OpenAI-OSS-120B。训练设置明确:1个epoch,学习率2e-5,全局batch size 128,最大序列长度16K-32K tokens,推理温度0.6。评估使用6个公开数学推理基准(GSM8K、MATH-500、AMC2023、AIME2024/2025、OlympiadBench),评估代码使用标准的answer-matching工具(math-verify)。行为标注使用Qwen2.5-32B-Instruct,标注prompt完整提供。Mole-Syn的合成提示词也完整公开。算力需求方面,20K条数据的蒸馏训练在8B模型上应可在单机多卡环境下完成,但RL训练(1000步,batch size 16)可能需要更多计算资源。总体而言,复现难度中等,主要挑战在于获取强推理模型的API访问权限和足够的计算资源进行RL训练。