arXiv:2506.07785cs.CVcs.AI2025-06ICML被引 5

用树搜索重排推理上下文,让视觉语言大模型回答更准更稳

Re-ranking Reasoning Context with Tree Search Makes Large Vision-Language Models Stronger

  • 构建含推理模式的知识库,提升上下文质量
  • 用蒙特卡洛树搜索优先筛选相关示例,准确率显著提升
  • 适合想改进视觉问答模型的开发者和研究者

大型视觉语言模型(LVLMs)在视觉问答(VQA)任务中表现优异,但现有方法仍面临推理样本稀缺和检索内容波动等问题。本文提出多模态检索增强生成框架RCTS,通过自洽评估机制构建富含推理模式的知识库,并引入带启发式奖励的蒙特卡洛树搜索(MCTS-HR)对检索结果进行重排,从而引导模型使用高质量的上下文推理。大量实验表明,该框架在多个VQA数据集上达到当前最优性能,显著优于上下文学习(ICL)和原始检索增强生成(Vanilla-RAG)方法,验证了知识库与重排机制的有效性。代码已开源。

原文摘要 · Abstract (English)

Recent advancements in Large Vision Language Models (LVLMs) have significantly improved performance in Visual Question Answering (VQA) tasks through multimodal Retrieval-Augmented Generation (RAG). However, existing methods still face challenges, such as the scarcity of knowledge with reasoning examples and erratic responses from retrieved knowledge. To address these issues, in this study, we propose a multimodal RAG framework, termed RCTS, which enhances LVLMs by constructing a Reasoning Context-enriched knowledge base and a Tree Search re-ranking method. Specifically, we introduce a self-consistent evaluation mechanism to enrich the knowledge base with intrinsic reasoning patterns. We further propose a Monte Carlo Tree Search with Heuristic Rewards (MCTS-HR) to prioritize the most relevant examples. This ensures that LVLMs can leverage high-quality contextual reasoning for better and more consistent responses. Extensive experiments demonstrate that our framework achieves state-of-the-art performance on multiple VQA datasets, significantly outperforming In-Context Learning (ICL) and Vanilla-RAG methods. It highlights the effectiveness of our knowledge base and re-ranking method in improving LVLMs. Our code is available at https://github.com/yannqi/RCTS-RAG.

视觉问答多模态检索增强树搜索

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