Listening to the Wise Few: Query-Key Alignment Unlocks Latent Correct Answers in Large Language Models
Not on the map. Why?
Abstract (arXiv)
Large language models (LLMs) routinely fail to output the correct option in multiple-choice question answering (MCQA) while encoding the answer internally. We expose this latent knowledge via the Query--Key (QK) score, defined for an attention head as the inner product between the last-token query and the key at the end-of-line token following option i, evaluated before rotary positional embedding is applied. Its argmax identifies a universal class of select-and-copy heads in middle layers that perform option selection through semantic query--key alignment, mechanistically distinct from induction and copy-suppression heads (Olsson et al., 2022): they are invariant to label symbols, and solve a synthetic task with zero surface overlap---properties no positional-copy account explains and that critically require stripping RoPE. Across 24 models from 1.5B to 72B parameters (LLaMA-2/3/3.1/3.3, Qwen-2.5, Gemma, Phi-3.5, DeepSeek-R1-Distill), a single head's QK-score exceeds the model's own zero-shot accuracy by up to +27.4 pp on HellaSwag and +49.8 pp on HaluDialogue; causal zero-ablation collapses MCQA accuracy to near-random. To remove any dependence on labeled validation data, we introduce an unsupervised HeadScore that ranks heads from unlabeled inputs and recovers the supervised top-k heads on every tested model. Against four positional-debiasing baselines (e.g., PriDe, Wiegrefe, Wang), QK-score is complementary by construction: debiasing re-weights output logits, whereas QK-score reads the model's selection from a middle-layer head before decoding. We release a one-line drop-in HeadScore script and per-model head indices, making every result one-command reproducible across all 24 models and four benchmarks.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 32 of its 40 references that are in the map. First posted on arXiv 3 Oct 2024.
Builds on
32 of 40 references.
References · 32
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2022Language Models (Mostly) Know What They Knowcited by 1,856
- 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,302
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,165
- 2022In-context Learning and Induction Headscited by 964
- 2022Discovering Latent Knowledge in Language Models Without Supervisioncited by 767
- 2022Transformers learn in-context by gradient descentcited by 732
- 2023The Internal State of an LLM Knows When It's Lyingcited by 659
- 2022MedMCQA : A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answeringcited by 543
- 2023Large Language Models Are Not Robust Multiple Choice Selectorscited by 486
20 more
- 2018What you can cram into a single vector: Probing sentence embeddings for linguistic propertiescited by 457
- 2019Cosmos QA: Machine Reading Comprehension with Contextual Commonsense Reasoningcited by 285
- 2023Large Language Models Sensitivity to The Order of Options in Multiple-Choice Questionscited by 263
- 2022Leveraging Large Language Models for Multiple Choice Question Answeringcited by 203
- 2023Does Circuit Analysis Interpretability Scale? Evidence from Multiple Choice Capabilities in Chinchillacited by 149
- 2024When Benchmarks are Targets: Revealing the Sensitivity of Large Language Model Leaderboardscited by 137
- 2024Benchmarking LLMs via Uncertainty Quantificationcited by 126
- 2024A Practical Review of Mechanistic Interpretability for Transformer-Based Language Modelscited by 118
- 2023Circuit Component Reuse Across Tasks in Transformer Language Modelscited by 108
- 2023A Mechanistic Interpretation of Arithmetic Reasoning in Language Models using Causal Mediation Analysiscited by 89
- 2023Overthinking the Truth: Understanding how Language Models Process False Demonstrationscited by 81
- 2024"My Answer is C": First-Token Probabilities Do Not Match Text Answers in Instruction-Tuned Language Modelscited by 69
- 2023Do Androids Know They're Only Dreaming of Electric Sheep?cited by 67
- 2024Attention Heads of Large Language Models: A Surveycited by 67
- 2023Cognitive Dissonance: Why Do Language Model Outputs Disagree with Internal Representations of Truthfulness?cited by 62
- 2024Changing Answer Order Can Decrease MMLU Accuracycited by 42
- 2024A Study on Large Language Models' Limitations in Multiple-Choice Question Answeringcited by 22
- 2021The heads hypothesis: A unifying statistical approach towards understanding multi-headed attention in BERTcited by 15
- 2024Correcting Negative Bias in Large Language Models through Negative Attention Score Alignmentcited by 12
- 2023Increasing Probability Mass on Answer Choices Does Not Always Improve Accuracycited by 11
Built on by
Citing papers per year: 2025: 1.
1 paper.
- 2025A Survey of Context Engineering for Large Language Modelscited by 126
- 2025A Survey of Context Engineering for Large Language Models17 Jul 2025