arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.18409cs.IR

生成式检索遇上多级相关性

Generative Retrieval Meets Multi-Graded Relevance

Yubao Tang, Ruqing Zhang, Jiafeng Guo, Maarten de Rijke, Wei Chen, Xueqi Cheng

AI总结:

提出 GR$^2$ 框架,通过联合优化标识符的相关性与区别性并采用多级受限对比训练,将生成式检索扩展至多级相关性场景,有效提升了检索性能。

摘要:

生成式检索代表了一种新颖的信息检索方法。它使用编码器-解码器架构直接为查询生成相关文档标识符(docids)。虽然这种方法具有优势,但当前的方法仅限于二元相关性数据的场景,忽略了文档具有多级相关性的潜力。将生成式检索扩展以适应多级相关性带来了挑战,包括需要协调 docid 对的似然概率,以及多个相关文档共享相同标识符的可能性。为了应对这些挑战,我们引入了一个名为 GRaded Generative Retrieval (GR$^2$) 的框架。GR$^2$ 专注于两个关键组件:确保相关且不同的标识符,以及实施多级受限对比训练。首先,我们创建既在语义上相关又充分不同的标识符,以有效地表示单个文档。这是通过结合 docid 生成和自编码器模型,联合优化 docids 的相关性和区别性来实现的。其次,我们整合关于相关性等级之间关系的信息以指导训练过程。我们使用受限对比训练策略,根据各自的相关性等级,将查询的表示及其相关文档的标识符拉近。在具有多级和二元相关性的数据集上的大量实验证明了 GR$^2$ 的有效性。

英文摘要:

Generative retrieval represents a novel approach to information retrieval. It uses an encoder-decoder architecture to directly produce relevant document identifiers (docids) for queries. While this method offers benefits, current approaches are limited to scenarios with binary relevance data, overlooking the potential for documents to have multi-graded relevance. Extending generative retrieval to accommodate multi-graded relevance poses challenges, including the need to reconcile likelihood probabilities for docid pairs and the possibility of multiple relevant documents sharing the same identifier. To address these challenges, we introduce a framework called GRaded Generative Retrieval (GR$^2$). GR$^2$ focuses on two key components: ensuring relevant and distinct identifiers, and implementing multi-graded constrained contrastive training. First, we create identifiers that are both semantically relevant and sufficiently distinct to represent individual documents effectively. This is achieved by jointly optimizing the relevance and distinctness of docids through a combination of docid generation and autoencoder models. Second, we incorporate information about the relationship between relevance grades to guide the training process. We use a constrained contrastive training strategy to bring the representations of queries and the identifiers of their relevant documents closer together, based on their respective relevance grades. Extensive experiments on datasets with both multi-graded and binary relevance demonstrate the effectiveness of GR$^2$.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑