虫洞记忆：用于跨对话检索的 Rubik's Cube
机器学习
2025-04-15 v5 人工智能
计算与语言
摘要
鉴于当前大型语言模型在跨对话间共享记忆方面的差距，本研究提出一种虫洞记忆模块（WMM），将记忆设计为可在不同对话之间任意检索的 Rubik's Cube。通过仿真实验，研究者基于 Python 环境构建了实验框架，并使用设置记忆屏障来模拟当前 LLM 对话之间记忆难以共享的局面。将 CoQA 开发数据集导入实验，并验证了 WMM 非线性索引和动态检索的跨对话记忆检索功能，同时对 Titans 与 MemGPT 记忆模块的能力进行了比较分析。实验结果表明，WMM 能够实现跨对话记忆检索，并在八项实验中展现出量化指标的稳定性。该工作为 LLM 记忆管理的优化提供了新的技术方法，并为未来实际应用提供了经验。
关键词
引用
@article{arxiv.2501.14846,
title = {Wormhole Memory: A Rubik's Cube for Cross-Dialogue Retrieval},
author = {Libo Wang},
journal= {arXiv preprint arXiv:2501.14846},
year = {2025}
}
备注
The experimental process and code have been uploaded to the Github repository, the link is: https://github.com/brucewang123456789/GeniusTrail/tree/main/Wormhole%20Memory%20Module