mlsanityData to 14 Oct 2026

Improving LLM First-Token Predictions in Multiple-Choice Question Answering via Output Prefilling

arXiv 2505.15323 ↗First posted 21 May 2025cs.CLcited by 6

Not on the map. Why?

Abstract (arXiv)

Large Language Models (LLMs) are increasingly evaluated on multiple-choice question answering (MCQA) tasks using *first-token probability* (FTP), which selects the answer option whose initial token has the highest likelihood. While efficient, FTP can be fragile: models may assign high probability to unrelated tokens (*misalignment*) or use a valid token merely as part of a generic preamble rather than as a clear answer choice (*misinterpretation*), undermining the reliability of symbolic evaluation. We propose a simple solution: the *prefilling attack*, a structured natural-language prefix (e.g., "*The correct option is:*") prepended to the model output. Originally explored in AI safety, we repurpose prefilling to steer the model to respond with a clean, valid option, without modifying its parameters. Empirically, the FTP with prefilling strategy substantially improves accuracy, calibration, and output consistency across a broad set of LLMs and MCQA benchmarks. It outperforms standard FTP and often matches the performance of open-ended generation approaches that require full decoding and external classifiers, while being significantly more efficient. Our findings suggest that prefilling is a simple, robust, and low-cost method to enhance the reliability of FTP-based evaluation in multiple-choice settings.

Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 41 of its 51 references that are in the map. First posted on arXiv 21 May 2025.

Builds on

41 of 51 references.

References · 41

  1. 2020Language Models are Few-Shot Learnerscited by 32,419
  2. 2023GPT-4 Technical Reportcited by 18,396
  3. 2022Training language models to follow instructions with human feedbackcited by 15,971
  4. 2024The Llama 3 Herd of Modelscited by 14,715
  5. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
  6. 2020Measuring Massive Multitask Language Understandingcited by 7,963
  7. 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,326
  8. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
  9. 2022Large Language Models are Zero-Shot Reasonerscited by 4,374
  10. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
  11. 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,512
  12. 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
29 more
  1. 2023Universal and Transferable Adversarial Attacks on Aligned Language Modelscited by 3,104
  2. 2022Emergent Abilities of Large Language Modelscited by 2,512
  3. 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
  4. 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
  5. 2023Jailbroken: How Does LLM Safety Training Fail?cited by 1,651
  6. 2022Rethinking the Role of Demonstrations: What Makes In-Context Learning Work?cited by 1,434
  7. 2021Calibrate Before Use: Improving Few-Shot Performance of Language Modelscited by 1,392
  8. 2023A Survey on Evaluation of Large Language Modelscited by 1,242
  9. 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,075
  10. 2017RACE: Large-scale ReAding Comprehension Dataset From Examinationscited by 861
  11. 2023Whose Opinions Do Language Models Reflect?cited by 773
  12. 2017Crowdsourcing Multiple Choice Science Questionscited by 765
  13. 2019MathQA: Towards Interpretable Math Word Problem Solving with Operation-Based Formalismscited by 662
  14. 2023A Comprehensive Overview of Large Language Modelscited by 648
  15. 2022Towards Reasoning in Large Language Models: A Surveycited by 596
  16. 2024Jailbreaking Leading Safety-Aligned LLMs with Simple Adaptive Attackscited by 486
  17. 2019Measuring Calibration in Deep Learningcited by 446
  18. 2016A Corpus and Evaluation Framework for Deeper Understanding of Commonsense Storiescited by 417
  19. 2021Surface Form Competition: Why the Highest Probability Answer Isn't Always Rightcited by 263
  20. 2020Moral Stories: Situated Reasoning about Norms, Intents, Actions, and their Consequencescited by 135
  21. 2019"Going on a vacation" takes longer than "Going for a walk": A Study of Temporal Commonsense Understandingcited by 123
  22. 2023Evaluating Open-Domain Question Answering in the Era of Large Language Modelscited by 111
  23. 2024"My Answer is C": First-Token Probabilities Do Not Match Text Answers in Instruction-Tuned Language Modelscited by 69
  24. 2024Can multiple-choice questions really be useful in detecting the abilities of LLMs?cited by 69
  25. 2025Which of These Best Describes Multiple Choice Evaluation with LLMs? A) Forced B) Flawed C) Fixable D) All of the Abovecited by 52
  26. 2025Right Answer, Wrong Score: Uncovering the Inconsistencies of LLM Evaluation in Multiple-Choice Question Answeringcited by 28
  27. 2024DeLLMa: Decision Making Under Uncertainty with Large Language Modelscited by 21
  28. 2024xFinder: Large Language Models as Automated Evaluators for Reliable Evaluationcited by 19
  29. 2024Look at the Text: Instruction-Tuned Language Models are More Robust Multiple Choice Selectors than You Thinkcited by 14

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 6.

6 papers.