SYNTHESIS NOTE
TopicsMemorythis note

Where do memorization errors arise in chain-of-thought reasoning?

Explores whether memorization in language model reasoning can be localized to specific token sources and which sources dominate error patterns during long generations.

Synthesis note · 2026-02-23 · sourced from Memory
How should we spend compute at inference time? What kind of thing is an LLM really? How do you navigate synthesis across fragmented research topics?

STIM (Source-aware Token-level Identification of Memorization) argues that memorization in long CoT generations must be identified at the token level, not the sequence level. A single faulty token — produced by memorization rather than reasoning — can trigger cascading errors through subsequent steps. Existing metrics report a single score for the entire sequence, missing where and why individual tokens go wrong.

Three distinct memorization sources influence each token:

  1. Local memorization — frequent continuations of immediately preceding tokens. The model generates the next token based on statistical co-occurrence with its local context, not reasoning. This is the dominant error source, responsible for up to 67% of wrong tokens.

  2. Mid-range memorization — tokens that frequently co-occur with the generation prefix. The model has seen this pattern in pretraining and reproduces it, even when the current reasoning context requires a different continuation.

  3. Long-range memorization — frequent co-occurrence with tokens in the input prompt. The prompt triggers a familiar pattern from pretraining that overrides the reasoning chain.

Key distributional findings:

This connects to the broader reasoning trace reliability cluster. Since Which sentences actually steer a reasoning trace?, STIM adds a complementary mechanism: specific tokens at the sub-sentence level carry memorization-driven influence that can derail even well-structured reasoning chains. The failure is more granular than thought-level — it operates at individual tokens.

The practical implication: high memorization scores are strong indicators of reasoning failures (measured via Precision@k and Recall@k). This offers a potential diagnostic tool for identifying where reasoning chains are unreliable, independent of whether the final answer is correct. This diagnostic capability directly addresses the faithfulness problem: since Do language models actually use their reasoning steps?, STIM's memorization scores provide a token-level mechanism for faithfulness failure — memorized tokens are causally unnecessary (the answer was determined by pattern-matching, not reasoning) and causally insufficient (the memorized continuation may diverge from the reasoning the chain appears to perform).

Inquiring lines that read this note 103

This note is a source for these research framings, grouped by the broader line of inquiry each explores. Scan the bold lines of inquiry; follow any specific question forward.

Why do reasoning models fail at systematic problem-solving and search? How do training data properties shape reasoning capability development? What memory architectures best support persistent reasoning across extended interactions? What structural biases does transformer attention create in language model outputs? Why do correct reasoning traces tend to be shorter than incorrect ones? What actually drives chain-of-thought reasoning improvements in language models? How does memorization interact with learning and generalization? How can AI systems learn from failures without cascading errors? Why does supervised fine-tuning improve accuracy while degrading reasoning quality? When do additional thinking tokens stop improving reasoning performance? How do transformer attention mechanisms implement memory and algorithmic functions? Why do continual learning scenarios trigger catastrophic forgetting and interference? Can prompting strategies overcome LLM biases without model fine-tuning? Do corrupted reasoning traces serve as effective supervision signals? Do reasoning traces faithfully represent or merely mimic actual model reasoning? How should memory consolidation strategies shape agent performance over time? Does parallel reasoning outperform sequential thinking under fixed compute budgets? How does latent reasoning compare to verbalized chain-of-thought? Why does verification consistently lag behind AI generation? Why do language models struggle with implicit discourse relations? Do language models understand semantics or rely on pattern matching? What memory abstraction level best enables agent knowledge reuse? Does self-reflection enable models to reliably correct their errors? How do LLMs distinguish causal reasoning from temporal and semantic associations? What pretraining choices and baseline capability constrain reinforcement learning gains? Can next-token prediction alone produce genuine language understanding? Why does consolidated memory sometimes degrade agent performance? What structural advantages do diffusion language models offer over autoregressive methods? What capability tradeoffs emerge when scaling model reasoning abilities? Why do benchmark improvements fail to reflect actual reasoning quality? How should iterative research systems allocate reasoning per search step? Why does finetuning cause catastrophic forgetting of model capabilities? How do training priors constrain what context information can override?

Related concepts in this collection 5

This note in its neighbourhood — explore the map, then jump to a related concept in the list below.

Concept map
14 direct connections · 137 in 2-hop network ·dense cluster Open in graph ↗

Click a node to walk · click center to open · click Open in graph to see this note in the full knowledge graph

your link semantically near linked from elsewhere

Related papers in this collection 8

Papers most semantically related to this note, ranked by cosine similarity in the embedding space.

Original note title

token-level memorization in CoT reasoning has three distinct sources and local memorization causes up to 67 percent of reasoning errors