提出分层检索的预训练微调方案,显著提升远距离文档召回率。
Hierarchical Retrieval: The Geometry and a Pretrain-Finetune Recipe
- 基于层级结构设计嵌入空间几何,证明维度需随层次深度线性增长
- 在WordNet上实现长距离匹配召回率从19%提升至76%
- 适合需要跨层级语义匹配的搜索系统与推荐场景
双编码器模型因结构简单和可扩展性强,在信息检索中广泛应用。然而,嵌入空间的欧几里得几何限制了其表达能力,可能影响性能。本文研究分层检索(HR)中的此类局限性,即文档集具有层级结构,查询匹配的是其所有祖先文档。我们首先证明,只要嵌入维度随层级深度线性增长、随文档数量对数增长,双编码器仍可有效用于HR。接着在标准检索设置下研究嵌入学习问题,发现存在‘远距离迷失’现象:层级越远的文档召回率越低。为此提出一种预训练-微调方案,显著提升远距离检索效果,同时不损害近邻文档性能。在真实层级结构WordNet上的实验显示,该方法将长距离匹配的召回率从19%提升至76%。最后,该方法在购物查询数据集上也提升了相关商品的检索效果。
原文摘要 · Abstract (English)
Dual encoder (DE) models, where a pair of matching query and document are embedded into similar vector representations, are widely used in information retrieval due to their simplicity and scalability. However, the Euclidean geometry of the embedding space limits the expressive power of DEs, which may compromise their quality. This paper investigates such limitations in the context of hierarchical retrieval (HR), where the document set has a hierarchical structure and the matching documents for a query are all of its ancestors. We first prove that DEs are feasible for HR as long as the embedding dimension is linear in the depth of the hierarchy and logarithmic in the number of documents. Then we study the problem of learning such embeddings in a standard retrieval setup where DEs are trained on samples of matching query and document pairs. Our experiments reveal a lost-in-the-long-distance phenomenon, where retrieval accuracy degrades for documents further away in the hierarchy. To address this, we introduce a pretrain-finetune recipe that significantly improves long-distance retrieval without sacrificing performance on closer documents. We experiment on a realistic hierarchy from WordNet for retrieving documents at various levels of abstraction, and show that pretrain-finetune boosts the recall on long-distance pairs from 19% to 76%. Finally, we demonstrate that our method improves retrieval of relevant products on a shopping queries dataset.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。