用序列建模和知识图谱提升对话推荐的准确性
Knowledge-Enhanced Conversational Recommendation via Transformer-based Sequential Modelling
- 将对话中的物品与属性作为序列建模,捕捉用户偏好变化
- 在电影等场景中,相比现有方法推荐准确率显著提升
- 融合知识图谱的多跳路径,适合有丰富实体关系数据的领域
在对话式推荐系统中,对话通常涉及多个物品及其相关实体或属性(如电影的导演)。这些物品和实体随对话发展被陆续提及,形成潜在的序列依赖关系。然而,多数现有方法忽略了这种依赖。本文提出基于Transformer的序列对话推荐方法TSCR,通过物品与相关实体构建用户序列,利用填空任务预测推荐项。针对部分领域已存在物品与实体构成的知识图谱的情况,进一步提出TSCRKG:利用知识图谱对模型进行离线初始化,并将对话序列扩展为多跳路径以增强表示。实验表明,TSCR显著优于现有基线,TSCRKG在此基础上进一步提升性能。
原文摘要 · Abstract (English)
In conversational recommender systems (CRSs), conversations usually involve a set of items and item-related entities or attributes, e.g., director is a related entity of a movie. These items and item-related entities are often mentioned along the development of a dialog, leading to potential sequential dependencies among them. However, most of existing CRSs neglect these potential sequential dependencies. In this article, we first propose a Transformer-based sequential conversational recommendation method, named TSCR, to model the sequential dependencies in the conversations to improve CRS. In TSCR, we represent conversations by items and the item-related entities, and construct user sequences to discover user preferences by considering both the mentioned items and item-related entities. Based on the constructed sequences, we deploy a Cloze task to predict the recommended items along a sequence. Meanwhile, in certain domains, knowledge graphs formed by the items and their related entities are readily available, which provide various different kinds of associations among them. Given that TSCR does not benefit from such knowledge graphs, we then propose a knowledge graph enhanced version of TSCR, called TSCRKG. In specific, we leverage the knowledge graph to offline initialize our model TSCRKG, and augment the user sequence of conversations (i.e., sequence of the mentioned items and item-related entities in the conversation) with multi-hop paths in the knowledge graph. Experimental results demonstrate that our TSCR model significantly outperforms state-of-the-art baselines, and the enhanced version TSCRKG further improves recommendation performance on top of TSCR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。