HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Learning Heterogeneous Knowledge Base Embeddings for Explainable Recommendation

Qingyao Ai; Vahid Azizi; Xu Chen; Yongfeng Zhang

Learning Heterogeneous Knowledge Base Embeddings for Explainable Recommendation

Abstract

Providing model-generated explanations in recommender systems is important to user experience. State-of-the-art recommendation algorithms - especially collaborative filtering (CF)-based approaches with shallow or deep models - usually work with various unstructured information sources for recommendation, such as textual reviews, visual images, and various implicit or explicit feedbacks. Though structured knowledge bases were considered in content-based approaches, they have been largely ignored recently due to the research focus on CF approaches. However, structured knowledge exhibit unique advantages in personalized recommendation systems. When the explicit knowledge about users and items is considered for recommendation, the system could provide highly customized recommendations based on users' historical behaviors and the knowledge is helpful for providing informed explanations regarding the recommended items. A great challenge for using knowledge bases for recommendation is how to integrate large-scale structured data, while taking advantage of collaborative filtering for highly accurate performance. Recent achievements in knowledge-base embedding (KBE) sheds light on this problem, which makes it possible to learn user and item representations while preserving the structure of their relationship with external knowledge for explanation. In this work, we propose to explain knowledge-base embeddings for explainable recommendation. Specifically, we propose a knowledge-base representation learning framework to embed heterogeneous entities for recommendation, and based on the embedded knowledge base, a soft matching algorithm is proposed to generate personalized explanations for the recommended items. Experimental results on real-world e-commerce datasets verified the superior recommendation performance and the explainability power of our approach compared with state-of-the-art baselines.

Code Repositories

fumiyo0607/KGAN
tf
Mentioned in GitHub
melissakou/KG-RecSys
Mentioned in GitHub
LunaBlack/KGAT-pytorch
pytorch
Mentioned in GitHub
giantonia/KGATv2
tf
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
link-prediction-on-movielens-25mCFKG
Hits@10: 0.8152
nDCG@10: 0.5196
link-prediction-on-yelpCFKG
HR@10: 0.8729
nDCG@10: 0.5826

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
Learning Heterogeneous Knowledge Base Embeddings for Explainable Recommendation | Papers | HyperAI