Enhancing user and item representation with collaborative signals for KG-based recommendation

Yanlin Zhang,Xiaodong Gu
DOI: https://doi.org/10.1007/s00521-024-09419-1
2024-02-21
Neural Computing and Applications
Abstract:Knowledge graph (KG) shows great potential in improving recommendation systems. Recent studies have focused on developing end-to-end models based on graph neural networks. However, these previous approaches do not fully utilize collaborative signals, which are the collective information and feedback provided by multiple users in a recommendation system. Specifically, we think that collaborative signals could be better utilized in two points. Firstly, users’ collective information leads to differences in node importance in KG, offering opportunities to improve item representation. Secondly, an item’s collaborative filtering (CF) feature can reveal the multiple interests of a user, helping to gain fine-grained user representation. In this paper, we highlight the effect of node importance estimation and CF-based item clustering and propose the collaborative signals injection module. We integrate this module in our Collaborative Signals Enhanced Knowledge Graph recommendation (CSEKG) model. CSEKG outperforms state-of-the-art methods like KGIN (Wang et al. in Learning intents behind interactions with knowledge graph for recommendation. CoRR arXiv:2102.07057, 2021) , KGCL (Yang et al. in Proceedings of the 45th international ACM SIGIR conference on research and development in information retrieval, 2022) , and KGRec (Yang et al. in Proceedings of the 29th ACM SIGKDD conference on knowledge discovery and data mining, 2023) through improving recall@20 by 3.86%$$3.86\%$$, 5.15%$$5.15\%$$, and 2.88%$$2.88\%$$ on Last.FM, Amazon-Book and MovieLens-1M, respectively. Furthermore, we flexibly integrate the collaborative signals injection module into existing methods, such as LightGCN, KGAT and KGIN, which improves their performance. Further visualizing recommendation cases show the interpretability benefits brought by collaborative signals.
computer science, artificial intelligence
What problem does this paper attempt to address?