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

Heterogeneous Information Network Embedding for Recommendation

About

Due to the flexibility in modelling data heterogeneity, heterogeneous information network (HIN) has been adopted to characterize complex and heterogeneous auxiliary data in recommender systems, called HIN based recommendation. It is challenging to develop effective methods for HIN based recommendation in both extraction and exploitation of the information from HINs. Most of HIN based recommendation methods rely on path based similarity, which cannot fully mine latent structure features of users and items. In this paper, we propose a novel heterogeneous network embedding based approach for HIN based recommendation, called HERec. To embed HINs, we design a meta-path based random walk strategy to generate meaningful node sequences for network embedding. The learned node embeddings are first transformed by a set of fusion functions, and subsequently integrated into an extended matrix factorization (MF) model. The extended MF model together with fusion functions are jointly optimized for the rating prediction task. Extensive experiments on three real-world datasets demonstrate the effectiveness of the HERec model. Moreover, we show the capability of the HERec model for the cold-start problem, and reveal that the transformed embedding information from HINs can improve the recommendation performance.

Chuan Shi, Binbin Hu, Wayne Xin Zhao, Philip S. Yu• 2017

Related benchmarks

TaskDatasetResultRank
Node ClassificationDBLP (test)
Macro-F192.8
70
Node ClassificationIMDB (test)
Macro F1 Score47.64
70
Node ClusteringACM
ARI37.13
57
Node ClusteringDBLP
NMI0.7673
39
RecommendationEpinions (test)--
33
Node ClusteringIMDB
NMI1.2
24
Node ClassificationDBLP (80% train)
Macro F192.25
20
Node ClassificationDBLP (20% train)
Macro F190.82
20
Node ClassificationIMDb (20% train)
Macro F1 Score45.61
20
RecommendationAli-Display (test)
NDCG@200.5408
17
Showing 10 of 20 rows

Other info

Follow for update