mlsanityData to 14 Oct 2026

Listening to the Wise Few: Query-Key Alignment Unlocks Latent Correct Answers in Large Language Models

arXiv 2410.02343 ↗First posted 3 Oct 2024cs.CL cs.LGcited by 1

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

  1. 2020Measuring Massive Multitask Language Understandingcited by 7,963
  2. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  3. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
  4. 2022Language Models (Mostly) Know What They Knowcited by 1,856
  5. 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,302
  6. 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,165
  7. 2022In-context Learning and Induction Headscited by 964
  8. 2022Discovering Latent Knowledge in Language Models Without Supervisioncited by 767
  9. 2022Transformers learn in-context by gradient descentcited by 732
  10. 2023The Internal State of an LLM Knows When It's Lyingcited by 659
  11. 2022MedMCQA : A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answeringcited by 543
  12. 2023Large Language Models Are Not Robust Multiple Choice Selectorscited by 486
20 more
  1. 2018What you can cram into a single vector: Probing sentence embeddings for linguistic propertiescited by 457
  2. 2019Cosmos QA: Machine Reading Comprehension with Contextual Commonsense Reasoningcited by 285
  3. 2023Large Language Models Sensitivity to The Order of Options in Multiple-Choice Questionscited by 263
  4. 2022Leveraging Large Language Models for Multiple Choice Question Answeringcited by 203
  5. 2023Does Circuit Analysis Interpretability Scale? Evidence from Multiple Choice Capabilities in Chinchillacited by 149
  6. 2024When Benchmarks are Targets: Revealing the Sensitivity of Large Language Model Leaderboardscited by 137
  7. 2024Benchmarking LLMs via Uncertainty Quantificationcited by 126
  8. 2024A Practical Review of Mechanistic Interpretability for Transformer-Based Language Modelscited by 118
  9. 2023Circuit Component Reuse Across Tasks in Transformer Language Modelscited by 108
  10. 2023A Mechanistic Interpretation of Arithmetic Reasoning in Language Models using Causal Mediation Analysiscited by 89
  11. 2023Overthinking the Truth: Understanding how Language Models Process False Demonstrationscited by 81
  12. 2024"My Answer is C": First-Token Probabilities Do Not Match Text Answers in Instruction-Tuned Language Modelscited by 69
  13. 2023Do Androids Know They're Only Dreaming of Electric Sheep?cited by 67
  14. 2024Attention Heads of Large Language Models: A Surveycited by 67
  15. 2023Cognitive Dissonance: Why Do Language Model Outputs Disagree with Internal Representations of Truthfulness?cited by 62
  16. 2024Changing Answer Order Can Decrease MMLU Accuracycited by 42
  17. 2024A Study on Large Language Models' Limitations in Multiple-Choice Question Answeringcited by 22
  18. 2021The heads hypothesis: A unifying statistical approach towards understanding multi-headed attention in BERTcited by 15
  19. 2024Correcting Negative Bias in Large Language Models through Negative Attention Score Alignmentcited by 12
  20. 2023Increasing Probability Mass on Answer Choices Does Not Always Improve Accuracycited by 11

Built on by

Cited by year

Citing papers per year: 2025: 1.

1 paper.