NTH

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

October 26, 2026 2 min read
Watch on YouTube
The one-line take

A single correct opening token can help an LLM recover factual answers it seemed unable to recall.

Key results

16
Evaluated open-weight models

Models from the Llama 3.1/3.2, Gemma 3, and Qwen 2.5 families.

20–50%
Baseline errors corrected

Share corrected by one-token cueing on multi-token English QA answers.

60–67%
Correct-start confidence preference

Matched-prior crossed-cue cases where the correct start had the more confident continuation.

0.60–0.85
First-token entropy AUROC

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 paper

More in Large Language Models

Browse all 90 papers →
01Llm

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.

Read analysis
02Llm

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.

Read analysis
03Llm

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.

Read analysis