Human-like Episodic Memory for Infinite Context LLMs
Authors: Zafeirios Fountas, Martin A Benfeghoul, Adnan Oomerjee, Fenia Christopoulou, Gerasimos Lampouras, Haitham Bou-Ammar, Jun Wang
AI-generated Key Points
- Large language models (LLMs) struggle with processing extensive contexts, limiting coherence and accuracy over long sequences.
- The human brain excels at organizing and retrieving episodic experiences across vast temporal scales.
- EM-LLM integrates human-like episodic memory and event cognition into LLMs to address this limitation.
- EM-LLM organizes token sequences into coherent episodic events using Bayesian surprise and graph-theoretic boundary refinement in real-time.
- It allows the model to handle practically infinite context lengths efficiently.
- Events are retrieved through a two-stage memory process that combines similarity-based and temporally contiguous retrieval for efficient access to relevant information.
- EM-LLM outperforms the state-of-the-art InfLLM model on the LongBench dataset, showing an overall relative improvement of 4.3% and a 33% enhancement specifically on the PassageRetrieval task.
- The model's event segmentation correlates strongly with human-perceived events, bridging artificial systems with biological counterparts.
- EM-LLM leverages insights from hierarchical and nested-timescale structures in memory formation to enhance long-term memory recall efficiency in complex tasks with extended contexts.
- It provides a computational framework for exploring human memory mechanisms and opens new avenues for research in AI and cognitive science by integrating key aspects of human cognition into artificial intelligence systems.
Abstract: Large language models (LLMs) have shown remarkable capabilities, but still struggle with processing extensive contexts, limiting their ability to maintain coherence and accuracy over long sequences. In contrast, the human brain excels at organising and retrieving episodic experiences across vast temporal scales, spanning a lifetime. In this work, we introduce EM-LLM, a novel approach that integrates key aspects of human episodic memory and event cognition into LLMs, enabling them to effectively handle practically infinite context lengths while maintaining computational efficiency. EM-LLM organises sequences of tokens into coherent episodic events using a combination of Bayesian surprise and graph-theoretic boundary refinement in an on-line fashion. When needed, these events are retrieved through a two-stage memory process, combining similarity-based and temporally contiguous retrieval for efficient and human-like access to relevant information. Experiments on the LongBench dataset demonstrate EM-LLM's superior performance, outperforming the state-of-the-art InfLLM model with an overall relative improvement of 4.3% across various tasks, including a 33% improvement on the PassageRetrieval task. Furthermore, our analysis reveals strong correlations between EM-LLM's event segmentation and human-perceived events, suggesting a bridge between this artificial system and its biological counterpart. This work not only advances LLM capabilities in processing extended contexts but also provides a computational framework for exploring human memory mechanisms, opening new avenues for interdisciplinary research in AI and cognitive science.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.