On the Tip of the Tongue: First-Token Cueing as a Behavioral Probe of Recall Failures in LLMs
Organizations: Department of Rural Systems Engineering, Seoul National University · Graduate School of Data Science, Seoul National University · Department of Aerospace Engineering, Seoul National University
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.
Figures & tables
| Errors recovered (%) | ||||
|---|---|---|---|---|
| Model | Stratum | Errors | Cue | Oracle |
| Llama-3.1 8B | IN | 2,004 (43%) | 54.0 | 63.0 |
| NOT-IN | 2,624 (57%) | 40.4 | 24.8 | |
| Llama-3.1 70B | IN | 1,265 (41%) | 49.6 | 57.9 |
| NOT-IN | 1,793 (59%) | 44.6 | 22.6 | |
| Changed | EM | |||||||
|---|---|---|---|---|---|---|---|---|
| Model | Cor. | Inc. | Jacc. | Pref. | Base | Sel. | Cue | |
| Llama-3.1 8B | 2,332 | 74 | 78 | 0.76 | 66.7 | 15.3 | 25.3 | 30.1 |
| Llama-3.1 70B | 1,485 | 60 | 63 | 0.62 | 65.0 | 26.9 | 29.5 | 35.7 |
| Qwen-2.5 7B | 2,845 | 83 | 89 | 0.82 | 64.2 | 11.6 | 25.3 | 31.1 |
| Gemma-3 12B | 1,760 | 80 | 83 | 0.78 | 60.0 | 14.7 | 25.8 | 34.2 |
Appendix figures & tables28 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Year | Description | License |
|---|---|---|---|
| TriviaQA ( Joshi et al., 2017 ) | 2017 | Open-domain trivia questions with Wikipedia-grounded short answers. We use the rc.wikipedia.nocontext subset, validation split. | Apache 2.0 |
| NQ-Open ( Kwiatkowski et al., 2019 ) | 2019 | Open-domain variant of Natural Questions: real Google search queries paired with short Wikipedia-grounded answers. Validation split. | CC BY-SA 3.0 |
| WebQuestions ( Berant et al., 2013 ) | 2013 | Real-world questions issued to the Google Suggest API, paired with Freebase-grounded short answers. Test split. | CC BY 4.0 |
| MKQA ( Longpre et al., 2021 ) | 2021 | Multilingual Knowledge Questions and Answers: NQ-Open queries translated and answered in 26 languages. We use 7 languages (en, es, ru, ar, ja, zh_cn, ko). | CC BY-SA 3.0 |
| Condition | Prompt | Max tokens | Temp. | Top- | |
|---|---|---|---|---|---|
| Baseline | Direct | 50 | 0.0 | – | 1 |
| First-Token Cueing | Direct + cue | 50 | 0.0 | – | 1 |
| Greedy CoT | CoT | 1024 | 0.0 | – | 1 |
| pass@16 (Direct sampling) | Direct | 50 | 0.7 | 0.95 | 16 |
| pass@ (CoT sampling) | CoT | 1024 | 0.7 | 0.95 | 16 |
| Sample | Reweighted to | |||||
| Agree. | FP / FN | Agree. | Gain judge | Gain labels | ||
| EN | 90.0 | 0.80 | 0 / 5 | 91.5 | ||
| KO | 92.0 | 0.84 | 1 / 3 | 90.6 | ||
| JA | 98.0 | 0.96 | 0 / 1 | 96.6 | ||
| ZH | 92.0 | 0.84 | 0 / 4 | 89.1 | ||
| Baseline | 95.0 | 0.90 | 1 / 4 | 93.2 | – | – |
| Model | Errors | Corrected (%) | Broken (%) | Ratio | |
|---|---|---|---|---|---|
| Gemma-3 270M | 11,593 | 10,875 | 21.6 | 37.2 | 8.8 |
| Gemma-3 1B | 11,593 | 9,253 | 32.7 | 18.8 | 6.9 |
| Gemma-3 4B | 11,593 | 6,623 | 41.4 | 9.3 | 5.9 |
| Gemma-3 12B | 11,593 | 4,829 | 43.3 | 6.4 | 4.9 |
| Gemma-3 27B | 11,593 | 3,856 | 44.6 | 4.9 | 4.6 |
| Llama-3.2 1B | 12,253 | 8,160 | 44.2 | 12.4 | 7.1 |
| Gemma-3 | Llama 3.1/3.2 | Qwen-2.5 | |||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Task | Metric | 270M | 1B | 4B | 12B | 27B | 1B | 3B | 8B | 70B | 0.5B | 1.5B | 3B | 7B | 14B | 32B | 72B |
| TriviaQA | Baseline | 6.7 | 23.9 | 54.1 | 74.4 | 83.0 | 40.2 | 63.3 | 77.0 | 90.9 | 16.8 | 36.9 | 44.9 | 58.1 | 72.5 | 74.8 | 83.2 |
| Cueing | 29.0 | 52.5 | 75.5 | 86.5 | 92.2 | 68.1 | 81.6 | 89.2 | 94.9 | 46.9 | 65.6 | 70.3 | 78.6 | 86.0 | 86.9 | 91.6 | |
| +22.3 | +28.6 | +21.4 | +12.1 | +9.2 | +27.9 | +18.3 | +12.2 | +4.0 | +30.1 | +28.7 | +25.4 | +20.4 | +13.5 | +12.2 | +8.4 | ||
| 6,560 | 6,560 | 6,560 | 6,560 | 6,560 | 7,075 | 7,075 | 7,075 | 7,075 | 7,157 | 7,157 | 7,157 | 7,157 | 7,157 | 7,157 | 7,157 | ||
| NQ-Open | Baseline | 3.3 | 11.7 | 24.2 | 33.7 | 43.9 | 20.5 | 33.0 | 40.0 | 53.9 | 9.0 | 16.8 | 20.3 | 27.8 | 37.3 | 37.5 | 44.6 |
| Gemma-3 | Llama 3.1/3.2 | Qwen-2.5 | |||||||||||||||
| Task | Metric | 270M | 1B | 4B | 12B | 27B | 1B | 3B | 8B | 70B | 0.5B | 1.5B | 3B | 7B | 14B | 32B | 72B |
| TriviaQA | Baseline | 8.5 | 26.9 | 56.6 | 75.7 | 83.9 | 41.5 | 64.3 | 77.2 | 90.9 | 18.8 | 39.0 | 46.9 | 59.5 | 73.5 | 75.6 | 83.8 |
| Random | 0.5 | 5.4 | 13.5 | 19.6 | 33.9 | 6.4 | 14.5 | 53.1 | 40.9 | 3.1 | 5.8 | 19.2 | 27.0 | 57.3 | 48.9 | 63.3 | |
| NQ-Open | Baseline | 3.9 | 12.7 | 25.5 | 35.2 | 44.8 | 20.9 | 33.3 | 40.4 | 54.0 | 9.6 | 17.4 | 21.1 | 28.9 | 38.0 | 38.0 | 45.3 |
| Random | 0.3 | 3.8 | 7.2 | 9.7 | 18.4 | 4.0 | 9.6 | 27.5 | 29.2 | 2.1 | 3.0 | 8.4 | 14.1 | 29.3 | 24.2 | 35.5 | |
| Exact Match (%) | Token F1 | |||||
| Model | Base | Cue | Base | Cue | ||
| Gemma-3 270M | 3.10 | 25.44 | +22.34 | 10.12 | 43.86 | +33.73 |
| Gemma-3 1B | 10.47 | 30.72 | +20.25 | 18.40 | 48.07 | +29.67 |
| Gemma-3 4B | 27.42 | 46.56 | +19.15 | 37.12 | 63.01 | +25.90 |
| Gemma-3 12B | 39.71 | 55.94 | +16.23 | 49.52 | 71.51 | +21.99 |
| Gemma-3 27B | 43.66 | 55.95 | +12.29 | 55.16 | 72.08 | +16.91 |
| Exact Match (%) | Token F1 | |||||
|---|---|---|---|---|---|---|
| Language | Base | Cue | Base | Cue | ||
| EN | 18.3 | 39.2 | 26.6 | 53.9 | ||
| ES | 13.3 | 33.5 | 19.2 | 46.9 | ||
| RU | 8.1 | 17.0 | 12.4 | 25.5 | ||
| AR | 4.3 | 13.2 | 7.8 | 23.0 | ||
| JA | 3.7 | 9.5 | 25.1 | 50.8 | ||
| Model | Stratum | Base | Cue | Oracle | Cue Oracle | |
| Gold answers of at least two tokens | ||||||
| Llama-3.1 8B | IN | 7,620 | 62.2 | 75.6 | 77.3 | |
| NOT-IN | 4,633 | 28.5 | 42.4 | 35.4 | ||
| Llama-3.1 70B | IN | 8,149 | 73.5 | 81.4 | 82.5 | |
| NOT-IN | 4,104 | 40.9 | 49.1 | 45.1 | ||
| All questions | ||||||
| Cue Base | Cue Oracle | |||||
|---|---|---|---|---|---|---|
| Stratum | (8B / 70B) | 8B | 70B | 8B | 70B | |
| IN | 1 | 3,221 / 5,044 | ||||
| 2 | 1,550 / 1,917 | |||||
| 3–4 | 1,034 / 943 | |||||
| 5–8 | 721 / 492 | |||||
| 9+ | 2,106 / 822 | |||||
| Cue Oracle | ||||
|---|---|---|---|---|
| Stratum | Ambiguity | (8B / 70B) | 8B | 70B |
| IN | 1 (unique) | 880 / 946 | ||
| 2–5 | 1,834 / 1,972 | |||
| 6–20 | 1,045 / 1,129 | |||
| 21+ | 567 / 587 | |||
| NOT-IN | 1 (unique) | 748 / 682 | ||
| Model | Dataset | Base | Gold cue | Plausible wrong cue | |
|---|---|---|---|---|---|
| Llama-3.1 8B | TriviaQA | 5,262 | 66.3 | 86.1 | 51.7 |
| NQ-Open | 3,280 | 35.4 | 58.2 | 27.4 | |
| WebQ | 1,725 | 38.9 | 53.9 | 30.6 | |
| Llama-3.1 70B | TriviaQA | 3,991 | 82.7 | 91.8 | 70.4 |
| NQ-Open | 2,927 | 44.6 | 65.8 | 37.4 | |
| WebQ | 1,569 | 42.5 | 57.1 | 35.6 |
| Dataset | Base | Output prefilling | Prefill Base | Cue Base | |
|---|---|---|---|---|---|
| TriviaQA | 7,993 | 77.2 | 78.0 | ||
| NQ-Open | 3,610 | 40.6 | 41.3 | ||
| WebQ | 2,032 | 47.5 | 46.8 |
| Model | Dataset | Base EM | Gold-cue EM | Plausible-wrong-cue EM | |
|---|---|---|---|---|---|
| Llama-3.1 8B | TriviaQA | 5,262 | 56.0 | 78.9 | 38.5 |
| NQ-Open | 3,280 | 18.9 | 38.4 | 10.3 | |
| WebQ | 1,725 | 13.0 | 23.1 | 5.0 | |
| Llama-3.1 70B | TriviaQA | 3,991 | 72.8 | 85.6 | 57.4 |
| NQ-Open | 2,927 | 25.4 | 44.5 | 14.7 | |
| WebQ | 1,569 | 13.5 | 22.1 | 6.8 |
| Model | Thinking pass@16 | Share of failures | Cue recovery | |
|---|---|---|---|---|
| Llama-3.1 8B | recoverable | 2,745 | 59.3% | 59.9% |
| not recoverable | 1,883 | 40.7% | 26.6% | |
| Llama-3.1 70B | recoverable | 1,442 | 47.2% | 61.0% |
| not recoverable | 1,616 | 52.8% | 33.8% |
| Gemma-3 | Llama 3.1/3.2 | Qwen-2.5 | ||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| First-token length | 270M | 1B | 4B | 12B | 27B | 1B | 3B | 8B | 70B | 0.5B | 1.5B | 3B | 7B | 14B | 32B | 72B |
| 1 char | +9.5 | +12.8 | +14.5 | +10.7 | +8.2 | +21.2 | +17.0 | +12.4 | +6.0 | +16.9 | +17.3 | +18.2 | +16.7 | +10.8 | +9.8 | +8.1 |
| 2 chars | +11.5 | +15.4 | +16.2 | +13.0 | +10.0 | +24.2 | +17.1 | +13.6 | +9.5 | +20.5 | +20.1 | +17.7 | +15.3 | +10.1 | +10.2 | +10.1 |
| 3 chars | +17.8 | +17.0 | +18.5 | +15.8 | +12.3 | +25.7 | +19.6 | +13.0 | +9.5 | +22.4 | +24.9 | +23.3 | +19.7 | +14.5 | +12.6 | +13.0 |
| 4+ chars | +31.5 | +25.5 | +22.3 | +19.5 | +14.7 | +29.9 | +23.2 | +16.2 | +12.1 | +29.2 | +31.2 | +27.2 | +23.4 | +17.2 | +17.0 | +16.6 |
| mean | first-token length (chars) | tok | tokens | ||||
|---|---|---|---|---|---|---|---|
| Family | chars | 1 | 2 | 3 | 4+ | = ans. | / answer |
| Gemma-3 | 3.84 | 19.9% | 10.1% | 22.9% | 47.1% | 15.0% | 3.21 |
| Llama-3 | 3.16 | 25.6% | 16.2% | 26.2% | 32.1% | 10.1% | 3.41 |
| Qwen-2.5 | 3.06 | 31.3% | 14.6% | 22.0% | 32.1% | 9.0% | 3.60 |
| First-token chars | 1-char first token (%) | Tokens per answer | |
|---|---|---|---|
| Data | (G / L / Q) | (G / L / Q) | (G / L / Q) |
| English QA | 3.84 / 3.16 / 3.06 | 19.9 / 25.6 / 31.3 | 3.21 / 3.41 / 3.60 |
| MKQA EN | 3.23 / 3.14 / 2.64 | 38.3 / 19.5 / 46.8 | 4.04 / 3.60 / 4.34 |
| ES | 3.01 / 2.88 / 2.39 | 39.5 / 21.9 / 49.3 | 4.23 / 3.96 / 4.69 |
| RU | 1.57 / 1.78 / 1.25 | 64.9 / 56.3 / 87.7 | 5.68 / 5.45 / 6.48 |
| AR | 1.63 / 1.99 / 1.70 | 58.5 / 38.1 / 54.3 | 5.49 / 5.08 / 5.82 |
| Llama-3.1 8B ( eligible) | Llama-3.1 70B ( ) | Qwen-2.5 7B ( ) | Gemma-3 12B ( ) | |||||||||||||
| Caliper | pairs | cov. | pref. | adv. | pairs | cov. | pref. | adv. | pairs | cov. | pref. | adv. | pairs | cov. | pref. | adv. |
| 1 nat | 873 | 37.9 | 65.6 | +0.37 | 469 | 23.5 | 63.4 | +0.34 | 1,109 | 40.5 | 63.4 | +0.31 | 513 | 16.2 | 60.1 | +0.08 |
| 2 nats | 1,171 | 50.8 | 65.5 | +0.37 | 710 | 35.5 | 64.5 | +0.32 | 1,488 | 54.3 | 64.0 | +0.31 | 843 | 26.7 | 59.4 | +0.07 |
| 3 nats | 1,371 | 59.5 | 66.7 | +0.38 | 878 | 44.0 | 65.2 | +0.31 | 1,681 | 61.3 | 64.4 | +0.31 | 1,074 | 34.0 | 60.5 | +0.07 |
| 5 nats | 1,615 | 70.1 | 68.4 | +0.38 | 1,174 | 58.8 | 68.3 | +0.32 | 1,951 | 71.2 | 65.6 | +0.30 | 1,372 | 43.4 | 60.8 | +0.07 |
| none | 2,297 | 99.7 | 74.8 | +0.41 | 1,989 | 99.6 | 76.5 | +0.36 | 2,707 | 98.8 | 68.4 | +0.31 | 3,147 | 99.5 | 65.8 | +0.07 |
| All | Greedy first char. wrong | |
| Method | ( ) | ( ) |
| Greedy (Baseline) | 50.0 | 12.1 |
| Beam search (width 8) | 53.0 | 16.9 |
| Self-consistency (16 samples) | 51.6 | 15.0 |
| 36 characters, total log-prob | 52.2 | 17.2 |
| 36 characters, continuation log-prob | 45.7 | 20.5 |
| Model | AUROC | Model | AUROC |
|---|---|---|---|
| Gemma-3 270M | 0.633 | Llama-3.1 70B | 0.832 |
| Gemma-3 1B | 0.668 | Qwen-2.5 0.5B | 0.729 |
| Gemma-3 4B | 0.757 | Qwen-2.5 1.5B | 0.801 |
| Gemma-3 12B | 0.778 | Qwen-2.5 3B | 0.758 |
| Gemma-3 27B | 0.763 | Qwen-2.5 7B | 0.739 |
| Llama-3.2 1B | 0.773 | Qwen-2.5 14B | 0.704 |
| Rank of gold first token | Share of errors (%) | Fixed by the cue (%) |
|---|---|---|
| 1 (the greedy first token) | 7.4 | 5.5 |
| 2–5 | 21.6 | 65.5 |
| 6–50 | 30.0 | 50.7 |
| 51–1,000 | 27.4 | 37.5 |
| 13.6 | 30.3 |
| Gemma-3 | Llama 3.1/3.2 | Qwen-2.5 | |||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Task | Metric | 270M | 1B | 4B | 12B | 27B | 1B | 3B | 8B | 70B | 0.5B | 1.5B | 3B | 7B | 14B | 32B | 72B |
| TriviaQA | Baseline | 8.5 | 26.9 | 56.6 | 75.7 | 83.9 | 41.5 | 64.3 | 77.2 | 90.9 | 18.8 | 39.0 | 46.9 | 59.5 | 73.5 | 75.6 | 83.8 |
| Cueing | 40.3 | 59.9 | 79.5 | 88.7 | 93.4 | 71.4 | 83.5 | 90.2 | 95.4 | 51.7 | 68.9 | 73.1 | 80.5 | 87.2 | 88.1 | 92.3 | |
| +31.8 | +33.0 | +22.9 | +12.9 | +9.4 | +29.9 | +19.3 | +13.0 | +4.6 | +33.0 | +29.9 | +26.2 | +21.1 | +13.7 | +12.5 | +8.6 | ||
| +373.4 | +122.5 | +40.4 | +17.0 | +11.3 | +72.2 | +29.9 | +16.9 | +5.0 | +175.4 | +76.8 | +55.9 | +35.5 | +18.7 | +16.5 | +10.2 | ||
| NQ-Open | Baseline | 3.9 | 12.7 | 25.5 | 35.2 | 44.8 | 20.9 | 33.3 | 40.4 | 54.0 | 9.6 | 17.4 | 21.1 | 28.9 | 38.0 | 38.0 | 45.3 |
Factual Knowledge in Language Models
Factual knowledge in language models concerns what facts they store, how reliably they retrieve them, and how that knowledge can be tested or changed.
See all 195 papers