The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasets

TL;DR

本研究发现大型语言模型在规模足够大时,能通过线性空间结构表示事实的真伪,采用差异均值探针实现良好迁移和因果干预。

cs.AI 🔴 高级 2023-10-11 677 引用 62 次浏览
Samuel Marks Max Tegmark
自然语言处理 模型解释 线性表示 因果推断 模型规模

核心发现

方法论

本文采用高质量的真假陈述数据集,结合主成分分析(PCA)可视化、迁移实验和因果干预,系统研究大规模LLMs中的真伪表示结构。通过在不同数据集上训练线性探针,验证其迁移能力;利用残差流激活的“补丁”技术定位关键信息存储位置;最后通过操控隐藏状态方向,验证其因果关系。研究重点在于揭示模型内部是否存在抽象的“真”或“假”空间,以及其在不同模型规模中的演变。

关键结果

  • 在LLaMA-2-70B模型中,真/假陈述的表示在主成分空间中表现出明显的线性分离,且随着模型规模的增加,线性空间的抽象程度增强。通过迁移实验发现,训练在某一数据集上的线性探针可以较好地迁移到不同结构和主题的数据集,迁移准确率超过95%。
  • 通过因果干预实验,操控特定隐藏状态的方向可以显著改变模型对真假陈述的判断。例如,操控第15层残差激活的“真/假”方向,能使模型将虚假陈述误判为真实,反之亦然,验证了这些方向的因果重要性。
  • 简单的差异均值探针在不同模型规模和数据集上表现出与复杂探针相当的迁移能力,同时在因果干预中表现出更强的因果相关性。这表明线性空间中的“真”空间具有较强的因果解释力,优于其他复杂的探针方法。

研究意义

本研究突破了以往关于模型内部“真”或“假”表示的争议,提供了实验证据表明大规模LLMs在内部空间中以线性方式编码事实的真伪。这不仅深化了我们对模型内部知识表示的理解,也为模型的可信性和可解释性提供了理论基础。通过因果干预验证,揭示了模型决策的因果机制,有助于未来设计更透明、更可靠的AI系统。此外,提出的差异均值探针为模型解释提供了简洁而有效的工具,有望推广到更广泛的模型解释任务中。

技术贡献

本文提出了系统的多角度分析框架,包括高质量数据集、可视化、迁移实验和因果干预,验证了大规模LLMs中真/假表示的线性结构。引入差异均值探针,简化了传统线性探针的复杂性,同时增强了因果解释能力。研究还揭示了模型规模与抽象能力的关系,展示了模型在逐步抽象“真”概念的演变过程。这些技术贡献为模型内部知识的结构化理解提供了新工具和新视角。

新颖性

本研究首次系统性地利用高质量的真假陈述数据集,结合多种分析方法,证明了大规模LLMs在内部空间中以线性方式编码“真”或“假”。不同于以往仅关注单一探针或统计相关性的方法,本文通过迁移和因果干预验证了空间的因果关系,强调了线性空间的因果可操作性。这在模型解释和可信性研究中具有重要创新意义,填补了模型内部抽象表示的研究空白。

局限性

  • 研究主要集中在简单、明确的真假陈述,尚未覆盖复杂、多层次的推理任务,模型在处理含糊或多义性陈述时的表现仍不明确。
  • 因果干预实验依赖于操控特定隐藏状态的线性方向,可能忽略了非线性或分布式的表示机制,未来需探索更复杂的因果模型。
  • 模型规模虽大,但仍存在对特定类型陈述的偏差,未来应考虑多模态、多任务的验证,以增强普适性和鲁棒性。

未来方向

未来工作将扩展到更复杂的推理和多模态任务,验证线性空间的普适性与局限性。同时,结合更先进的因果推断技术,探索非线性或分布式的知识表示机制。还计划开发实时可解释工具,将模型的线性“真”空间可视化,辅助模型调试和信任评估。此外,研究如何利用这些空间结构优化模型训练和微调策略,以提升模型的真实性和可靠性。

AI 总览摘要

在人工智能快速发展的今天,理解大型语言模型(LLMs)内部的知识表示成为AI研究的核心难题之一。尽管这些模型在自然语言处理任务中表现出色,但其决策过程的透明度和可信度仍受到质疑。特别是在判断陈述真伪时,模型可能会输出虚假信息,影响其应用的安全性和可靠性。

本研究由Samuel Marks和Max Tegmark合作完成,系统探讨了大规模LLMs中关于“真”与“假”的线性空间结构。通过构建高质量的真假陈述数据集,结合主成分分析(PCA)可视化、迁移实验和因果干预技术,研究发现:随着模型规模的扩大,模型内部逐渐形成抽象的“真”空间,这一空间可以用线性方向准确区分真假陈述。

具体而言,研究中训练的线性探针在不同数据集间表现出极强的迁移能力,准确率超过95%,验证了“真”空间的普适性。同时,操控隐藏状态的线性方向可以显著改变模型的判断结果,证明这些空间方向具有因果关系。这一发现不仅为模型的解释提供了新工具,也为提升模型的可信性奠定了基础。

研究还提出了差异均值探针(差异-均值方向)的方法,简洁高效,优于复杂的线性探针,在因果干预中表现出更强的因果相关性。这些技术的结合,为理解和操控大规模模型的知识表示提供了全新视角。

总之,本研究揭示了大规模LLMs在内部空间中以线性方式编码事实的真伪,为模型的透明度、可信度和安全性提供了理论基础。未来,结合更复杂的推理任务和多模态数据,有望推动AI向更可信、更可解释的方向发展。

深度解读

原文摘要

Large Language Models (LLMs) have impressive capabilities, but are prone to outputting falsehoods. Recent work has developed techniques for inferring whether a LLM is telling the truth by training probes on the LLM's internal activations. However, this line of work is controversial, with some authors pointing out failures of these probes to generalize in basic ways, among other conceptual issues. In this work, we use high-quality datasets of simple true/false statements to study in detail the structure of LLM representations of truth, drawing on three lines of evidence: 1. Visualizations of LLM true/false statement representations, which reveal clear linear structure. 2. Transfer experiments in which probes trained on one dataset generalize to different datasets. 3. Causal evidence obtained by surgically intervening in a LLM's forward pass, causing it to treat false statements as true and vice versa. Overall, we present evidence that at sufficient scale, LLMs linearly represent the truth or falsehood of factual statements. We also show that simple difference-in-mean probes generalize as well as other probing techniques while identifying directions which are more causally implicated in model outputs.

cs.AI

参考文献 (20)

LLaMA: Open and Efficient Foundation Language Models

Hugo Touvron, Thibaut Lavril, Gautier Izacard 等

2023 21315 引用 ⭐ 高影响力 查看解读 →

Causal Analysis of Syntactic Agreement Mechanisms in Neural Language Models

Matthew Finlayson, Aaron Mueller, Stuart M. Shieber 等

2021 154 引用 ⭐ 高影响力 查看解读 →

Emergent World Representations: Exploring a Sequence Model Trained on a Synthetic Task

Kenneth Li, Aspen K. Hopkins, David Bau 等

2022 525 引用 ⭐ 高影响力 查看解读 →

Discovering Latent Knowledge in Language Models Without Supervision

Collin Burns, Haotian Ye, D. Klein 等

2022 829 引用 ⭐ 高影响力 查看解读 →

The Internal State of an LLM Knows When its Lying

A. Azaria, Tom M. Mitchell

2023 762 引用 ⭐ 高影响力 查看解读 →

Inference-Time Intervention: Eliciting Truthful Answers from a Language Model

Kenneth Li, Oam Patel, Fernanda B. Viégas 等

2023 1383 引用 ⭐ 高影响力 查看解读 →

Still no lie detector for language models: probing empirical and conceptual roadblocks

B. A. Levinstein, Daniel A. Herrmann

2023 109 引用 ⭐ 高影响力 查看解读 →

Explore, Establish, Exploit: Red Teaming Language Models from Scratch

Stephen Casper, Jason Lin, Joe Kwon 等

2023 141 引用 ⭐ 高影响力 查看解读 →

What Is One Grain of Sand in the Desert? Analyzing Individual Neurons in Deep NLP Models

Fahim Dalvi, Nadir Durrani, Hassan Sajjad 等

2018 226 引用 查看解读 →

What Is Data Science

Longbing Cao

2018 571 引用

Implicit Representations of Meaning in Neural Language Models

Belinda Z. Li, Maxwell Nye, Jacob Andreas

2021 246 引用 查看解读 →

Multimodal Neurons in Artificial Neural Networks

Gabriel Goh, Nick Cammarata, Chelsea Voss 等

2021 420 引用

Probing Classifiers: Promises, Shortcomings, and Advances

Yonatan Belinkov

2021 983 引用 查看解读 →

Understanding the role of individual units in a deep neural network

David Bau, Jun-Yan Zhu, Hendrik Strobelt 等

2020 542 引用 查看解读 →

Neural Natural Language Inference Models Partially Embed Theories of Lexical Entailment and Negation

Atticus Geiger, Kyle Richardson, Christopher Potts

2020 135 引用

Designing and Interpreting Probes with Control Tasks

John Hewitt, Percy Liang

2019 804 引用 查看解读 →

Investigating Gender Bias in Language Models Using Causal Mediation Analysis

Jesse Vig, Sebastian Gehrmann, Yonatan Belinkov 等

2020 776 引用

Mapping Language Models to Grounded Conceptual Spaces

Roma Patel, Ellie Pavlick

2022 211 引用

The Implicit Bias of Gradient Descent on Separable Data

Daniel Soudry, Elad Hoffer, Suriya Gunasekar 等

2017 1156 引用 查看解读 →

TruthfulQA: Measuring How Models Mimic Human Falsehoods

Stephanie C. Lin, Jacob Hilton, Owain Evans

2021 3797 引用 查看解读 →

被引用 (20)

Linear representations of grammaticality in neural language models

2026 ⭐ 高影响力 查看解读 →

Language Models Encode the Contextual Truth of Propositions

2026 ⭐ 高影响力 查看解读 →

On the Robustness of LLMs'Internal Representation of Code Correctness

2026 ⭐ 高影响力 查看解读 →

STEREODISCO: Discovering Stereotypicality in LLMs

2026 ⭐ 高影响力 查看解读 →

Efficient LLM Adversarial Training via Low-Rank Defense and Circuit-Guided Surrogates

The Refusal Residue: When Probes Catch Alignment Faking and When They Don't

Steering Robustness into World Action Models via Mechanistic Interpretability and Optimal Control

2026 1 引用 查看解读 →

How Does Alignment Tuning Shape Representations of Sycophancy and Related Cue-Induced Biases in LLMs?

The Anatomy of a Truth Direction: Knowledge-Dependent Dimensionality, a Relational Law, and a Convergent Category Geometry in Small Language Models

Operational Proto-Introspection in Looped Language Models: Process-Quality Taps, Executable Branching, and the Readout-Control Boundary

Diagnosing Correctness Probes under Self-Judgement Confounding

Peirce's revenge on the Chinese Room

2026

Misalignment Has a Personality: A Big Five Account of Emergent Misalignment

Isotropy Cliffs: The Geometric Signature of Decision-Making in Large Language Models

Cloud-ScPO: Hidden-State Geometry for Semi-Supervised Preference Optimization in LLM Reasoning

Probing the 3D Object-Level Understanding of Pre-Trained Detection Transformers

Semantic Alignment of AI Models: Concept Collapse, Checkpoint Dynamics, and Cross-Lingual Transfer

Cultural Awareness is Represented but Not Decoded: Tracing Mythological Knowledge across 18 Open-Source LLMs

When Outputs Disperse, Does Epistemic Revision Follow? A Black-Box Diagnostic for Machine Collectives

Test, then Route: How Language Models Execute In-Context Conditional Rules Across Models and Languages