From Isolated Conversations to Hierarchical Schemas: Dynamic Tree Memory Representation for LLMs

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Rezazadeh, Alireza, Li, Zichao, Wei, Wei, Bao, Yujia
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916657552687104
author Rezazadeh, Alireza
Li, Zichao
Wei, Wei
Bao, Yujia
author_facet Rezazadeh, Alireza
Li, Zichao
Wei, Wei
Bao, Yujia
contents Recent advancements in large language models have significantly improved their context windows, yet challenges in effective long-term memory management remain. We introduce MemTree, an algorithm that leverages a dynamic, tree-structured memory representation to optimize the organization, retrieval, and integration of information, akin to human cognitive schemas. MemTree organizes memory hierarchically, with each node encapsulating aggregated textual content, corresponding semantic embeddings, and varying abstraction levels across the tree's depths. Our algorithm dynamically adapts this memory structure by computing and comparing semantic embeddings of new and existing information to enrich the model's context-awareness. This approach allows MemTree to handle complex reasoning and extended interactions more effectively than traditional memory augmentation methods, which often rely on flat lookup tables. Evaluations on benchmarks for multi-turn dialogue understanding and document question answering show that MemTree significantly enhances performance in scenarios that demand structured memory management.
format Preprint
id arxiv_https___arxiv_org_abs_2410_14052
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle From Isolated Conversations to Hierarchical Schemas: Dynamic Tree Memory Representation for LLMs
Rezazadeh, Alireza
Li, Zichao
Wei, Wei
Bao, Yujia
Computation and Language
Artificial Intelligence
Machine Learning
Recent advancements in large language models have significantly improved their context windows, yet challenges in effective long-term memory management remain. We introduce MemTree, an algorithm that leverages a dynamic, tree-structured memory representation to optimize the organization, retrieval, and integration of information, akin to human cognitive schemas. MemTree organizes memory hierarchically, with each node encapsulating aggregated textual content, corresponding semantic embeddings, and varying abstraction levels across the tree's depths. Our algorithm dynamically adapts this memory structure by computing and comparing semantic embeddings of new and existing information to enrich the model's context-awareness. This approach allows MemTree to handle complex reasoning and extended interactions more effectively than traditional memory augmentation methods, which often rely on flat lookup tables. Evaluations on benchmarks for multi-turn dialogue understanding and document question answering show that MemTree significantly enhances performance in scenarios that demand structured memory management.
title From Isolated Conversations to Hierarchical Schemas: Dynamic Tree Memory Representation for LLMs
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2410.14052