arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.02429cs.IRcs.LG
收藏

PEFA:用于大规模基于嵌入的检索模型的无参数适配器

PEFA: Parameter-Free Adapters for Large-scale Embedding-based Retrieval Models

Amazon(亚马逊公司);UCLA(加利福尼亚大学洛杉矶分校)

Wei-Cheng Chang, Jyun-Yu Jiang, Jiong Zhang, Mutasem Al-Darabsah, Choon Hui Teo, Cho-Jui Hsieh, Hsiang-Fu Yu, S. V. N. Vishwanathan

AI总结:

本文提出无参数适配器框架 PEFA,通过 kNN 与 ERM 评分凸组合实现免反向传播调优,并在文档检索和商品搜索中显著提升 Recall@100。

摘要:

基于嵌入的检索模型(ERMs)凭借强大的大语言模型,已成为解决大规模文本检索问题的一种很有前景的框架。然而,由于数据规模极大以及多阶段流水线(例如预训练、微调、蒸馏)的复杂性,微调 ERM 以达到最先进结果可能成本高昂。在本文中,我们提出 PEFA 框架,即 ParamEter-Free Adapters(无参数适配器),用于在优化过程中不进行任何反向传播的情况下快速调优 ERM。在索引构建阶段,PEFA 为 ERM 配备一个非参数化的 k 近邻(kNN)组件。在推理阶段,PEFA 对两个评分函数执行凸组合,其中一个来自 ERM,另一个来自 kNN。基于邻域定义,PEFA 框架产生两种实现,即使用双 ANN 索引的 PEFA-XL(即 extra large,超大)和使用单个 ANN 索引的 PEFA-XS(即 extra small,超小)。在实验上,PEFA 在两个检索应用上取得显著提升。对于文档检索,在 Recall@100 指标方面,PEFA 不仅将 Trivia-QA 上预训练 ERM 的结果平均提升 13.2%,还分别将 NQ-320K 上微调 ERM 的结果平均提升 5.5%。对于商品搜索,就 PEFA-XS 和 PEFA-XL 而言,PEFA 将微调 ERM 的 Recall@100 分别平均提升 5.3% 和 14.5%。我们的代码可在 https://github.com/amzn/pecos/tree/mainline/examples/pefa-wsdm24 获取。

英文摘要:

Embedding-based Retrieval Models (ERMs) have emerged as a promising framework for large-scale text retrieval problems due to powerful large language models. Nevertheless, fine-tuning ERMs to reach state-of-the-art results can be expensive due to the extreme scale of data as well as the complexity of multi-stages pipelines (e.g., pre-training, fine-tuning, distillation). In this work, we propose the PEFA framework, namely ParamEter-Free Adapters, for fast tuning of ERMs without any backward pass in the optimization. At index building stage, PEFA equips the ERM with a non-parametric k-nearest neighbor (kNN) component. At inference stage, PEFA performs a convex combination of two scoring functions, one from the ERM and the other from the kNN. Based on the neighborhood definition, PEFA framework induces two realizations, namely PEFA-XL (i.e., extra large) using double ANN indices and PEFA-XS (i.e., extra small) using a single ANN index. Empirically, PEFA achieves significant improvement on two retrieval applications. For document retrieval, regarding Recall@100 metric, PEFA improves not only pre-trained ERMs on Trivia-QA by an average of 13.2%, but also fine-tuned ERMs on NQ-320K by an average of 5.5%, respectively. For product search, PEFA improves the Recall@100 of the fine-tuned ERMs by an average of 5.3% and 14.5%, for PEFA-XS and PEFA-XL, respectively. Our code is available at https://github.com/amzn/pecos/tree/mainline/examples/pefa-wsdm24.

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

↑