Abstract:We propose a two-stage memory retrieval dynamics for modern Hopfield models, termed $\mathtt{U\text{-}Hop}$, with enhanced memory capacity. Our key contribution is a learnable feature map $\Phi$ which transforms the Hopfield energy function into kernel space. This transformation ensures convergence between the local minima of energy and the fixed points of retrieval dynamics within the kernel space. Consequently, the kernel norm induced by $\Phi$ serves as a novel similarity measure. It utilizes the stored memory patterns as learning data to enhance memory capacity across all modern Hopfield models. Specifically, we accomplish this by constructing a separation loss $\mathcal{L}_\Phi$ that separates the local minima of kernelized energy by separating stored memory patterns in kernel space. Methodologically, $\mathtt{U\text{-}Hop}$ memory retrieval process consists of: (Stage I) minimizing separation loss for a more uniform memory (local minimum) distribution, followed by (Stage II) standard Hopfield energy minimization for memory retrieval. This results in a significant reduction of possible metastable states in the Hopfield energy function, thus enhancing memory capacity by preventing memory confusion. Empirically, with real-world datasets, we demonstrate that $\mathtt{U\text{-}Hop}$ outperforms all existing modern Hopfield models and state-of-the-art similarity measures, achieving substantial improvements in both associative memory retrieval and deep learning tasks. Code is available at <a class="link-external link-https" href="https://github.com/MAGICS-LAB/UHop" rel="external noopener nofollow">this https URL</a> ; future updates are on <a class="link-https" data-arxiv-id="2404.03827" href="https://arxiv.org/abs/2404.03827">arXiv:2404.03827</a>

On Computational Limits of Modern Hopfield Models: A Fine-Grained Complexity Analysis

On the Expressive Power of Modern Hopfield Networks

Nonparametric Modern Hopfield Models

On Sparse Modern Hopfield Model

Analysis of the Hopfield Model with Discrete Coupling

Provably Optimal Memory Capacity for Modern Hopfield Models: Transformer-Compatible Dense Associative Memories as Spherical Codes

Uniform Memory Retrieval with Larger Capacity for Modern Hopfield Models

The capacity of the Hopfield associative memory

Storage Capacity of the Hopfield Network Associative Memory

On the Critical Capacity of the Hopfield Model

Dynamic Capacity Estimation in Hopfield Networks

A generalized Hopfield model to store and retrieve mismatched memory patterns

Input-Driven Dynamics for Robust Memory Retrieval in Hopfield Networks

The Complex Hopfield Model

High storage capacity in the Hopfield model with auto-interactions - stability analysis

Estimation of domain of attraction and exponential convergence rate of Hopfield's continuous-time associative memory and its applications

On a model of associative memory with huge storage capacity

A new mechanical approach to handle generalized Hopfield neural networks

Large Associative Memory Problem in Neurobiology and Machine Learning

Long Sequence Hopfield Memory

Quantum Hopfield Model with Dilute Memories