提出新型稀疏自编码器,精准解析大模型推理过程的每一步
Step-Level Sparse Autoencoder for Reasoning Process Interpretation
- 在步骤层级构建稀疏自编码器,通过上下文控制稀疏性分离关键信息
- 提取特征可预测推理长度、首词分布及正确性等复杂属性
- 适合研究模型可解释性与自我验证机制的研究者
大型语言模型(LLM)通过思维链(CoT)推理展现出强大的复杂推理能力,但其推理过程仍过于复杂难以分析。尽管稀疏自编码器(SAE)已成为可解释性的重要工具,现有方法多在标记层级操作,与更关键的步骤级信息(如推理方向、语义转换)存在粒度不匹配。本文提出步骤级稀疏自编码器(SSAE),作为解析LLM推理步骤中不同方面的分析工具。通过精确控制基于上下文的步骤特征稀疏性,形成步骤重构的信息瓶颈,将增量信息与背景信息分离,并分解为多个稀疏激活维度。在多个基础模型和推理任务上的实验表明,所提取特征有效。线性探测显示,这些特征可轻松预测生成长度、首词分布等表面信息,以及步骤正确性和逻辑性等复杂属性。这表明模型在生成过程中已部分掌握这些性质,为模型的自我验证能力提供了基础。代码开源:https://github.com/Miaow-Lab/SSAE。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have achieved strong complex reasoning capabilities through Chain-of-Thought (CoT) reasoning. However, their reasoning patterns remain too complicated to analyze. While Sparse Autoencoders (SAEs) have emerged as a powerful tool for interpretability, existing approaches predominantly operate at the token level, creating a granularity mismatch when capturing more critical step-level information, such as reasoning direction and semantic transitions. In this work, we propose step-level sparse autoencoder (SSAE), which serves as an analytical tool to disentangle different aspects of LLMs' reasoning steps into sparse features. Specifically, by precisely controlling the sparsity of a step feature conditioned on its context, we form an information bottleneck in step reconstruction, which splits incremental information from background information and disentangles it into several sparsely activated dimensions. Experiments on multiple base models and reasoning tasks show the effectiveness of the extracted features. By linear probing, we can easily predict surface-level information, such as generation length and first token distribution, as well as more complicated properties, such as the correctness and logicality of the step. These observations indicate that LLMs should already at least partly know about these properties during generation, which provides the foundation for the self-verification ability of LLMs. Our code is available at https://github.com/Miaow-Lab/SSAE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。