Knowledge Graph Embedding Based Collaborative Filtering

Yuhang Zhang

doi:10.1109/ACCESS.2020.3011105

Return

Knowledge Graph Embedding Based Collaborative Filtering

2020-01-01

Yuhang Zhang

Jun Wang *

Jie Luo *

DOI：10.1109/ACCESS.2020.3011105

Original

Save

View PDF

Abstract

En 中文

Along with the rapidly increasing massive online data, recommender systems have been used as an effective approach for filtering useful information, which have been widely adopted in many web applications. In recent years, new techniques such as neural network and deep learning have demonstrated its effectiveness in studies of recommender systems. In this paper, we propose a novel approach for collaborative filtering with implicit feedback based on knowledge graph embedding. The basic ideal is to model the interactions between users and items as an interaction knowledge graph with a single relation, whose vector representation is learned through knowledge graph embedding. Base on the learned representation, the collaborative filtering problem is converted into link prediction in the interaction knowledge graph. KGECF, a neural network for knowledge graph embedding based collaborative filtering, is proposed based on this ideal and the RotatE knowledge graph embedding model. Experimental results on five datasets with different characteristics show that the KGECF model achieves the state-of-the-art (SOTA) performance on all datasets. And unlike other SOTA models that have obvious performance drops on AMusic and AToy datasets, our model's performance is very stable across all datasets.

Keywords:

Collaboration

Recommender systems

Neural networks

Computational modeling

Measurement

Task analysis

Collaborative filtering

link prediction

knowledge graph embedding

recommender system

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.