Probing Language Models on Their Knowledge Source
Not on the map. Why?
Abstract (arXiv)
Large Language Models (LLMs) often encounter conflicts between their learned, internal (parametric knowledge, PK) and external knowledge provided during inference (contextual knowledge, CK). Understanding how LLMs models prioritize one knowledge source over the other remains a challenge. In this paper, we propose a novel probing framework to explore the mechanisms governing the selection between PK and CK in LLMs. Using controlled prompts designed to contradict the model's PK, we demonstrate that specific model activations are indicative of the knowledge source employed. We evaluate this framework on various LLMs of different sizes and demonstrate that mid-layer activations, particularly those related to relations in the input, are crucial in predicting knowledge source selection, paving the way for more reliable models capable of handling knowledge conflicts effectively.
Key facts. Cited by 11 papers in the map (data through 14 Oct 2026). Builds on 19 of its 31 references that are in the map. First posted on arXiv 8 Oct 2024.
Builds on
19 of 31 references.
References · 19
- 2017Attention Is All You Needcited by 60,361
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
- 2022Locating and Editing Factual Associations in GPTcited by 2,691
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,411
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
- 2019Can You Trust Your Model's Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shiftcited by 1,287
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,165
- 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 623
- 2023Textbooks Are All You Need II: phi-1.5 technical reportcited by 565
- 2023Dissecting Recall of Factual Associations in Auto-Regressive Language Modelscited by 528
7 more
- 2021Measuring and Improving Consistency in Pretrained Language Modelscited by 364
- 2023Adaptive Chameleon or Stubborn Sloth: Revealing the Behavior of Large Language Models in Knowledge Conflictscited by 341
- 2023Do Large Language Models Know What They Don't Know?cited by 243
- 2022A Review on Language Models as Knowledge Basescited by 157
- 2024Infini-gram: Scaling Unbounded n-gram Language Models to a Trillion Tokenscited by 153
- 2020Language Models as Knowledge Bases: On Entity Representations, Storage Capacity, and Paraphrased Queriescited by 107
- 2023A Glitch in the Matrix? Locating and Detecting Language Model Grounding with Fakepediacited by 24
Built on by
Citing papers per year: 2025: 9; 2026: 2.
11 papers.
- 2025How does Chain of Thought Think? Mechanistic Interpretability of Chain-of-Thought Reasoning with Sparse Autoencodingcited by 19
- 2025Language Models Can Predict Their Own Behaviorcited by 8
- 2025Probing the Difficulty Perception Mechanism of Large Language Modelscited by 8
- 2025RECALL: REpresentation-aligned Catastrophic-forgetting ALLeviation via Hierarchical Model Mergingcited by 5
- 2025Bi-directional Model Cascading with Proxy Confidencecited by 2
- 2025Context Copying Modulation: The Role of Entropy Neurons in Managing Parametric and Contextual Knowledge Conflictscited by 1
- 2026Probing for Knowledge Attribution in Large Language Modelscited by 1
- 2026Do LLMs have core beliefs?cited by 1
- 2025Does This Look Familiar to You? Knowledge Analysis via Model Internal Representationscited by 0
- 2025That's Deprecated! Understanding, Detecting, and Steering Knowledge Conflicts in Language Models for Code Generationcited by 0
- 2025Classification EM-PCA for clustering and embeddingcited by 0
- 2026Do LLMs have core beliefs?5 May 2026
- 2026Probing for Knowledge Attribution in Large Language Models26 Feb 2026
- 2025Classification EM-PCA for clustering and embedding24 Nov 2025
- 2025RECALL: REpresentation-aligned Catastrophic-forgetting ALLeviation via Hierarchical Model Merging23 Oct 2025
- 2025That's Deprecated! Understanding, Detecting, and Steering Knowledge Conflicts in Language Models for Code Generation21 Oct 2025
- 2025Probing the Difficulty Perception Mechanism of Large Language Models7 Oct 2025
- 2025Context Copying Modulation: The Role of Entropy Neurons in Managing Parametric and Contextual Knowledge Conflicts12 Sep 2025
- 2025Does This Look Familiar to You? Knowledge Analysis via Model Internal Representations9 Sep 2025
- 2025How does Chain of Thought Think? Mechanistic Interpretability of Chain-of-Thought Reasoning with Sparse Autoencoding24 Jul 2025
- 2025Bi-directional Model Cascading with Proxy Confidence27 Apr 2025
- 2025Language Models Can Predict Their Own Behavior18 Feb 2025