A Multi-Level Contrastive Learning Framework for Knowledge Graph-Based Recommendation Systems
In recent years, researchers have introduced knowledge information and structural information extracted from knowledge graphs into recommendation systems to improve their performance. However, existing knowledge graph-based recommendation algorithms face challenges in extracting sufficient knowledge and identifying noise unrelated to the recommendation task. To address the aforementioned challenges, we introduce a multi-level contrastive learning framework for knowledge graph-based recommendation systems, MCKGRec. The model integrates information from three levels: user-item interactions, item-entity knowledge, and collaborative knowledge structures. It utilizes lightweight graph convolutional networks to capture interactive signals, graph attention networks to learn knowledge information, and a collaborative knowledge propagation module to capture global structural information. Additionally, a multi-level contrastive learning task is introduced to enhance the recommendation accuracy and robustness. Comprehensive testing across three different datasets confirms that our method significantly enhances the efficacy of recommendation outcomes. By leveraging knowledge graph information and multi-level contrastive learning, MCKGRec better captures complex user-item relationships and filters out irrelevant noise.
Authors
- Haiyuan Zou (ORCID: https://orcid.org/0000-0002-7551-0297)
- Quanwang Wu (ORCID: https://orcid.org/0000-0001-8155-6200)
- Xiaodong Zhang (ORCID: https://orcid.org/0000-0001-9786-7038)
- Tianhao Sun
- Yuxiao Chen (ORCID: https://orcid.org/0009-0000-9856-7053)
Institutions
- Chongqing University (CN)
Publication Details
- Published
- 2024-10-06
- DOI
- https://doi.org/10.1109/smc54092.2024.10831511
- Citations
- 2
- Primary Topic
- Advanced Graph Neural Networks
- Type
- article
- Field-Weighted Citation Impact
- 0.69
Funders
- Natural Science Foundation of Chongqing