Factual recall in linear associative memories: sharp asymptotics and mechanistic insights
Not on the map. Why?
Abstract (arXiv)
Large language models demonstrate remarkable ability in factual recall, yet the fundamental limits of storing and retrieving input--output associations with neural networks remain unclear. We study these limits in a minimal setting: a linear associative memory that maps p input embeddings in ℝd to their corresponding~d-dimensional targets via a single layer, requiring each mapped input to be well separated from all other targets. Unlike in supervised classification, this strict separation induces~p constraints per association and produces strong correlations between constraints that make a direct characterisation of the storage capacity difficult. Here, we provide a precise characterisation of this capacity in the following way. We first introduce a decoupled model in which each input has its own independent set of competing outputs, and provide numerical and analytical evidence that this decoupled model is equivalent to the original model in terms of storage capacity, spectra of the learnt weights, and storage mechanism. Using tools from statistical physics, we show that the decoupled model can store up to pc log pc / d2 = 1 / 2 associations, and generalise the computation of pc to linear two-layer architectures. Our analysis also gives mechanistic insight into how the optimal solution improves over a na"ive Hebbian learning rule: rather than boosting input-output alignments with broad fluctuations, the optimal solution raises the correct scores just above the extreme-value threshold set by the competing outputs. These findings give a sharp statistical-physics characterisation of factual storage in linear networks and provide a baseline for understanding the memory capacity of more realistic neural architectures.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 33 of its 59 references that are in the map. First posted on arXiv 11 May 2026.
Builds on
33 of 59 references.
References · 33
- 2022Locating and Editing Factual Associations in GPTcited by 2,609
- 2019Language Models as Knowledge Bases?cited by 1,922
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
- 2020How Much Knowledge Can You Pack Into the Parameters of a Language Model?cited by 713
- 2020Hopfield Networks is All You Needcited by 461
- 2016Dense Associative Memory for Pattern Recognitioncited by 230
- 2024Physics of Language Models: Part 3.3, Knowledge Capacity Scaling Lawscited by 137
- 2017Optimal Errors and Phase Transitions in High-Dimensional Generalized Linear Modelscited by 115
- 2020The Gaussian equivalence of generative models for learning with shallow neural networkscited by 113
- 2020Large Associative Memory Problem in Neurobiology and Machine Learningcited by 110
- 2022Universality of empirical risk minimizationcited by 73
- 2024Understanding Factual Recall in Transformers via Associative Memoriescited by 45
21 more
- 2023Scaling Laws for Associative Memoriescited by 30
- 2022Gaussian Universality of Perceptrons with Random Labelscited by 26
- 2020Asymptotic Errors for Teacher-Student Convex Generalized Linear Models (or : How to Prove Kabashima's Replica Formula)cited by 23
- 2024Bayes-optimal learning of an extensive-width neural network from quadratically many samplescited by 23
- 2019Mean-field inference methods for neural networkscited by 16
- 2024Learning Associative Memories with Gradient Descentcited by 14
- 2021Tractability from overparametrization: The example of the negative perceptroncited by 12
- 2025Asymptotics of Non-Convex Generalized Linear Models in High-Dimensions: A proof of the replica formulacited by 11
- 2025Fundamental Limits of Matrix Sensing: Exact Asymptotics, Universality, and Applicationscited by 11
- 2025The Nuclear Route: Sharp Asymptotics of ERM in Overparameterized Quadratic Networkscited by 11
- 2023Injectivity of ReLU networks: perspectives from statistical physicscited by 10
- 2025Statistical mechanics of extensive-width Bayesian neural networks near interpolationcited by 9
- 2023Fitting an ellipsoid to random points: predictions using the replica methodcited by 7
- 2023Exact threshold for approximate ellipsoid fitting of random pointscited by 7
- 2020Strong replica symmetry for high-dimensional disordered log-concave Gibbs measurescited by 6
- 2026Sharp Capacity Scaling of Spectral Optimizers in Learning Associative Memorycited by 6
- 2025When does Gaussian equivalence fail and how to fix it: Non-universal behavior of random features with quadratic scalingcited by 5
- 2026High-Dimensional Analysis of Gradient Flow for Extensive-Width Quadratic Neural Networkscited by 5
- 2024Bilinear Sequence Regression: A Model for Learning from Long Sequences of High-dimensional Tokenscited by 3
- 2026Sharp Capacity Thresholds in Linear Associative Memory: From Top-1 Retrieval to Tail-Average Learningcited by 2
- 2026Learning to Recall with Transformers Beyond Orthogonal Embeddingscited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.