arXiv:2510.02745cs.CV2025-10NeurIPS被引 11

用推理驱动框架提升多模态检索的准确率与效率

Retrv-R1: A Reasoning-Driven MLLM Framework for Universal and Efficient Multimodal Retrieval

  • 引入分步推理与信息压缩模块,减少冗余计算
  • 在多个基准上达到最优性能,推理效率显著提升
  • 适合需要高精度多模态检索的研究与应用

DeepSeek-R1的成功展示了强化学习增强大模型推理能力的巨大潜力。本文提出Retrv-R1,首个专为多模态通用检索设计的R1风格多模态大模型,通过分步推理实现更精准的检索结果。直接套用DeepSeek-R1方法于检索任务不可行,主要因(1)多候选生成时需大量令牌进行推理,导致计算成本过高;(2)直接应用强化学习训练检索任务易产生不稳定与次优结果。为此,Retrv-R1引入信息压缩模块与细节检查机制,在降低令牌数量的同时保留关键信息。此外,提出新训练范式:先使用面向检索的合成思维链数据集进行激活阶段优化,再通过新型课程奖励进行强化学习,兼顾性能与效率。实验表明,该框架在多个基准和任务中均实现当前最优表现,兼具高效性与强泛化能力。

原文摘要 · Abstract (English)

The success of DeepSeek-R1 demonstrates the immense potential of using reinforcement learning (RL) to enhance LLMs' reasoning capabilities. This paper introduces Retrv-R1, the first R1-style MLLM specifically designed for multimodal universal retrieval, achieving higher performance by employing step-by-step reasoning to produce more accurate retrieval results. We find that directly applying the methods of DeepSeek-R1 to retrieval tasks is not feasible, mainly due to (1) the high computational cost caused by the large token consumption required for multiple candidates with reasoning processes, and (2) the instability and suboptimal results when directly applying RL to train for retrieval tasks. To address these issues, Retrv-R1 introduces an information compression module with a details inspection mechanism, which enhances computational efficiency by reducing the number of tokens while ensuring that critical information for challenging candidates is preserved. Furthermore, a new training paradigm is proposed, including an activation stage using a retrieval-tailored synthetic CoT dataset for more effective optimization, followed by RL with a novel curriculum reward to improve both performance and efficiency. Incorporating these novel designs, Retrv-R1 achieves SOTA performance, high efficiency, and strong generalization ability, as demonstrated by experiments across multiple benchmarks and tasks.

多模态检索推理驱动强化学习高效模型

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