cs.LGSep 28, 2026

Intrinsic Associative Memory on Riemannian Manifolds: Curvature, Capacity, and Emergent Modes

Authors: Krishnakumar Balasubramanian, Zhaoyang Shi

Organizations: Department of Statistics, University of California, Davis. · Center for Applied Mathematics, Fudan University.

Abstract

Geometry does more than constrain an associative memory: curvature determines what it remembers and which states it creates. We develop intrinsic dense associative memories on Riemannian manifolds by casting memory as Epanechnikov kernel-density mode seeking. We compare geodesic and volume-corrected energies and show that curvature separates their behavior. We prove that geodesic memory always retains an isolated pattern, while corrected memory obeys a sharp Ricci-curvature threshold: positive curvature can erase memories in high dimensions, while negative curvature reinforces them. We derive geodesic capacity scalings of qβ−1/2q_β^{-1/2} for retaining every pattern and qβ−1q_β^{-1} for a typical one, where qβq_β is the pairwise kernel-overlap probability. We show how overlap \emph{creates} novel memories: designed NN-pattern configurations realize all 2N−12^N-1 subset modes, but random data at the storage threshold yield only a Poisson number. We establish exact one-step recall using Riemannian mean shift. In simulations, we recover the predicted curvature transition and every designed mode. On WordNet's full noun hierarchy, we demonstrate that volume correction improves low-capacity retrieval. Together, our work shows that curvature is a design variable for associative memory, not merely a property of the data.

Figures & tables

Appendix figures & tables1 asset

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

May 11, 2026stat.ML

Factual recall in linear associative memories: sharp asymptotics and mechanistic insights

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 pp input embeddings in Rd\mathbb{R}^d to their corresponding~dd-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~pp 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 pclog⁡pc/d2=1/2p_c \log p_c / d^2 = 1 / 2 associations, and generalise the computation of pcp_c to linear two-layer architectures. Our analysis also gives mechanistic insight into how the optimal solution improves over a naïve 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.
Jul 21, 2026cond-mat.dis-nn

Free energy landscape of Dense Associative Memory

Using large deviations theory, we solve and obtain a general expression for the free energy functional for a broad class of associative memories, including dense associative memories. We illustrate the method by reproducing classical results for the Hopfield model. For a finite number of patterns, we derive the temperature-dependent free energy functional for dense associative memories featuring polynomial interactions and Log-Sum-Exponential (LSE) activation. We also evaluate the disorder-averaged ground-state energy of these systems in the extensive limit. Our analytical framework reveals how memory retrieval depends on the initial state in higher-order dense networks, and gives the exact full-retrieval threshold for the LSE model. This method provides a systematic procedure for analyzing diverse, complex architectures in associative memory.
Sep 15, 2026cond-mat.dis-nn

Bias-Induced Crossover in Absolute Capacity of Dense Associative Memory

The absolute capacity of dense associative memory has mainly been analyzed for unbiased patterns. Here we examine the effect of bias in centered binary patterns under the Krotov-Hopfield single-site criterion Perror=1/NP_{\mathrm{error}}=1/N, where PerrorP_{\mathrm{error}} is the probability that a single-site flip lowers the energy of a stored pattern and NN is the number of neurons. Each pattern component takes 1−q1-q with probability qq and −q-q otherwise, where 0<q≤1/20<q\le1/2. For polynomial interactions of order nn, a signal-to-noise analysis gives an absolute capacity of order Nn−1/ln⁡NN^{n-1}/\ln N at q=1/2q=1/2. For fixed q<1/2q<1/2, however, the capacity is O(Nn/2)O(N^{n/2}) for even n≥4n\ge4 and O(N(n+1)/2)O(N^{(n+1)/2}) for odd n≥5n\ge5. For n=3n=3, both the unbiased and fixed-bias capacities remain O(N2/ln⁡N)O(N^2/\ln N). For n≥4n\ge4, these different asymptotic forms imply a nonuniform large-NN limit near q=1/2q=1/2. Asymptotic matching predicts a bias-induced crossover in the region 1−2q=O(ln⁡N/N⌊n/2⌋−1)1-2q=O(\ln N/N^{\lfloor n/2\rfloor-1}). The crossover originates from a bias-dependent crosstalk mean that reduces the stability of sites carrying the more frequent value −q-q. Computer simulations are compared with the finite-size conditioned-Gaussian predictions. An activity-dependent control potential that cancels the conditional crosstalk mean restores the Nn−1/ln⁡NN^{n-1}/\ln N capacity for fixed 0<q<1/20<q<1/2 within the conditioned-Gaussian approximation.