用单个隐向量代替多模态证据,大幅降低问答系统资源消耗。
One Token per Multimodal Evidence: Latent Memory for Resource-Constrained QA

- 每条证据压缩为一个高维隐向量,实现统一隐空间检索与生成。
- 相比先进RAG方法,生成令牌消耗减少3至10倍,性能相当。
- 适合边缘设备或低算力场景的多模态问答应用。
外部记忆能有效将大语言模型(LLMs)和视觉语言模型(VLMs)驱动的问答系统锚定在相关多模态证据上。然而,现有记忆范式以原始文本或图像形式表示每条记忆项,导致基于检索的系统必须将检索到的文本或图像传递给生成模型,造成高令牌消耗和存储压力,难以应用于资源受限场景。本文提出隐空间记忆(Latent Memory),用小型压缩模型生成的单个高维隐向量替代每条原始文本或图像证据。查询嵌入该隐空间后,直接检索相关隐向量,并将其作为提示输入预训练的语言或视觉语言模型进行答案生成。为使每个隐向量同时支持重建、检索与生成,压缩器通过重建、对比和蒸馏目标在端到端框架下联合训练。在七个纯文本问答基准(如HotpotQA)和多模态问答基准上评估,其性能媲美先进RAG基线,生成令牌消耗减少3至10倍;在WebQA上实现了最强的图像引导问答表现。代码已公开于https://github.com/zz1358m/Latent-Memory-Master。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。