HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Towards Unified Conversational Recommender Systems via Knowledge-Enhanced Prompt Learning

Xiaolei Wang Kun Zhou Ji-Rong Wen Wayne Xin Zhao

Towards Unified Conversational Recommender Systems via Knowledge-Enhanced Prompt Learning

Abstract

Conversational recommender systems (CRS) aim to proactively elicit user preference and recommend high-quality items through natural language conversations. Typically, a CRS consists of a recommendation module to predict preferred items for users and a conversation module to generate appropriate responses. To develop an effective CRS, it is essential to seamlessly integrate the two modules. Existing works either design semantic alignment strategies, or share knowledge resources and representations between the two modules. However, these approaches still rely on different architectures or techniques to develop the two modules, making it difficult for effective module integration. To address this problem, we propose a unified CRS model named UniCRS based on knowledge-enhanced prompt learning. Our approach unifies the recommendation and conversation subtasks into the prompt learning paradigm, and utilizes knowledge-enhanced prompts based on a fixed pre-trained language model (PLM) to fulfill both subtasks in a unified approach. In the prompt design, we include fused knowledge representations, task-specific soft tokens, and the dialogue context, which can provide sufficient contextual information to adapt the PLM for the CRS task. Besides, for the recommendation subtask, we also incorporate the generated response template as an important part of the prompt, to enhance the information interaction between the two subtasks. Extensive experiments on two public CRS datasets have demonstrated the effectiveness of our approach.

Code Repositories

rucaibox/unicrs
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
recommendation-systems-on-redialUniCRS
Recall@1: 0.051
Recall@10: 0.224
Recall@50: 0.428
text-generation-on-redialUniCRS
Distinct-2: 0.492
Distinct-3: 0.648
Distinct-4: 0.832

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Towards Unified Conversational Recommender Systems via Knowledge-Enhanced Prompt Learning | Papers | HyperAI