Relevance Matters: A Multi-Task and Multi-Stage Large Language Model Approach for E-commerce Query Rewriting

📅 2026-03-02
📈 Citations: 0
✨ Influential: 0
📄 PDF
🤖 AI Summary
This work addresses the semantic gap between user queries and product descriptions in e-commerce search by proposing a multi-task, multi-stage query rewriting framework based on large language models. The approach uniquely integrates explicit relevance modeling into the rewriting process, combining supervised fine-tuning (SFT) with Group Relative Policy Optimization (GRPO)—a reinforcement learning algorithm tailored to business objectives—to jointly optimize query rewriting, relevance estimation, and user conversion. Experiments leveraging JD.com’s pretrained large language model demonstrate significant improvements in both offline relevance metrics and online user conversion rate (UCVR) in A/B tests. The method has been deployed on JD.com’s search platform since August 2025.

Technology Category

Search and Optimization: Learning to SearchMachine Learning: Large Multimodal Models (LMMs)Natural Language Processing: Learning & Optimization for NLP

Application Category

Search and Retrieval-Augmented AI: Ad search and search for Web retailEconomics, Online Markets and Human Computation: LLM based quality controls for crowd workUser Modeling, Personalization and Recommendation: ML for personalized search and recommendations
📝 Abstract
For e-commerce search, user experience is measured by users' behavioral responses to returned products, like click-through rate and conversion rate, as well as the relevance between returned products and search queries. Consequently, relevance and user conversion constitute the two primary objectives in query rewriting, a strategy to bridge the lexical gap between user expressions and product descriptions. This research proposes a multi-task and multi-stage query rewriting framework grounded in large language models (LLMs). Critically, in contrast to previous works that primarily emphasized rewritten query generation, we inject the relevance task into query rewriting. Specifically, leveraging a pretrained model on user data and product information from JD.com, the approach initiates with multi-task supervised fine-tuning (SFT) comprising of the rewritten query generation task and the relevance tagging task between queries and rewrites. Subsequently, we employ Group Relative Policy Optimization (GRPO) for the model's objective alignment oriented toward enhancing the relevance and stimulating user conversions. Through offline evaluation and online A/B test, our framework illustrates substantial improvements in the effectiveness of e-commerce query rewriting, resulting in elevating the search results' relevance and boosting the number of purchases made per user (UCVR). Since August 2025, our approach has been implemented on JD.com, one of China's leading online shopping platforms.
Problem

Research questions and friction points this paper is trying to address.

query rewriting
relevance
e-commerce search
lexical gap
user conversion
Innovation

Methods, ideas, or system contributions that make the work stand out.

multi-task learning
query rewriting
relevance modeling
large language models
Group Relative Policy Optimization
🔎 Similar Papers
No similar papers found.
A
Aijun Dai
JD.com, Beijing, China
J
Jixiang Zhang
Tsinghua University
H
Haiqing Hu
JD.com, Beijing, China
G
Guoyu Tang
JD.com, Beijing, China
L
Lin Liu
JD.com, Beijing, China
Z
Ziguang Cheng
JD.com, Beijing, China