Causal Inference for Knowledge Graph Based Recommendation

被引:11
作者
Wei, Yinwei [1 ]
Wang, Xiang [2 ]
Nie, Liqiang [3 ]
Li, Shaoyu [3 ]
Wang, Dingxian [4 ]
Chua, Tat-Seng [1 ]
机构
[1] Natl Univ Singapore, Sch Comp, Singapore 119077, Singapore
[2] Univ Sci & Technol China, Sch Informat Sci & Technol, Hefei 230052, Anhui, Peoples R China
[3] Shandong Univ, Jinan 250100, Peoples R China
[4] eBay Inc, Search Sci Dept, Ranking Team, Shanghai 200001, Peoples R China
关键词
Causal inference; counterfactual inference; knowledge graph; recommender system;
D O I
10.1109/TKDE.2022.3231352
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Knowledge Graph (KG), as a side-information, tends to be utilized to supplement the collaborative filtering (CF) based recommendation model. By mapping items with the entities in KGs, prior studies mostly extract the knowledge information from the KGs and inject it into the representations of users and items. Despite their remarkable performance, they fail to model the user preference on attribute in the KG, since they ignore that (1) the structure information of KG may hinder the user preference learning, and (2) the user's interacted attributes will result in the bias issue on the similarity scores. With the help of causality tools, we construct the causal-effect relation between the variables in KG-based recommendation and identify the reasons causing the mentioned challenges. Accordingly, we develop a new framework, termed Knowledge Graph-based Causal Recommendation (KGCR), which implements the deconfounded user preference learning and adopts counterfactual inference to eliminate bias in the similarity scoring. Ultimately, we evaluate our proposed model on three datasets, including Amazon-book, LastFM, and Yelp2018 datasets. By conducting extensive experiments on the datasets, we demonstrate that KGCR outperforms several state-of-the-art baselines, such as KGNN-LS (Wang et al., 2019), KGAT (Wang et al., 2019) and KGIN (Wang et al., 2021).
引用
收藏
页码:11153 / 11164
页数:12
相关论文
共 49 条
[1]   Some new estimates of the 'Jensen gap' [J].
Abramovich, Shoshana ;
Persson, Lars-Erik .
JOURNAL OF INEQUALITIES AND APPLICATIONS, 2016, :1-9
[2]  
Bordes A., 2013, P 26 INT C NEURAL IN, P2787
[3]  
Cadene R, 2019, ADV NEUR IN, V32
[4]   Unifying Knowledge Graph Learning and Recommendation: Towards a Better Understanding of User Preferences [J].
Cao, Yixin ;
Wang, Xiang ;
He, Xiangnan ;
Hu, Zikun ;
Chua, Tat-Seng .
WEB CONFERENCE 2019: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2019), 2019, :151-161
[5]   Personalized Video Recommendation Using Rich Contents from Videos [J].
Du, Xingzhong ;
Yin, Hongzhi ;
Chen, Ling ;
Wang, Yang ;
Yang, Yi ;
Zhou, Xiaofang .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2020, 32 (03) :492-505
[6]  
Feng FL, 2021, Arxiv, DOI arXiv:2106.03046
[7]   Should Graph Convolution Trust Neighbors? A Simple Causal Inference Method [J].
Feng, Fuli ;
Huang, Weiran ;
He, Xiangnan ;
Xin, Xin ;
Wang, Qifan ;
Chua, Tat-Seng .
SIGIR '21 - PROCEEDINGS OF THE 44TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL, 2021, :1208-1218
[8]   MAGNN: Metapath Aggregated Graph Neural Network for Heterogeneous Graph Embedding [J].
Fu, Xinyu ;
Zhang, Jiani ;
Men, Ziqiao ;
King, Irwin .
WEB CONFERENCE 2020: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2020), 2020, :2331-2341
[9]  
Gao X., 2017, arXiv
[10]  
Glorot X., 2010, International conference on artificial intelligence and statistics, P249