The Differentiable Search Index (DSI) is a novel information retrieval (IR) framework that utilizes a differentiable function to generate a sorted list of document identifiers in response to a given query. However, due to the black-box nature of the end-to-end neural architecture, it remains to be understood to what extent DSI possesses the basic indexing and retrieval abilities. To mitigate this gap, in this study, we define and examine three important abilities that a functioning IR framework should possess, namely, exclusivity, completeness, and relevance ordering. Our analytical experimentation shows that while DSI demonstrates proficiency in memorizing the unidirectional mapping from pseudo queries to document identifiers, it falls short in distinguishing relevant documents from random ones, thereby negatively impacting its retrieval effectiveness. To address this issue, we propose a multi-task distillation approach to enhance the retrieval quality without altering the structure of the model and successfully endow it with improved indexing abilities. Through experiments conducted on various datasets, we demonstrate that our proposed method outperforms previous DSI baselines.
翻译:可微搜索索引(Differentiable Search Index, DSI)是一种新型信息检索(IR)框架,它利用可微函数生成针对给定查询的文档标识符排序列表。然而,由于端到端神经架构的黑箱特性,DSI在多大程度上具备基本的索引和检索能力仍有待理解。为弥补这一不足,本研究定义并考察了功能性信息检索框架应具备的三项重要能力,即排他性、完整性和相关性排序。我们的分析性实验表明,虽然DSI在记忆从伪查询到文档标识符的单向映射方面表现出色,但在区分相关文档与随机文档方面存在不足,从而对其检索效果产生负面影响。为解决这一问题,我们提出了一种多任务蒸馏方法,在不改变模型结构的情况下提升检索质量,并成功赋予其改进的索引能力。通过在多个数据集上的实验,我们证明了所提方法优于以往的DSI基线方法。