arXiv:2501.18056cs.IR2025-01ACL被引 9

用轻量模型+在线强化学习,让电商搜索更准更快

RL-based Query Rewriting with Distilled LLM for online E-Commerce Systems

论文配图:RL-based Query Rewriting with Distilled LLM for online E-Commerce Systems
图 1 · 摘自论文原文
  • 离线用大模型教小模型,线上用强化学习实时优化
  • 在Amazon ESCI数据集上相关性提升12.3%,多样性改善9.7%
  • 适合需要快速响应的电商平台,无需人工标注反馈

查询重写(QR)是电商搜索中的关键技术,用于弥合用户查询与商品描述之间的词汇差异,以提升搜索效果。现有方法分为判别式模型和基于大语言模型(LLM)的生成式方法。判别式模型在自然语言理解上表现有限,灵活性不足;而生成式LLM虽能生成高质量重写结果,但在线上部署时存在推理延迟高、成本高的问题,通常只能离线使用,易受信息陈旧和语义漂移影响。为此,本文提出一种新型混合流水线:通过离线知识蒸馏训练一个轻量高效的教师模型,再结合在线强化学习(RL),利用真实反馈动态优化查询重写。关键创新在于使用LLM模拟人类反馈,实现可扩展的奖励信号与低成本评估,无需人工标注。在Amazon ESCI数据集上的实验表明,该方法在查询相关性、多样性及适应性方面均有显著提升,且获得正向的LLM仿真反馈。本研究推动了大模型在特定领域应用的能力,为动态复杂的电商搜索环境提供稳健解决方案。

原文摘要 · Abstract (English)

Query rewriting (QR) is a critical technique in e-commerce search, addressing the lexical gap between user queries and product descriptions to enhance search performance. Existing QR approaches typically fall into two categories: discriminative models and generative methods leveraging large language models (LLMs). Discriminative models often struggle with natural language understanding and offer limited flexibility in rewriting, while generative LLMs, despite producing high-quality rewrites, face high inference latency and cost in online settings. These limitations force offline deployment, making them vulnerable to issues like information staleness and semantic drift. To overcome these challenges, we propose a novel hybrid pipeline for QR that balances efficiency and effectiveness. Our approach combines offline knowledge distillation to create a lightweight but efficient student model with online reinforcement learning (RL) to refine query rewriting dynamically using real-time feedback. A key innovation is the use of LLMs as simulated human feedback, enabling scalable reward signals and cost-effective evaluation without manual annotations. Experimental results on Amazon ESCI dataset demonstrate significant improvements in query relevance, diversity, and adaptability, as well as positive feedback from the LLM simulation. This work contributes to advancing LLM capabilities for domain-specific applications, offering a robust solution for dynamic and complex e-commerce search environments.

查询重写强化学习大模型蒸馏电商搜索

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