让大模型在隐藏层中进行逻辑推理,提升可解释性与可控性
ActivationReasoning: Logical Reasoning in Latent Activation Spaces
- 将稀疏自编码器提取的潜在概念组织成逻辑命题库
- 在多跳问答等任务中实现可扩展的结构化推理,准确率提升显著
- 适合研究模型透明性、可控生成与安全对齐的研究者
大语言模型虽能生成流畅文本,但其内部推理过程仍不透明且难以控制。稀疏自编码器(SAEs)可揭示与人类概念对齐的潜在特征,但这些特征脆弱且被动,缺乏系统性推理机制。为此,我们提出激活推理(ActivationReasoning, AR)框架,将显式逻辑推理嵌入大模型的潜在空间。该框架分三阶段:(1) 找到潜在表示,通过SAEs识别并构建概念字典;(2) 激活命题,在推理时检测激活概念并映射为逻辑命题;(3) 逻辑推理,基于命题应用逻辑规则,推导高层结构、组合新概念并引导模型行为。我们在多跳推理(PrOntoQA)、抽象与间接线索鲁棒性(Rail2Country)、自然语言多样性推理(ProverQA)及上下文敏感安全性(BeaverTails)任务上评估。结果表明,AR在推理复杂度上升时仍具鲁棒性,泛化至抽象与情境敏感任务,并跨模型主干迁移。这证明在潜在激活中嵌入逻辑结构不仅增强透明性,还能实现结构化推理、可靠控制与目标行为对齐,为更可信、可审计的人工智能提供路径。
原文摘要 · Abstract (English)
Large language models (LLMs) excel at generating fluent text, but their internal reasoning remains opaque and difficult to control. Sparse autoencoders (SAEs) make hidden activations more interpretable by exposing latent features that often align with human concepts. Yet, these features are fragile and passive, offering no mechanism for systematic reasoning or model control. To address this, we introduce ActivationReasoning (AR), a framework that embeds explicit logical reasoning into the latent space of LLMs. It proceeds in three stages: (1) Finding latent representations, first latent concept representations are identified (e.g., via SAEs) and organized into a dictionary; (2) Activating propositions, at inference time AR detects activating concepts and maps them to logical propositions; and (3)Logical reasoning, applying logical rules over these propositions to infer higher-order structures, compose new concepts, and steer model behavior. We evaluate AR on multi-hop reasoning (PrOntoQA), abstraction and robustness to indirect concept cues (Rail2Country), reasoning over natural and diverse language (ProverQA), and context-sensitive safety (BeaverTails). Across all tasks, AR scales robustly with reasoning complexity, generalizes to abstract and context-sensitive tasks, and transfers across model backbones. These results demonstrate that grounding logical structure in latent activations not only improves transparency but also enables structured reasoning, reliable control, and alignment with desired behaviors, providing a path toward more reliable and auditable AI.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。