Knowledge-Enhanced Recommendation with User-Centric Subgraph Network (2403.14377v1)
Abstract: Recommendation systems, as widely implemented nowadays on various platforms, recommend relevant items to users based on their preferences. The classical methods which rely on user-item interaction matrices has limitations, especially in scenarios where there is a lack of interaction data for new items. Knowledge graph (KG)-based recommendation systems have emerged as a promising solution. However, most KG-based methods adopt node embeddings, which do not provide personalized recommendations for different users and cannot generalize well to the new items. To address these limitations, we propose Knowledge-enhanced User-Centric subgraph Network (KUCNet), a subgraph learning approach with graph neural network (GNN) for effective recommendation. KUCNet constructs a U-I subgraph for each user-item pair that captures both the historical information of user-item interactions and the side information provided in KG. An attention-based GNN is designed to encode the U-I subgraphs for recommendation. Considering efficiency, the pruned user-centric computation graph is further introduced such that multiple U-I subgraphs can be simultaneously computed and that the size can be pruned by Personalized PageRank. Our proposed method achieves accurate, efficient, and interpretable recommendations especially for new items. Experimental results demonstrate the superiority of KUCNet over state-of-the-art KG-based and collaborative filtering (CF)-based methods.
- J. J. Levandoski, M. Sarwat, A. Eldawy, and M. F. Mokbel, “Lars: A location-aware recommender system,” in ICDE. IEEE, 2012, pp. 450–461.
- S. Wang, H. Li, C. C. Cao, X.-H. Li, N. N. Fai et al., “Tower bridge net (tb-net): Bidirectional knowledge graph aware embedding propagation for explainable recommender systems,” in ICDE. IEEE, 2022, pp. 3268–3279.
- B. Hui, D. Yan, H. Chen, and W.-S. Ku, “Time-sensitive poi recommendation by tensor completion with side information,” in ICDE. IEEE, 2022, pp. 205–217.
- P. Resnick and H. R. Varian, “Recommender systems,” Communications of the ACM, vol. 40, no. 3, pp. 56–58, 1997.
- F. O. Isinkaye, Y. O. Folajimi, and B. A. Ojokoh, “Recommendation systems: Principles, methods and evaluation,” Egyptian informatics journal, vol. 16, no. 3, pp. 261–273, 2015.
- X. He, L. Liao, H. Zhang, L. Nie, X. Hu, and T.-S. Chua, “Neural collaborative filtering,” in WWW, 2017, pp. 173–182.
- X. He, Z. He, J. Song, Z. Liu, Y.-G. Jiang, and T.-S. Chua, “Nais: Neural attentive item similarity model for recommendation,” TKDE, vol. 30, no. 12, pp. 2354–2366, 2018.
- Y. Koren, R. Bell, and C. Volinsky, “Matrix factorization techniques for recommender systems,” Computer, vol. 42, no. 8, pp. 30–37, 2009.
- S. Rendle, C. Freudenthaler, Z. Gantner, and L. Schmidt-Thieme, “Bayesian personalized ranking from implicit feedback,” in Proc. of Uncertainty in Artificial Intelligence, 2014, pp. 452–461.
- S. Rendle, Z. Gantner, C. Freudenthaler, and L. Schmidt-Thieme, “Fast context-aware recommendations with factorization machines,” in SIGIR, 2011, pp. 635–644.
- Y. Cao, L. Hou, J. Li, and Z. Liu, “Neural collective entity linking,” in COLING, 2018, pp. 675–686.
- F. Zhang, N. J. Yuan, D. Lian, X. Xie, and W.-Y. Ma, “Collaborative knowledge base embedding for recommender systems,” in SIGKDD, 2016, pp. 353–362.
- Q. Guo, F. Zhuang, C. Qin, H. Zhu, X. Xie, H. Xiong, and Q. He, “A survey on knowledge graph-based recommender systems,” TKDE, vol. 34, no. 8, pp. 3549–3568, 2020.
- Q. Wang, Z. Mao, B. Wang, and L. Guo, “Knowledge graph embedding: A survey of approaches and applications,” TKDE, vol. 29, no. 12, pp. 2724–2743, 2017.
- S. Ji, S. Pan, E. Cambria, P. Marttinen, and S. Y. Philip, “A survey on knowledge graphs: Representation, acquisition, and applications,” TNNLS, vol. 33, no. 2, pp. 494–514, 2021.
- X. Wang, X. He, Y. Cao, M. Liu, and T.-S. Chua, “Kgat: Knowledge graph attention network for recommendation,” in SIGKDD, 2019, pp. 950–958.
- H. Wang, F. Zhang, M. Zhang, J. Leskovec, M. Zhao, W. Li, and Z. Wang, “Knowledge-aware graph neural networks with label smoothness regularization for recommender systems,” in SIGKDD, 2019, pp. 968–977.
- Z. Wang, G. Lin, H. Tan, Q. Chen, and X. Liu, “Ckan: collaborative knowledge-aware attentive network for recommender systems,” in SIGIR, 2020, pp. 219–228.
- X. Wang, T. Huang, D. Wang, Y. Yuan, Z. Liu, X. He, and T.-S. Chua, “Learning intents behind interactions with knowledge graph for recommendation,” in WWW, 2021, pp. 878–887.
- T. N. Kipf and M. Welling, “Semi-supervised classification with graph convolutional networks,” in ICLR, 2016.
- W. Hamilton, Z. Ying, and J. Leskovec, “Inductive representation learning on large graphs,” NeurIPS, vol. 30, 2017.
- X. He, K. Deng, X. Wang, Y. Li, Y. Zhang, and M. Wang, “Lightgcn: Simplifying and powering graph convolution network for recommendation,” in SIGIR, 2020, pp. 639–648.
- L. Page, S. Brin, R. Motwani, and T. Winograd, “Pagerank: Bringing order to the web,” Stanford Digital Libraries Working Paper, Tech. Rep., 1997.
- T. H. Haveliwala, “Topic-sensitive pagerank,” in WWW, 2002, pp. 517–526.
- S. Baluja, R. Seth, D. Sivakumar, Y. Jing, J. Yagnik, S. Kumar, D. Ravichandran, and M. Aly, “Video suggestion and discovery for youtube: taking random walks through the view graph,” in WWW, 2008, pp. 895–904.
- T. Bogers, “Movie recommendation using random walks over the contextual graph,” 2010.
- X. Li and H. Chen, “Recommendation as link prediction in bipartite graphs: A graph kernel-based machine learning approach,” Decision Support Systems, vol. 54, no. 2, pp. 880–890, 2013.
- C. Musto, P. Lops, M. de Gemmis, and G. Semeraro, “Context-aware graph-based recommendations exploiting personalized pagerank,” Knowledge-Based Systems, vol. 216, p. 106806, 2021.
- Y. Lin, Z. Liu, M. Sun, Y. Liu, and X. Zhu, “Learning entity and relation embeddings for knowledge graph completion,” in AAAI, vol. 29, no. 1, 2015.
- B. Hu, C. Shi, W. X. Zhao, and P. S. Yu, “Leveraging meta-path based context for top-n recommendation with a neural co-attention model,” in SIGKDD, 2018, pp. 1531–1540.
- H. Wang, F. Zhang, J. Wang, M. Zhao, W. Li, X. Xie, and M. Guo, “Ripplenet: Propagating user preferences on the knowledge graph for recommender systems,” in CIKM, 2018, pp. 417–426.
- A. Bordes, N. Usunier, A. Garcia-Duran, J. Weston, and O. Yakhnenko, “Translating embeddings for modeling multi-relational data,” Advances in neural information processing systems, vol. 26, 2013.
- M. Schlichtkrull, T. N. Kipf, P. Bloem, R. Van Den Berg, I. Titov, and M. Welling, “Modeling relational data with graph convolutional networks,” in The Semantic Web: 15th International Conference, ESWC 2018, Heraklion, Crete, Greece, June 3–7, 2018, Proceedings 15. Springer, 2018, pp. 593–607.
- S. Vashishth, S. Sanyal, V. Nitin, and P. Talukdar, “Composition-based multi-relational graph convolutional networks,” in ICLR, 2019.
- M. Zhang and Y. Chen, “Link prediction based on graph neural networks,” NeurIPS, vol. 31, 2018.
- K. Teru, E. Denis, and W. Hamilton, “Inductive relation prediction by subgraph reasoning,” in ICML. PMLR, 2020, pp. 9448–9457.
- Y. Zhang and Q. Yao, “Knowledge graph reasoning with relational digraph,” in WWW, 2022, pp. 912–924.
- Z. Zhu, Z. Zhang, L.-P. Xhonneux, and J. Tang, “Neural bellman-ford networks: A general graph neural network framework for link prediction,” NeurIPS, vol. 34, pp. 29 476–29 490, 2021.
- J. Gilmer, S. S. Schoenholz, P. F. Riley, O. Vinyals, and G. E. Dahl, “Neural message passing for quantum chemistry,” in ICML, 2017, pp. 1263–1272.
- Y. Zhang, Z. Zhou, Q. Yao, X. Chu, and B. Han, “Adaprop: Learning adaptive propagation for graph neural network based knowledge graph reasoning,” in Proceedings of the 29th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, 2023, pp. 3446–3457.
- D. P. Kingma and J. Ba, “Adam: A method for stochastic optimization,” arXiv:1412.6980, Tech. Rep., 2014.
- W. Krichene and S. Rendle, “On sampled metrics for item recommendation,” in SIGKDD, 2020, pp. 1748–1757.
- Y. Sun, J. Han, X. Yan, P. S. Yu, and T. Wu, “Pathsim: Meta path-based top-k similarity search in heterogeneous information networks,” VLDB, vol. 4, no. 11, pp. 992–1003, 2011.
- J. Piñero, À. Bravo, N. Queralt-Rosinach, A. Gutiérrez-Sacristán, J. Deu-Pons, E. Centeno, J. García-García, F. Sanz, and L. I. Furlong, “Disgenet: a comprehensive platform integrating information on human disease-associated genes and variants,” Nucleic acids research, p. gkw943, 2016.
- X. Wang, T. Jia, C. Wang, K. Xu, Z. Shu, K. Yang, and X. Zhou, “Knowledge graph completion based on tensor decomposition for disease gene prediction,” arXiv:2302.09335, Tech. Rep., 2023.
- Y. Zhang, Q. Yao, L. Yue, X. Wu, Z. Zhang, Z. Lin, and Y. Zheng, “Emerging drug interaction prediction enabled by flow-based graph neural network with biomedical network,” Nature Computational Science, 2023.
- Guangyi Liu (104 papers)
- Quanming Yao (102 papers)
- Yongqi Zhang (33 papers)
- Lei Chen (485 papers)