让AI更懂图文混排文档,通过动态推理提升问答准确率
ViDoRAG: Visual Document Retrieval-Augmented Generation via Dynamic Iterative Reasoning Agents
- 用混合检索策略融合图文特征,提升信息定位精度
- 引入迭代式推理机制,使模型可多轮思考并优化答案
- 专为复杂图文文档设计,适合需要深度理解的场景
从视觉丰富的文档中理解信息仍是传统检索增强生成(RAG)方法的重大挑战。现有基准主要聚焦图像问答,忽视了密集图文文档中高效检索、理解与推理的核心难题。为此,我们提出ViDoSeek数据集,用于评估RAG在需复杂推理的视觉文档上的表现。基于此,我们发现当前RAG方法存在两大局限:(i) 纯视觉检索难以有效融合文本与视觉特征;(ii) 过去方法推理令牌分配不足,限制其效果。为此,我们提出ViDoRAG,一种面向视觉文档复杂推理的多智能体RAG框架。该框架采用基于高斯混合模型(GMM)的混合检索策略,有效处理多模态信息。为进一步激发模型推理能力,我们引入包含探索、总结与反思的迭代式智能体流程,构建测试时扩展的研究框架。在ViDoSeek上的大量实验验证了该方法的有效性与泛化能力。显著地,ViDoRAG在竞争性基准上超越现有方法超过10%。代码已开源:https://github.com/Alibaba-NLP/ViDoRAG。
原文摘要 · Abstract (English)
Understanding information from visually rich documents remains a significant challenge for traditional Retrieval-Augmented Generation (RAG) methods. Existing benchmarks predominantly focus on image-based question answering (QA), overlooking the fundamental challenges of efficient retrieval, comprehension, and reasoning within dense visual documents. To bridge this gap, we introduce ViDoSeek, a novel dataset designed to evaluate RAG performance on visually rich documents requiring complex reasoning. Based on it, we identify key limitations in current RAG approaches: (i) purely visual retrieval methods struggle to effectively integrate both textual and visual features, and (ii) previous approaches often allocate insufficient reasoning tokens, limiting their effectiveness. To address these challenges, we propose ViDoRAG, a novel multi-agent RAG framework tailored for complex reasoning across visual documents. ViDoRAG employs a Gaussian Mixture Model (GMM)-based hybrid strategy to effectively handle multi-modal retrieval. To further elicit the model's reasoning capabilities, we introduce an iterative agent workflow incorporating exploration, summarization, and reflection, providing a framework for investigating test-time scaling in RAG domains. Extensive experiments on ViDoSeek validate the effectiveness and generalization of our approach. Notably, ViDoRAG outperforms existing methods by over 10% on the competitive ViDoSeek benchmark. The code is available at https://github.com/Alibaba-NLP/ViDoRAG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。