Beyond Multiplicity Correction: Sharp Retrieval Thresholds for Attention with Correlated Distractors
Abstract
Redundant context affects attention in two distinct ways: multiplicity adds normalization mass, while score variation within correlated groups can produce extreme individual scores. We separate these effects in a planted retrieval model with exponentially many Gaussian distractors organized into supplied clusters whose number and size both grow exponentially with dimension. The model is a two-level generalized random energy model, whose classical free energy yields exact retrieval thresholds. Gaussian interpolation further shows that increasing positive correlation at fixed token count lowers the distractor free energy, unlike adding duplicates. Our main result establishes a sharp limitation of count correction, which subtracts the log cluster size from pre-softmax scores: this correction removes multiplicity but retains within-cluster extremes. Even when the inverse temperature is chosen after observing all scores, count correction requires a signal at least , where is the exponential growth rate of the number of clusters. By contrast, averaging scores before exponentiation suppresses token-level fluctuations and lowers the optimal signal threshold to , where is the within-cluster correlation. This gap persists after adding exponentially many singleton distractors, so group size alone does not identify the target, and extends, under explicit cumulant regularity assumptions, to fixed finite hierarchies of independent-increment scores. Finite-sample, stability, and participation results complement the asymptotic analysis. A controlled retrieval experiment with frozen sentence embeddings illustrates both sides of the distinction: averaging helps when the relevant item is a singleton, but can discard useful evidence when a relevant group contains multiple matching items.
Figures & tables
| Context | Ordinary | Count | Average | Difference [95% interval] |
|---|---|---|---|---|
| One key per intent | 0.379 | 0.379 | 0.379 | [ , ] |
| Exact copies (8) | 0.288 | 0.379 | 0.379 | [ , ] |
| Same intent (4) | 0.267 | 0.345 | 0.406 | [ , ] |
| Same intent (8) | 0.234 | 0.342 | 0.440 | [ , ] |
| Four relevant keys | 0.614 | 0.626 | 0.465 | [ , ] |
| Mixed relevant group | 0.234 | 0.243 | 0.005 | [ , ] |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Context | Method | Accuracy (%) | ||
|---|---|---|---|---|
| One key per intent | Ordinary | 14.42 | 0.379 | 51.8 |
| One key per intent | Count correction | 14.42 | 0.379 | 51.8 |
| One key per intent | Averaging | 14.42 | 0.379 | 51.8 |
| One key per intent | One representative | 14.42 | 0.379 | 51.8 |
| Exact copies (8) | Ordinary | 17.75 | 0.288 | 36.3 |
| Exact copies (8) | Count correction | 14.42 | 0.379 | 51.8 |
| Condition | relevant mass | candidate prob. | |
|---|---|---|---|
| All singleton | .00000 | .00000 | |
| Repeated facts (4) | .00032 [.00030,.00035] | -.00215 [-.00253,-.00181] | |
| Related (4), selected | -.00139 [-.00291,.00001] | -.02042 [-.03285,-.00946] | |
| Related (4), native | .00018 [.00017,.00020] | -.00159 [-.00186,-.00133] | |
| Four relevant | -.00083 [-.00093,-.00074] | .00054 [.00014,.00098] |
Attention Mechanisms
Attention mechanisms let neural networks weight information from other tokens or regions according to relevance, enabling context-dependent processing.
See all 517 papers