mlsanityData to 7 Oct 2026

A decodability criterion predicts when hidden-state selection beats majority voting in large language models

arXiv 2608.17124 ↗First posted 17 Aug 2026cs.AIcited by 2

Not on the map. Why?

Abstract (arXiv)

Combining the answers a large language model (LLM) samples for a question into one decision is a test-time information fusion problem, usually solved by majority voting. Voting is unreliable on difficult questions, where the sampled answers share correlated errors, so the wrong answer can win and drawing more samples makes the decision worse. Selecting a candidate by reading a correctness signal from the model's hidden states is a promising alternative, but its accuracy varies across models and tasks, and no measure indicates when it can be trusted. In this paper, we propose CASE (Correctness-Axis SElection), a dynamic selection combiner that trains a linear gate on the answer-token hidden state and selects the highest-scoring candidate. Its main contribution is decodability, a leakage-free measure of how well the gate ranks a question's correct candidates above its incorrect ones, which predicts whether hidden-state selection will outperform voting. A conventional probe appears accurate only because of question-identity leakage, which vanishes under question-grouped evaluation. On held-out data, decodability predicts the accuracy gain of selection over voting with a Pearson correlation r=0.75 and a decision threshold near AUC=0.60. Across general and medical LLMs, CASE improves over voting by up to 19 points on medium-difficulty questions and 16.8 points on hard questions. Decodability depends on the aligned knowledge a model must recall, not on its scale, and its prediction transfers to an unseen scientific domain within 3.8 points. It thus provides a practical criterion, measurable in advance for a given model and task, for choosing between learned selection and majority voting.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 39 of its 50 references that are in the map. First posted on arXiv 17 Aug 2026.

Builds on

39 of 50 references.

References · 39

  1. 2024The Llama 3 Herd of Modelscited by 14,192
  2. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  3. 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
  4. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  5. 2017On Calibration of Modern Neural Networkscited by 4,860
  6. 2024Qwen2.5 Technical Reportcited by 4,713
  7. 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,748
  8. 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,928
  9. 2022Language Models (Mostly) Know What They Knowcited by 1,783
  10. 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,324
  11. 2016Understanding intermediate layers using linear classifier probescited by 1,279
  12. 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,271
27 more
  1. 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,259
  2. 2024Large Language Monkeys: Scaling Inference Compute with Repeated Samplingcited by 946
  3. 2022Solving math word problems with process- and outcome-based feedbackcited by 871
  4. 2023Semantic Uncertainty: Linguistic Invariances for Uncertainty Estimation in Natural Language Generationcited by 795
  5. 2022Discovering Latent Knowledge in Language Models Without Supervisioncited by 733
  6. 2022Teaching Models to Express Their Uncertainty in Wordscited by 731
  7. 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 679
  8. 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 673
  9. 2023The Internal State of an LLM Knows When It's Lyingcited by 644
  10. 2022MedMCQA : A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answeringcited by 525
  11. 2024Generative Verifiers: Reward Modeling as Next-Token Predictioncited by 465
  12. 2024BioMistral: A Collection of Open-Source Pretrained Large Language Models for Medical Domainscited by 299
  13. 2023MEDITRON-70B: Scaling Medical Pretraining for Large Language Modelscited by 272
  14. 2024Semantic Entropy Probes: Robust and Cheap Hallucination Detection in LLMscited by 246
  15. 2024LLMs Know More Than They Show: On the Intrinsic Representation of LLM Hallucinationscited by 230
  16. 2025Scalable Best-of-N Selection for Large Language Models via Self-Certaintycited by 182
  17. 2025Reasoning Models Know When They're Right: Probing Hidden States for Self-Verificationcited by 154
  18. 2024Towards Building Multilingual Language Model for Medicinecited by 91
  19. 2024Learning How Hard to Think: Input-Adaptive Allocation of LM Computationcited by 67
  20. 2024Med42 -- Evaluating Fine-Tuning Strategies for Medical LLMs: Full-Parameter vs. Parameter-Efficient Approachescited by 51
  21. 2025Probing the Geometry of Truth: Consistency and Generalization of Truth Directions in LLMs Across Logical Transformations and Question Answering Taskscited by 18
  22. 2025Mining Intrinsic Rewards from LLM Hidden States for Efficient Best-of-N Samplingcited by 9
  23. 2025Tracing the Traces: Latent Temporal Signals for Efficient and Accurate Reasoningcited by 8
  24. 2026The Confidence Manifold: Geometric Structure of Correctness Representations in Language Modelscited by 4
  25. 2025Lightweight Latent Verifiers for Efficient Meta-Generation Strategiescited by 3
  26. 2025TrajSelector: Harnessing Latent Representations for Efficient and Effective Best-of-N in Large Reasoning Modelcited by 3
  27. 2025Quantifying Genuine Awareness in Hallucination Prediction Beyond Question-Side Shortcutscited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.