← Search

Dmitry Krotov

13 accepted papers

2026

A Biologically Plausible Dense Associative Memory with Exponential Capacity

ICLR 2026poster

Krotov and Hopfield (2021) proposed a biologically plausible two-layer associative memory network with memory storage capacity exponential in the number of visible neurons. However, the capacity was only linear in the number of hidden neurons. This limitation arose from the choice of nonlinearity be…

Cited by 0SourceScholar
2026

NRGPT: An Energy-based Alternative for GPT

ICLR 2026poster

Generative Pre-trained Transformer (GPT) architectures are the most popular design for language modeling. Energy-based modeling is a different paradigm that views inference as a dynamical process operating on an energy landscape. We propose a minimal modification of the GPT setting to unify it with…

Cited by 0SourceScholar
2025

Dense Associative Memory with Epanechnikov Energy

NeurIPS 2025spotlight

We propose a novel energy function for Dense Associative Memory (DenseAM) networks, the log-sum-ReLU (LSR), inspired by optimal kernel density estimation. Unlike the common log-sum-exponential (LSE) function, LSR is based on the Epanechnikov kernel and enables exact memory retrieval with exponential…

Cited by 0SourceScholar
2025

M+: Extending MemoryLLM with Scalable Long-Term Memory

ICML 2025poster

Equipping large language models (LLMs) with latent-space memory has attracted increasing attention as they can extend the context window of existing language models. However, retaining information from the distant past remains a challenge. For example, MemoryLLM (Wang et al., 2024a), as a representa…

2024

Dense Associative Memory Through the Lens of Random Features

NeurIPS 2024poster

Dense Associative Memories are high storage capacity variants of the Hopfield networks that are capable of storing a large number of memory patterns in the weights of the network of a given size. Their common formulations typically require storing each pattern in a separate set of synaptic weights,…

2023

End-to-end Differentiable Clustering with Associative Memories

ICML 2023poster

Clustering is a widely used unsupervised learning technique involving an intensive discrete optimization problem. Associative Memory models or AMs are differentiable neural networks defining a recursive dynamical system, which have been integrated with various deep learning architectures. We uncover…

2023

Energy Transformer

NeurIPS 2023poster

Our work combines aspects of three promising paradigms in machine learning, namely, attention mechanism, energy-based models, and associative memory. Attention is the power-house driving modern deep learning successes, but it lacks clear theoretical foundations. Energy-based models allow a principle…

Cited by 60SourcePDFScholar
2023

Long Sequence Hopfield Memory

NeurIPS 2023poster

Sequence memory is an essential attribute of natural and artificial intelligence that enables agents to encode, store, and retrieve complex sequences of stimuli and actions. Computational models of sequence memory have been proposed where recurrent Hopfield-like neural networks are trained with temp…

2023

Sparse Distributed Memory is a Continual Learner

ICLR 2023poster

Continual learning is a problem for artificial neural networks that their biological counterparts are adept at solving. Building on work using Sparse Distributed Memory (SDM) to connect a core neural circuit with the powerful Transformer model, we create a modified Multi-Layered Perceptron (MLP) tha…

2021

Can a Fruit Fly Learn Word Embeddings?

ICLR 2021poster

The mushroom body of the fruit fly brain is one of the best studied systems in neuroscience. At its core it consists of a population of Kenyon cells, which receive inputs from multiple sensory modalities. These cells are inhibited by the anterior paired lateral neuron, thus creating a sparse high di…

Cited by 40SourcePDFScholar
2020

Bio-Inspired Hashing for Unsupervised Similarity Search

ICML 2020poster

The fruit fly Drosophila’s olfactory circuit has inspired a new locality sensitive hashing (LSH) algorithm, FlyHash. In contrast with classical LSH algorithms that produce low dimensional hash codes, FlyHash produces sparse high-dimensional hash codes and has also been shown to have superior empiric…

Cited by 37SourcePDFScholar