Knowledge graphs are used to alleviate the problems of data sparsity and cold starts in recommendation systems. However, most existing approaches ignore the hierarchical structure of the knowledge graph. In this paper, we propose a box embedding method for knowledge graph-enhanced recommendation system. Specifically, the box embedding represents not only the interaction between the user and the item, but also the head entity, the tail entity and the relation between them in the knowledge graph. Then the interaction between the item and the corresponding entity is calculated by the multi-task attention unit. Experimental results show that our method provides a large improvement over previous models in terms of Area Under Curve (AUC) and accuracy in publicly available recommendation datasets with three different domains.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Knowledge Graph-Enhanced Recommendation with Box Embeddings

  • Qiuyu Liang,
  • Weihua Wang,
  • Lei Lv,
  • Feilong Bao

摘要

Knowledge graphs are used to alleviate the problems of data sparsity and cold starts in recommendation systems. However, most existing approaches ignore the hierarchical structure of the knowledge graph. In this paper, we propose a box embedding method for knowledge graph-enhanced recommendation system. Specifically, the box embedding represents not only the interaction between the user and the item, but also the head entity, the tail entity and the relation between them in the knowledge graph. Then the interaction between the item and the corresponding entity is calculated by the multi-task attention unit. Experimental results show that our method provides a large improvement over previous models in terms of Area Under Curve (AUC) and accuracy in publicly available recommendation datasets with three different domains.