AssoMem: Scalable Memory QA with Multi-Signal Associative Retrieval
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 state-of-the-art baselines, verifying its superiority in context-aware memory recall.
Cite
Text
Zhang et al. "AssoMem: Scalable Memory QA with Multi-Signal Associative Retrieval." International Conference on Learning Representations, 2026.Markdown
[Zhang et al. "AssoMem: Scalable Memory QA with Multi-Signal Associative Retrieval." International Conference on Learning Representations, 2026.](https://mlanthology.org/iclr/2026/zhang2026iclr-assomem/)BibTeX
@inproceedings{zhang2026iclr-assomem,
title = {{AssoMem: Scalable Memory QA with Multi-Signal Associative Retrieval}},
author = {Zhang, Kai and Zhang, Xinyuan and Ahmed, Ejaz and Jiang, Hongda and Kumar, Caleb and Sun, Kai and Lin, Zhaojiang and Sharma, Sanat and Oraby, Shereen and Colak, Aaron and Aly, Ahmed A and Kumar, Anuj and Liu, Xiaozhong and Dong, Xin Luna},
booktitle = {International Conference on Learning Representations},
year = {2026},
url = {https://mlanthology.org/iclr/2026/zhang2026iclr-assomem/}
}