CPRM: A LLM-based Continual Pre-training Framework for Relevance Modeling in Commercial Search

Kaixin Wu,Yixin Ji,Zeyuan Chen,Qiang Wang,Cunxiang Wang,Hong Liu,Baijun Ji,Jia Xu,Zhongyi Liu,Jinjie Gu,Yuan Zhou,Linjian Mo
2024-12-02
Abstract:Relevance modeling between queries and items stands as a pivotal component in commercial search engines, directly affecting the user experience. Given the remarkable achievements of large language models (LLMs) in various natural language processing (NLP) tasks, LLM-based relevance modeling is gradually being adopted within industrial search systems. Nevertheless, foundational LLMs lack domain-specific knowledge and do not fully exploit the potential of in-context learning. Furthermore, structured item text remains underutilized, and there is a shortage in the supply of corresponding queries and background knowledge. We thereby propose CPRM (Continual Pre-training for Relevance Modeling), a framework designed for the continual pre-training of LLMs to address these issues. Our CPRM framework includes three modules: 1) employing both queries and multi-field item to jointly pre-train for enhancing domain knowledge, 2) applying in-context pre-training, a novel approach where LLMs are pre-trained on a sequence of related queries or items, and 3) conducting reading comprehension on items to produce associated domain knowledge and background information (e.g., generating summaries and corresponding queries) to further strengthen LLMs. Results on offline experiments and online A/B testing demonstrate that our model achieves convincing performance compared to strong baselines.
Artificial Intelligence,Computation and Language,Information Retrieval,Machine Learning
What problem does this paper attempt to address?
This paper attempts to solve the problem of query - item relevance modeling in commercial search engines. Specifically, the paper points out that although large - language models (LLMs) have achieved remarkable success in various natural - language - processing tasks, there are several challenges when applied to relevance modeling in commercial search: 1. **Lack of domain - specific knowledge**: Basic LLMs are pre - trained on a wide range of data sources, which do not pay special attention to specific domains, resulting in the models lacking domain - specific knowledge. 2. **Semantic gap**: Queries are usually in the form of short, colloquial texts, while items are usually expressed in more formal, long - text forms, which leads to a "semantic gap" between their representations. 3. **Task - agnostic pre - training**: The pre - training stage of LLMs is "task - agnostic", which hinders their direct connection with downstream tasks and excludes the possibility of context - pre - training enhancements customized for these tasks. 4. **Under - utilization of structured data**: Item data is highly structured and difficult to utilize, which prevents LLMs from fully realizing their potential. To address the above problems, the authors propose CPRM (Continual Pre - training for Relevance Modeling), a framework for continuously pre - training LLMs to improve relevance modeling. The CPRM framework consists of three modules: 1. **Domain Knowledge Enhancement (DKE)**: Use queries and multi - field items for joint pre - training to enhance domain knowledge. 2. **In - Context Pre - training (ICP)**: Pre - train LLMs by constructing a sequence of related queries or items, which is a novel method. 3. **Reading Comprehension Distillation (RCD)**: Perform reading comprehension on items to generate relevant domain knowledge and background information (such as generating summaries and corresponding queries), further strengthening the capabilities of LLMs. Through these methods, CPRM aims to improve the performance of LLMs in relevance modeling in commercial search. Experimental results show that CPRM performs well in both offline experiments and online A/B tests, significantly outperforming the baseline model.