arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2410.20163cs.IRcs.CL
收藏

UniHGKR:统一的指令感知异构知识检索器

UniHGKR: Unified Instruction-aware Heterogeneous Knowledge Retrievers

Southeast University(东南大学);Stony Brook University(石溪大学);Virginia Tech(弗吉尼亚理工大学);UC Davis(加州大学戴维斯分校)

Dehai Min, Zhiyang Xu, Guilin Qi, Lifu Huang, Chenyu You

AI总结:

针对现有信息检索模型假设知识源和查询同构的局限,提出统一指令感知异构知识检索器UniHGKR,构建统一检索空间并遵循指令检索指定类型知识,在CompMix-IR基准和ConvMix任务上取得显著提升。

摘要:

现有信息检索(IR)模型通常假设知识源和用户查询具有同构结构,这限制了它们在检索本质上具有异构性和多样性的现实场景中的适用性。在本文中,我们介绍了UniHGKR,一种统一的指令感知异构知识检索器,它(1)为异构知识构建统一的检索空间,(2)遵循多样化的用户指令检索指定类型的知识。UniHGKR包括三个主要阶段:异构自监督预训练、文本锚定嵌入对齐和指令感知检索器微调,使其能够在各种检索上下文中泛化。该框架具有高度可扩展性,包含基于BERT的版本和在大型语言模型上训练的UniHGKR-7B版本。此外,我们引入了CompMix-IR,首个原生异构知识检索基准,它包含两个带有各种指令的检索场景、超过9400个问答(QA)对以及一个涵盖四种不同数据类型的1000万条目语料库。大量实验表明,UniHGKR在CompMix-IR上持续优于最先进的方法,在两个场景中分别实现了高达6.36%和54.23%的相对提升。最后,通过将我们的检索器应用于开放域异构QA系统,我们在流行的ConvMix任务上取得了新的最先进结果,绝对提升高达5.90分。

英文摘要:

Existing information retrieval (IR) models often assume a homogeneous structure for knowledge sources and user queries, limiting their applicability in real-world settings where retrieval is inherently heterogeneous and diverse. In this paper, we introduce UniHGKR, a unified instruction-aware heterogeneous knowledge retriever that (1) builds a unified retrieval space for heterogeneous knowledge and (2) follows diverse user instructions to retrieve knowledge of specified types. UniHGKR consists of three principal stages: heterogeneous self-supervised pretraining, text-anchored embedding alignment, and instruction-aware retriever fine-tuning, enabling it to generalize across varied retrieval contexts. This framework is highly scalable, with a BERT-based version and a UniHGKR-7B version trained on large language models. Also, we introduce CompMix-IR, the first native heterogeneous knowledge retrieval benchmark. It includes two retrieval scenarios with various instructions, over 9,400 question-answer (QA) pairs, and a corpus of 10 million entries, covering four different types of data. Extensive experiments show that UniHGKR consistently outperforms state-of-the-art methods on CompMix-IR, achieving up to 6.36% and 54.23% relative improvements in two scenarios, respectively. Finally, by equipping our retriever for open-domain heterogeneous QA systems, we achieve a new state-of-the-art result on the popular ConvMix task, with an absolute improvement of up to 5.90 points.

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

↑