3 个月前

基于插值与扰动的文档表示增强方法用于稠密检索

基于插值与扰动的文档表示增强方法用于稠密检索

摘要

稠密检索模型旨在在稠密表示空间中为输入查询检索最相关的文档,因其卓越的表现而受到广泛关注。然而,稠密模型通常需要大量标注的训练数据才能达到优异性能,而人工标注的查询-文档配对数据往往难以获取。为解决这一问题,我们提出了一种简单但高效的稠密检索文档增强框架——DAR(Document Augmentation for dense Retrieval),该框架通过文档表示的插值与扰动来增强其表达。我们在两个基准数据集上对DAR的检索性能进行了验证,结果表明,所提出的DAR在有标注与无标注文档的稠密检索任务中,均显著优于现有基线方法。

代码仓库

starsuzi/dar
官方
pytorch
GitHub 中提及

基准测试

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于插值与扰动的文档表示增强方法用于稠密检索 | 论文 | HyperAI超神经