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

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Wang, Qiuchen, Wang, Shihang, Zeng, Yu, Zhang, Qiang, Zhang, Fanrui, Guo, Zhuoning, Zhang, Bosi, Huang, Wenxuan, Chen, Lin, Chen, Zehui, Xie, Pengjun, Ding, Ruixue
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910151761461248
author Wang, Qiuchen
Wang, Shihang
Zeng, Yu
Zhang, Qiang
Zhang, Fanrui
Guo, Zhuoning
Zhang, Bosi
Huang, Wenxuan
Chen, Lin
Chen, Zehui
Xie, Pengjun
Ding, Ruixue
author_facet Wang, Qiuchen
Wang, Shihang
Zeng, Yu
Zhang, Qiang
Zhang, Fanrui
Guo, Zhuoning
Zhang, Bosi
Huang, Wenxuan
Chen, Lin
Chen, Zehui
Xie, Pengjun
Ding, Ruixue
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2602_12735
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle VimRAG: Navigating Massive Visual Context in Retrieval-Augmented Generation via Multimodal Memory Graph
Wang, Qiuchen
Wang, Shihang
Zeng, Yu
Zhang, Qiang
Zhang, Fanrui
Guo, Zhuoning
Zhang, Bosi
Huang, Wenxuan
Chen, Lin
Chen, Zehui
Xie, Pengjun
Ding, Ruixue
Computer Vision and Pattern Recognition
Computation and Language
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.
title VimRAG: Navigating Massive Visual Context in Retrieval-Augmented Generation via Multimodal Memory Graph
topic Computer Vision and Pattern Recognition
Computation and Language
url https://arxiv.org/abs/2602.12735