Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

KGAT: Knowledge Graph Attention Network for Recommendation

About

To provide more accurate, diverse, and explainable recommendation, it is compulsory to go beyond modeling user-item interactions and take side information into account. Traditional methods like factorization machine (FM) cast it as a supervised learning problem, which assumes each interaction as an independent instance with side information encoded. Due to the overlook of the relations among instances or items (e.g., the director of a movie is also an actor of another movie), these methods are insufficient to distill the collaborative signal from the collective behaviors of users. In this work, we investigate the utility of knowledge graph (KG), which breaks down the independent interaction assumption by linking items with their attributes. We argue that in such a hybrid structure of KG and user-item graph, high-order relations --- which connect two items with one or multiple linked attributes --- are an essential factor for successful recommendation. We propose a new method named Knowledge Graph Attention Network (KGAT) which explicitly models the high-order connectivities in KG in an end-to-end fashion. It recursively propagates the embeddings from a node's neighbors (which can be users, items, or attributes) to refine the node's embedding, and employs an attention mechanism to discriminate the importance of the neighbors. Our KGAT is conceptually advantageous to existing KG-based recommendation methods, which either exploit high-order relations by extracting paths or implicitly modeling them with regularization. Empirical results on three public benchmarks show that KGAT significantly outperforms state-of-the-art methods like Neural FM and RippleNet. Further studies verify the efficacy of embedding propagation for high-order relation modeling and the interpretability benefits brought by the attention mechanism.

Xiang Wang, Xiangnan He, Yixin Cao, Meng Liu, Tat-Seng Chua• 2019

Related benchmarks

TaskDatasetResultRank
RecommendationAmazon-Book (test)
Recall@200.1489
101
RecommendationYelp 2018 (test)
Recall@207.12
90
Sequential RecommendationMovieLens
ValidRatio1
41
Item RecommendationYelp (test)
HR@1087.62
21
List-wise RecommendationLastFM
Hit Ratio@129.18
16
Pair-wise RecommendationLastFM
Hit Ratio@175.89
16
RecommendationYelp 2018
Recall0.0705
15
RecommendationMovieLens small (test)
Hit Rate@1052.14
15
RecommendationMovieLens-25M (test)
HR@100.8147
15
RecommendationAlibaba-iFashion--
11
Showing 10 of 16 rows

Other info

Code

Follow for update