HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

KGAT: Knowledge Graph Attention Network for Recommendation

Xiang Wang; Xiangnan He; Yixin Cao; Meng Liu; Tat-Seng Chua

KGAT: Knowledge Graph Attention Network for Recommendation

Abstract

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.

Code Repositories

zju-dbl/metakg
pytorch
Mentioned in GitHub
Userepic/KGAT-pytorch-master
pytorch
Mentioned in GitHub
fumiyo0607/KGAN
tf
Mentioned in GitHub
LunaBlack/KGAT-pytorch
pytorch
Mentioned in GitHub
giantonia/KGATv2
tf
Mentioned in GitHub
jennyzhang0215/DGL-KGAT
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
link-prediction-on-movielens-25mKGAT
Hits@10: 0.8147
nDCG@10: 0.5236
link-prediction-on-yelpKGAT
HR@10: 0.8762
nDCG@10: 0.6136

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
KGAT: Knowledge Graph Attention Network for Recommendation | Papers | HyperAI