arXiv:2505.01946cs.IR2025-05被引 7

优化双塔模型,提升电商长尾搜索的精准匹配与转化

Embedding based retrieval for long tail search queries in ecommerce

  • 用大语言模型增强稀疏点击信号,改善长尾查询表示
  • 在贝斯特百汇商品数据上预训练通用模型,提升语义理解能力
  • 融合查询-查询与查询-商品对微调,效果优于传统方法

本文针对贝斯特百汇网站搜索中常见的帕累托分布现象——少数高频查询占据多数流量,而大量低频长尾查询缺乏交互信号的问题,提出一系列针对双塔模型架构的优化。通过引入大语言模型增强稀疏转化信号,利用贝斯特百汇商品数据预训练通用Transformer模型,并在微调阶段结合查询-查询对与查询-商品对进行联合训练。此外,还探索了多模型权重融合策略以提升评估指标。最终构建的召回机制在在线A/B测试中使转化率提升3%。同时提供一套带人工反馈的评估数据集构建方案,用于持续监控模型表现。

原文摘要 · Abstract (English)

In this abstract we present a series of optimizations we performed on the two-tower model architecture [14], and training and evaluation datasets to implement semantic product search at Best Buy. Search queries on bestbuy.com follow the pareto distribution whereby a minority of them account for most searches. This leaves us with a long tail of search queries that have low frequency of issuance. The queries in the long tail suffer from very spare interaction signals. Our current work focuses on building a model to serve the long tail queries. We present a series of optimizations we have done to this model to maximize conversion for the purpose of retrieval from the catalog. The first optimization we present is using a large language model to improve the sparsity of conversion signals. The second optimization is pretraining an off-the-shelf transformer-based model on the Best Buy catalog data. The third optimization we present is on the finetuning front. We use query-to-query pairs in addition to query-to-product pairs and combining the above strategies for finetuning the model. We also demonstrate how merging the weights of these finetuned models improves the evaluation metrics. Finally, we provide a recipe for curating an evaluation dataset for continuous monitoring of model performance with human-in-the-loop evaluation. We found that adding this recall mechanism to our current term match-based recall improved conversion by 3% in an online A/B test.

电商搜索长尾查询双塔模型语义召回

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。