HyperAIHyperAI

Command Palette

Search for a command to run...

通过知识增强的提示学习构建统一的对话推荐系统

Xiaolei Wang Kun Zhou Ji-Rong Wen Wayne Xin Zhao

摘要

对话式推荐系统(Conversational Recommender Systems, CRS)旨在通过自然语言对话主动挖掘用户偏好,并推荐高质量的项目。典型的CRS由两个核心模块构成:推荐模块用于预测用户偏好的项目,对话模块用于生成恰当的回应。为了构建高效的CRS,实现两个模块的无缝集成至关重要。现有方法通常采用语义对齐策略,或在两个模块之间共享知识资源与表示。然而,这些方法仍依赖于不同的架构或技术来分别构建两个模块,导致模块间难以实现高效融合。为解决上述问题,本文提出一种基于知识增强提示学习(knowledge-enhanced prompt learning)的统一CRS模型——UniCRS。该方法将推荐与对话两个子任务统一纳入提示学习范式,并基于固定的预训练语言模型(Pre-trained Language Model, PLM),利用知识增强型提示来统一完成两项任务。在提示设计中,我们融合了知识表示、任务特定的软标记(soft tokens)以及对话上下文,从而为PLM提供充分的上下文信息,以适配CRS任务。此外,在推荐子任务中,我们还将生成的回复模板作为提示的重要组成部分,以增强两个子任务之间的信息交互。在两个公开的CRS数据集上进行的大量实验表明,所提方法在性能上具有显著优势,验证了其有效性。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供