Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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

About

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.

Qiuchen Wang, Shihang Wang, Yu Zeng, Qiang Zhang, Fanrui Zhang, Zhuoning Guo, Bosi Zhang, Wenxuan Huang, Lin Chen, Zehui Chen, Pengjun Xie, Ruixue Ding• 2026

Related benchmarks

TaskDatasetResultRank
Video Question AnsweringLVBench
Accuracy24.5
50
Document Visual Question AnsweringSlideVQA
Accuracy0.624
30
Multimodal Document Question AnsweringMMLongBench
Accuracy33.4
19
Cross-video UnderstandingXVBench
Accuracy37.1
14
General Text Question AnsweringHotpotQA
Accuracy79.1
14
General Text Question AnsweringSQuAD
Accuracy76.4
14
Image-based Question AnsweringWebQA
Accuracy53.9
14
Multimodal RAGVimRAG Overall
Overall Accuracy50.1
14
Video Corpus UnderstandingWikiHowQA
Accuracy29.7
14
Video Corpus UnderstandingSyntheticQA
Accuracy54.5
14
Showing 10 of 10 rows

Other info

Follow for update