arXiv:2602.12735cs.CVcs.CL2026-02被引 3

用图结构管理视觉记忆,让AI更聪明地处理多模态长任务。

VimRAG: Navigating Massive Visual Context in Retrieval-Augmented Generation via Multimodal Memory Graph

  • 构建动态有向无环图,组织多模态推理状态与证据
  • 按节点位置智能分配高分辨率令牌,关键信息优先保留
  • 适合复杂多轮视觉问答、视频理解等长上下文任务

有效检索、推理和理解多模态信息仍是智能体系统的关键挑战。传统检索增强生成(RAG)方法依赖线性交互历史,在长上下文任务中表现不佳,尤其在迭代推理场景下面对信息稀疏但标记量大的视觉数据时。为此,我们提出VimRAG,一个面向文本、图像和视频的多模态检索增强推理框架。受系统性研究启发,我们将推理过程建模为动态有向无环图,以结构化方式组织代理状态与检索到的多模态证据。在此结构化记忆基础上,引入图调制视觉记忆编码机制,通过拓扑位置评估记忆节点重要性,实现对关键证据的高分辨率令牌动态分配,同时压缩或丢弃次要线索。为实现该范式,我们提出图引导策略优化方法,通过剪枝冗余动作关联的记忆节点,将步骤有效性与轨迹级奖励解耦,促进细粒度信用分配。大量实验表明,VimRAG在多个多模态RAG基准上持续达到最先进性能。代码已开源:https://github.com/Alibaba-NLP/VRAG。

原文摘要 · Abstract (English)

Effectively retrieving, reasoning, and understanding multimodal information remains a critical challenge for agentic systems. Traditional Retrieval-augmented Generation (RAG) methods rely on linear interaction histories, which struggle to handle long-context tasks, especially those involving information-sparse yet token-heavy visual data in iterative reasoning scenarios. To bridge this gap, we introduce VimRAG, a framework tailored for multimodal Retrieval-augmented Reasoning across text, images, and videos. Inspired by our systematic study, we model the reasoning process as a dynamic directed acyclic graph that structures the agent states and retrieved multimodal evidence. Building upon this structured memory, we introduce a Graph-Modulated Visual Memory Encoding mechanism, with which the significance of memory nodes is evaluated via their topological position, allowing the model to dynamically allocate high-resolution tokens to pivotal evidence while compressing or discarding trivial clues. To implement this paradigm, we propose a Graph-Guided Policy Optimization strategy. This strategy disentangles step-wise validity from trajectory-level rewards by pruning memory nodes associated with redundant actions, thereby facilitating fine-grained credit assignment. Extensive experiments demonstrate that VimRAG consistently achieves state-of-the-art performance on diverse multimodal RAG benchmarks. The code is available at https://github.com/Alibaba-NLP/VRAG.

多模态RAG图神经网络视觉推理

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