Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

One Token per Multimodal Evidence: Latent Memory for Resource-Constrained QA

About

External memory effectively grounds large language models (LLMs) and vision-language models (VLMs)-based question answering (QA) in relevant multimodal evidence. However, existing memory paradigms represent each memory item in raw text and image forms, so retrieval-based systems must pass the retrieved text or images to the generation LLMs/VLMs, resulting in high token consumption and storage pressure, making it unaffordable for resource-constrained applications. We propose Latent Memory, a latent-space memory paradigm that replaces each raw text or image evidence item with a single high-dimensional latent token produced by a small compressor LLM/VLM. Rather than retrieving raw evidence for generation, Latent Memory operates in a unified latent representation space: the query is embedded into this space to retrieve relevant latent tokens, and the retrieved latent tokens are directly prompted to a pretrained LLM or VLM for answer generation. To make each latent token simultaneously informative for reconstruction, retrieval, and generation, we train the compressor with reconstruction, contrastive, and distillation objectives in a unified end-to-end manner. Latent Memory is evaluated on seven text-only QA benchmarks (e.g., HotpotQA) and multimodal QA benchmarks, where it achieves competitive QA performance compared to advanced RAG baselines while consuming 3x to 10x fewer generator tokens. It can also deliver the strongest image-grounded QA performance on WebQA. Code is available at https://github.com/zz1358m/Latent-Memory-Master.

Zhi Zheng, Ziqiao Meng, Hao Luan, Wei Liu, Wee Sun Lee• 2026

Related benchmarks

TaskDatasetResultRank
Open-domain Question AnsweringNQ--
74
Multimodal Question AnsweringWebQA Average
F1 Score50
32
Text-based Question AnsweringMuSiQue
EM3.6
21
Text-based Question AnsweringHotpotQA
Exact Match (EM)17.5
21
Text-based Question Answering2WikiMultihopQA
EM2.9
21
Claim VerificationWiCE
Exact Match60.3
16
Multimodal Question AnsweringWebQA Image-grounded n=2,511 (test)
F1 Score69.4
16
Text-based Question Answering2WikiMultihopQA (val)
EM24.3
16
Text-based Question Answering2WikiMultihopQA & MuSiQue Out-of-domain Average (val)
EM16.5
16
Multimodal Question AnsweringWebQA Image
F1 Score11.7
16
Showing 10 of 17 rows

Other info

GitHub

Follow for update