Feb, 2024

UniRetriever:多任务候选选择用于各种上下文自适应式会话检索

TL;DR我们提出了一种多任务框架,作为会话中三个主要检索任务(个人选择、知识选择和回复选择)的通用检索器,通过设计一个双编码器体系结构来检索相关上下文,通过点积方法检索适合的候选项,并引入两个损失约束来捕捉对话上下文和不同候选项之间的微妙关系。实验证明我们的模型在训练领域内外都能取得最先进的检索质量,揭示了其成为同时用于不同候选项选择任务的通用检索器的潜在潜力和泛化能力。