EM-LLM: Human-Inspired Episodic Memory for Infinite Context LLMs

Note that this works within a single sequence of tokens. It might be consistent with &quot;episodic memory&quot; metaphor if we consider a particular transformer run as its experience.<p>But this might be very different from what people expect from &quot;memory&quot; - i.e. ability to learn vast amounts of information and retrieve it as necessary.<p>This is more like a refinement of transformer attention: instead of running attention over all tokens (which is very expensive as it&#x27;s quadratic), it selects a subset of token spans and runs fine-grained attention only on those. So it essentially breaks transformer attention into two parts - coarse-grained (k-NN over token spans) and fine-grained (normal).<p>It might be a great thing for long-context situations. But it doesn&#x27;t make sense when you want millions of different facts to be considered - making them into long context is rather inefficient.

TTT, cannon layers, and titans seem like a stronger approach IMO.<p>Information needs to be compressed into latent space or it becomes computationally intractable

Interesting. Before there even was attention I was thinking that the episodic memory model offers something that could be very useful for neural nets, so its cool to see people testing that

So, infinite context length by making it compute bound instead of memory bound. Curious how much longer this takes to run and when it makes sense to use vs RAG.