用强化学习动态分配检索资源,让文档问答更关注关键信息。
MAB-DQA: Addressing Query Aspect Importance in Document Question Answering with Multi-Armed Bandits

- 将查询拆解为多个子查询,用多臂老虎机模型评估各部分重要性。
- 在4个数据集上比顶尖方法平均提升5%-18%,显著改善文档理解。
- 适合需要精准定位复杂文档信息的研究者与开发者。
文档问答(DQA)需基于用户查询从文档中生成答案,是文档理解的关键任务。该任务需解析视觉版式,促使近期研究采用多模态检索增强生成(RAG)来处理页面图像以生成答案。然而,多模态RAG在视觉DQA中难以有效利用大量图像,因检索阶段通常仅保留少数候选页面(如前4页),导致信息丰富但视觉不显著的内容被忽略,而常见但低信息量的页面占据优势。为此,我们提出基于多臂老虎机的DQA框架(MAB-DQA),显式建模查询中多个隐含方面的重要性。具体地,MAB-DQA将查询分解为感知方面的子查询,并为每个子查询检索特定候选集;将每个子查询视为一个“臂”,利用少量代表性页面的初步推理结果作为奖励信号,估算各方面的效用。在探索-利用策略引导下,动态调整检索预算至高价值方面。结合最具信息量的页面及其关联,生成预期答案。在四个基准测试上,MAB-DQA相比最先进方法平均提升5%-18%,持续增强文档理解能力。代码已公开于 https://github.com/ElephantOH/MAB-DQA。
原文摘要 · Abstract (English)
Document Question Answering (DQA) involves generating answers from a document based on a user's query, representing a key task in document understanding. This task requires interpreting visual layouts, which has prompted recent studies to adopt multimodal Retrieval-Augmented Generation (RAG) that processes page images for answer generation. However, in multimodal RAG, visual DQA struggles to utilize a large number of images effectively, as the retrieval stage often retains only a few candidate pages (e.g., Top-4), causing informative but less visually salient content to be overlooked in favor of common yet low-information pages. To address this issue, we propose a Multi-Armed Bandit-based DQA framework (MAB-DQA) to explicitly model the varying importance of multiple implicit aspects in a query. Specifically, MAB-DQA decomposes a query into aspect-aware subqueries and retrieves an aspect-specific candidate set for each. It treats each subquery as an arm and uses preliminary reasoning results from a small number of representative pages as reward signals to estimate aspect utility. Guided by an exploration-exploitation policy, MAB-DQA dynamically reallocates retrieval budgets toward high-value aspects. With the most informative pages and their correlations, MAB-DQA generates the expected results. On four benchmarks, MAB-DQA shows an average improvement of 5%-18% over the state-of-the-art method, consistently enhancing document understanding. Codes are available at https://github.com/ElephantOH/MAB-DQA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。