arXiv:2505.17670cs.LGcs.AI2025-05NeurIPS被引 18

用8个连续向量实现高效多模态记忆,让视觉语言模型更擅长复杂推理。

Towards General Continuous Memory for Vision-Language Models

  • 用视觉语言模型自身编码知识,生成8个紧凑的连续记忆向量。
  • 仅用1.2%参数和15.6K合成数据,即可实现高性能记忆模块。
  • 内存模块可即插即用,适合需要灵活增强推理能力的场景。

语言模型及其扩展——视觉语言模型(VLM)在各类任务中表现卓越,但在需要多模态或多语言现实知识的复杂推理任务上仍存在困难。为此,需要一个能高效提供相关多模态信息的外部记忆系统。现有方法通常将图像和文本标记拼接成长序列作为记忆,但会显著增加上下文长度,甚至降低性能。本文提出使用连续记忆——一组紧凑的稠密嵌入,以更高效地表示多模态与多语言知识。核心思路是利用VLM自身作为连续记忆编码器。实验证明该设计提升了复杂多模态推理任务的表现。在此基础上,我们提出一种数据高效且参数高效的微调方法,仅需1.2%模型参数和15.6K自合成样本,即可将VLM转化为记忆编码器。所提方法CoMEM利用VLM原有能力,将任意多模态与多语言知识编码为仅8个连续嵌入。由于推理时VLM保持冻结,该记忆模块可即插即用,灵活集成。在八个多模态推理基准上的实验验证了其有效性。

原文摘要 · Abstract (English)

Language models (LMs) and their extension, vision-language models (VLMs), have achieved remarkable performance across various tasks. However, they still struggle with complex reasoning tasks that require multimodal or multilingual real-world knowledge. To support such capabilities, an external memory system that can efficiently provide relevant multimodal information is essential. Existing approaches generally concatenate image and text tokens into a long sequence as memory, which, however, may drastically increase context length and even degrade performance. In contrast, we propose using continuous memory, a compact set of dense embeddings to more effectively and efficiently represent multimodal and multilingual knowledge. Our key insight is that a VLM can serve as its own continuous memory encoder. We empirically show that this design improves performance on complex multimodal reasoning tasks. Building on this, we introduce a data-efficient and parameter-efficient method to fine-tune the VLM into a memory encoder, requiring only 1.2% of the model's parameters and a small corpus of 15.6K self-synthesized samples. Our approach CoMEM utilizes VLM's original capabilities to encode arbitrary multimodal and multilingual knowledge into just 8 continuous embeddings. Since the inference-time VLM remains frozen, our memory module is plug-and-play and can be flexibly integrated as needed. Extensive experiments across eight multimodal reasoning benchmarks demonstrate the effectiveness of our approach.

视觉语言模型连续记忆多模态推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。