On the Tip of the Tongue: First-Token Cueing as a Behavioral Probe of Recall Failures in LLMs
AuthorsJaewon Sok, Jewon Yeom, Seonghyeon Park, Jeongjae Park, Hwiyeong Lee, Taesup Kim
AffiliationsDepartment of Rural Systems Engineering, Seoul National University · Graduate School of Data Science, Seoul National University · Department of Aerospace Engineering, Seoul National University
A single correct opening token can help an LLM recover factual answers it seemed unable to recall.
Key results
Models from the Llama 3.1/3.2, Gemma 3, and Qwen 2.5 families.
Share corrected by one-token cueing on multi-token English QA answers.
Matched-prior crossed-cue cases where the correct start had the more confident continuation.
Range across models on English QA benchmarks for predicting answer correctness.
What the paper found
This paper introduces First-Token Cueing, a behavioral probe that gives a language model the gold answer’s first token and tests whether it can complete the answer. Across 16 open-weight models from Meta’s Llama 3.1/3.2, Google’s Gemma 3, and Alibaba’s Qwen 2.5 families, the cue corrected 20–50% of baseline errors on multi-token answers in TriviaQA, NQ-Open, and WebQuestions. It also improved accuracy in all seven tested MKQA languages. First-token entropy predicted answer correctness with AUROC scores of 0.60–0.85, supporting the finding that uncertainty concentrates at the start of an answer. A crossed-cue control found that, when correct and incorrect starts had similar prior probabilities, models continued the correct start more confidently in 60–67% of cases. The results suggest that many factual errors are not simply missing knowledge: models can often recognize and complete an answer after failing to produce its opening token. Because the cue uses the gold answer, it is a diagnostic oracle rather than a deployable fix; the paper points to retrieval or other inference-time methods as possible ways to supply useful starts without gold answers.
Original abstract
Large language models often answer factual questions incorrectly in closed-book settings, and such errors are usually attributed to missing knowledge. Yet models often encode answers they fail to produce. We show that many closed-book errors are first-token recall failures: the model commits to a wrong first token, but once given the right one it completes the answer and often recognizes it as correct. We study this with First-Token Cueing, a behavioral probe that prefills the response with the first token of the gold answer. Model uncertainty peaks at the first answer token, where it already predicts correctness, and among gold prefixes of any length the first token alone accounts for the largest share of recovered errors. Across 16 open models, this one token corrects between a fifth and a half of the errors on multi-token answers in English QA and improves accuracy in all seven languages of a multilingual benchmark. Most of the errors it corrects are also solved by the model's own sampled reasoning, a criterion for encoded knowledge, and models recognize correct starts they do not produce: they continue a correct start more confidently than an incorrect start of the same answer type and prior probability, yet rarely find it without the gold answer. These results locate a recoverable class of factual errors at the first decoding step, where inference-time methods may gain the most.
Read the original paperMore in Large Language Models
Browse all 90 papers →Belief-Trajectory Energy: Measuring the Path to a Prediction
Jiahao Ying, Wei Tang, Boxian Ai, Yaoning Wang, Haotian Chen, Wenhe Sun, Caijun Xu, Haozhan Cai, Changyi Xiao, Yixin Cao
Belief-Trajectory Energy tracks how an LLM’s predictions change across layers to reveal information that its final answer alone can miss.
Competing Learning in Hybrid Architectures
Xingyu Shen, Yingfa Chen, Ziqing Qiao, Chaojun Xiao, Xu Han, Zhiyuan Liu
A staged training approach helps hybrid language models balance recurrence and attention, improving language modeling and retrieval.
Copy-Paste Language Models
Nathan Godey, Yoav Artzi
Copy-Paste Language Models add a low-overhead copying mechanism that improves language-model training and performance, especially in long-context and retrieval settings.