arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.00770cs.CL
收藏

推理作为状态转换:大型语言模型推理演变的表征分析

Reasoning as State Transition: A Representational Analysis of Reasoning Evolution in Large Language Models

Dept. of Comp. Sci. and Tech., Institute for AI, Tsinghua-Bosch Joint ML Center, THBI Lab, BNRist Center, Tsinghua University, Beijing 100084, China(计算机科学与技术系,人工智能研究所,清华-博世联合机器学习中心,THBI实验室,BNRist中心,清华大学,北京100084,中国);College of AI, Tsinghua University, Beijing 100084, China(人工智能学院,清华大学,北京100084,中国)

Siyuan Zhang, Jialian Li, Yichi Zhang, Xiao Yang, Yinpeng Dong, Hang Su

AI总结:

本研究从表征视角分析大型语言模型推理过程的演变,揭示训练后模型通过优化表示分布提升推理能力的机制。

摘要:

大型语言模型在推理任务上取得了显著的性能,这促使研究者探讨这种能力在训练过程中的演变机制。以往的研究主要通过显式的生成结果来分析这种演变,将推理过程视为一个黑箱,掩盖了内部的变化。为了解决这种不透明性,我们引入了表征视角来研究模型内部状态的动力学。通过在不同训练阶段的模型上进行全面实验,我们发现训练后仅在静态初始表征质量上产生有限的提升。此外,我们揭示了与非推理任务不同,推理过程中在生成期间表示会经历显著的连续分布变化。比较分析表明,训练后使模型能够推动这种转变朝着更优的分布发展以完成任务。为了明确内部状态与外部输出之间的关系,统计分析证实生成正确性与最终表示之间存在高度相关性;而反事实实验则识别出生成的token的语义,而不是推理期间的额外计算或内在参数差异,是转变的主要驱动因素。总体而言,我们提供了对推理过程以及训练对推理增强影响的全新理解,为未来模型分析和优化提供了有价值的见解。

英文摘要:

Large Language Models have achieved remarkable performance on reasoning tasks, motivating research into how this ability evolves during training. Prior work has primarily analyzed this evolution via explicit generation outcomes, treating the reasoning process as a black box and obscuring internal changes. To address this opacity, we introduce a representational perspective to investigate the dynamics of the model's internal states. Through comprehensive experiments across models at various training stages, we discover that post-training yields only limited improvement in static initial representation quality. Furthermore, we reveal that, distinct from non-reasoning tasks, reasoning involves a significant continuous distributional shift in representations during generation. Comparative analysis indicates that post-training empowers models to drive this transition toward a better distribution for task solving. To clarify the relationship between internal states and external outputs, statistical analysis confirms a high correlation between generation correctness and the final representations; while counterfactual experiments identify the semantics of the generated tokens, rather than additional computation during inference or intrinsic parameter differences, as the dominant driver of the transition. Collectively, we offer a novel understanding of the reasoning process and the effect of training on reasoning enhancement, providing valuable insights for future model analysis and optimization.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑