arXiv:2504.05181cs.IRcs.AI2025-04被引 16

用直接排序优化替代强化学习,提升生成式检索效果

Lightweight and Direct Document Relevance Optimization for Generative Information Retrieval

  • 通过成对排序直接优化文档相关性,避免逐标记预测偏差
  • MS MARCO上MRR@10提升7.4%,Natural Questions提升19.9%
  • 无需奖励建模与强化微调,简化流程且更稳定

生成式信息检索(GenIR)将文档检索建模为文档标识符(docid)生成任务,实现端到端的全局优化。然而现有模型存在标记级错位问题:训练目标为预测下一个标记,难以有效捕捉文档级相关性。尽管基于强化学习的反馈方法(如RLRF)尝试通过奖励建模解决此问题,但需优化辅助奖励函数并进行强化微调,计算成本高且常不稳定。为此,我们提出直接文档相关性优化(DDRO),通过成对排序直接对齐标记级生成与文档级相关性估计,无需显式奖励建模和强化学习。在MS MARCO document和Natural Questions等基准数据集上的实验表明,DDRO优于强化学习方法,在MS MARCO上实现MRR@10提升7.4%,在Natural Questions上提升19.9%。结果证明,该方法以简化优化流程显著提升了检索效果,为生成式检索提供了一种高效可行的替代方案。

原文摘要 · Abstract (English)

Generative information retrieval (GenIR) is a promising neural retrieval paradigm that formulates document retrieval as a document identifier (docid) generation task, allowing for end-to-end optimization toward a unified global retrieval objective. However, existing GenIR models suffer from token-level misalignment, where models trained to predict the next token often fail to capture document-level relevance effectively. While reinforcement learning-based methods, such as reinforcement learning from relevance feedback (RLRF), aim to address this misalignment through reward modeling, they introduce significant complexity, requiring the optimization of an auxiliary reward function followed by reinforcement fine-tuning, which is computationally expensive and often unstable. To address these challenges, we propose direct document relevance optimization (DDRO), which aligns token-level docid generation with document-level relevance estimation through direct optimization via pairwise ranking, eliminating the need for explicit reward modeling and reinforcement learning. Experimental results on benchmark datasets, including MS MARCO document and Natural Questions, show that DDRO outperforms reinforcement learning-based methods, achieving a 7.4% improvement in MRR@10 for MS MARCO and a 19.9% improvement for Natural Questions. These findings highlight DDRO's potential to enhance retrieval effectiveness with a simplified optimization approach. By framing alignment as a direct optimization problem, DDRO simplifies the ranking optimization pipeline of GenIR models while offering a viable alternative to reinforcement learning-based methods.

生成式检索相关性优化轻量级

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