arXiv:2603.00655cs.CV2026-03被引 1

用记忆模块增强视觉语言理解,让模型更好记住细节

Mema: Memory-Augmented Adapter for Enhanced Vision-Language Understanding

  • 设计可记忆的视觉适配器,动态融合多层视觉特征
  • 在多个基准上提升性能,显著改善细粒度视觉理解
  • 轻量插件式设计,适合快速集成到现有视觉模型

多模态大语言模型通过对齐预训练视觉表征与大语言模型中的语言知识,取得了显著进展。然而,现有方法通常依赖于最后一层的视觉特征或可学习的多层融合,缺乏显式的跨层交互设计,难以充分挖掘分层视觉线索。本文提出在视觉编码器中引入记忆增强适配器(Mema)。Mema维护一个状态化的记忆体,逐层累积视觉表征,并根据查询嵌入和逐步输入的视觉特征动态演化。部分记忆内容通过反馈机制有选择性地注入到令牌表示中,缓解浅层细粒度视觉线索的衰减问题。该模块轻量且即插即用,无需修改原始骨干网络结构,仅需极少新增参数训练,即可实现自适应视觉特征优化并降低训练开销。大量实验表明,Mema在多个基准上持续提升性能,验证了其在复杂多模态推理任务中的有效性。代码已开源:https://github.com/Sisiliu312/Mema。

原文摘要 · Abstract (English)

Multimodal Large Language Models (MLLMs) have achieved remarkable performance by aligning pretrained visual representations with the linguistic knowledge embedded in Large Language Models (LLMs). However, existing approaches typically rely on final-layer visual features or learnable multi-layer fusion, which often fail to sufficiently exploit hierarchical visual cues without explicit cross-layer interaction design. In this work, we propose a Memory-Augmented Adapter (Mema) within the vision encoder. Specifically, Mema maintains a stateful memory that accumulates hierarchical visual representations across layers, with its evolution conditioned on both query embeddings and step-wise visual features. A portion of this memory is selectively injected into token representations via a feedback mechanism, thereby mitigating the attenuation of fine-grained visual cues from shallow layers. Designed as a lightweight and plug-and-play module, Mema integrates seamlessly into pretrained vision encoders without modifying the vanilla backbone architecture. Only a minimal set of additional parameters requires training, enabling adaptive visual feature refinement while reducing training overhead. Extensive experiments across multiple benchmarks demonstrate that Mema consistently improves performance, validating its effectiveness in complex multimodal reasoning tasks. The code have been released at https://github.com/Sisiliu312/Mema.

视觉语言多模态记忆模块轻量适配

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