AssoMem: Scalable Memory QA with Multi-Signal Associative Retrieval

Kai Zhang, Xinyuan Zhang, Ejaz Ahmed, Hongda Jiang, Caleb Kumar +9 more
10/12/2025
cs.CL

No official code found. Best implementation: facebookresearch/AssoMem (11 stars, last commit 2026-03-27).

Abstract

Accurate recall from large scale memories remains a core challenge for memory augmented AI assistants performing question answering (QA), especially in similarity dense scenarios where existing methods mainly rely on semantic distance to the query for retrieval. Inspired by how humans link information associatively, we propose AssoMem, a novel framework constructing an associative memory graph that anchors dialogue utterances to automatically extracted clues. This structure provides a rich organizational view of the conversational context and facilitates importance aware ranking. Further, AssoMem integrates multi-dimensional retrieval signals-relevance, importance, and temporal alignment using an adaptive mutual information (MI) driven fusion strategy. Extensive experiments across three benchmarks and a newly introduced dataset, MeetingQA, demonstrate that AssoMem consistently outperforms SOTA baselines, verifying its superiority in context-aware memory recall.

View on arXivView PDF

Code Implementations(1)

Source implementation and dataset for paper "AssoMem Scalable Memory QA with Multi-Signal Associative Retrieval"

111Shell, PythonOct 10, 20256 months ago

Similar Papers

Cite this paper

@article{zhang2025assomem,
  title  = {AssoMem: Scalable Memory QA with Multi-Signal Associative Retrieval},
  author = {Kai Zhang and Xinyuan Zhang and Ejaz Ahmed and Hongda Jiang and Caleb Kumar and Kai Sun and Zhaojiang Lin and Sanat Sharma and Shereen Oraby and Aaron Colak and Ahmed Aly and Anuj Kumar and Xiaozhong Liu and Xin Luna Dong},
  year   = {2025},
  eprint = {2510.10397},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2510.10397v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2510.10397.svg)](https://codeofpaper.com/paper/2510.10397)

Discussion