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

Jerry Yao Chieh Hu*, Thomas Lin*, Zhao Song*, Han Liu*

*Corresponding author for this work

Research output: Contribution to journalConference articlepeer-review

1 Scopus citations

Abstract

We investigate the computational limits of the memory retrieval dynamics of modern Hopfield models from the fine-grained complexity analysis. Our key contribution is the characterization of a phase transition behavior in the efficiency of all possible modern Hopfield models based on the norm of patterns. Specifically, we establish an upper bound criterion for the norm of input query patterns and memory patterns. Only below this criterion, sub-quadratic (efficient) variants of the modern Hopfield model exist, assuming the Strong Exponential Time Hypothesis (SETH). To showcase our theory, we provide a formal example of efficient constructions of modern Hopfield models using low-rank approximation when the efficient criterion holds. This includes a derivation of a lower bound on the computational time, scaling linearly with max{# of stored memory patterns, length of input query sequence}. In addition, we prove its memory retrieval error bound and exponential memory capacity.

Original languageEnglish (US)
Pages (from-to)19327-19343
Number of pages17
JournalProceedings of Machine Learning Research
Volume235
StatePublished - 2024
Event41st International Conference on Machine Learning, ICML 2024 - Vienna, Austria
Duration: Jul 21 2024Jul 27 2024

Funding

JH would like to thank Feng Ruan, Dino Feng and Andrew Chen for enlightening discussions on related topics, and the Red Maple Family for support. TL would like to thank Gamma Paradigm Research and NTU ABC-Labs for support. The authors would like to thank the anonymous reviewers and program chairs for constructive comments. JH is partially supported by the Walter P. Murphy Fellowship. HL is partially supported by NIH R01LM1372201, NSF CAREER1841569, DOE DE-AC02-07CH11359, DOE LAB 20-2261 and a NSF TRIPODS1740735. The content is solely the responsibility of the authors and does not necessarily represent the official views of the funding agencies.

ASJC Scopus subject areas

  • Artificial Intelligence
  • Software
  • Control and Systems Engineering
  • Statistics and Probability

Fingerprint

Dive into the research topics of 'On Computational Limits of Modern Hopfield Models: A Fine-Grained Complexity Analysis'. Together they form a unique fingerprint.

Cite this