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